Opto-Electronic Engineering, Volume. 51, Issue 4, 240011-1(2024)
Graph neural network-based WSI cancer survival prediction method
Figures & Tables(10)
Fig. 1. Architecture of the BC GraphSurv model mainly including modules such as WSI preprocessing, (a) Feature extraction and graph structure generation, (b) WA-GAT branch, (c) MP-GCN branches, and (d) Feature fusion
Fig. 4. Comparison of KM curves and P-values of several commonly used methods
Table 1. Comparison of experimental results of feature extraction network
View tableView in ArticleTable 1. Comparison of experimental results of feature extraction network
C-index SD Parameters/MB ResNet50 0.7225 0.011 97.49 EfficientNet-b5 0.7306 0.022 115.93 HF-Net 0.7950 0.013 113.72
Table 2. Comparison of ablation experiment results
View tableView in ArticleTable 2. Comparison of ablation experiment results
WA-GAT MP-GCN GAT GCN GF C-index SD √ 0.7506 0.021 √ 0.7217 0.019 √ 0.7842 0.007 √ 0.7812 0.008 √ √ 0.7910 0.003 √ √ √ 0.7950 0.013
Table 3. First comparison experiments of different methods
View tableView in ArticleTable 3. First comparison experiments of different methods
Method BRCA KIRC C-index SD C-index SD MLP 0.6176 0.027 0.5862 0.019 Attention-MIL[ 26 ] (2018)0.7091 0.052 0.6590 0.044 WSISA[ 8 ] (2017)0.6802 0.083 0.6151 0.057 DeepGraphSurv[ 12 ] (2018)0.7402 0.012 0.6945 0.045 Patch GCN[ 13 ] (2021)0.7514 0.036 0.6775 0.067 H2-MIL[ 27 ] (2022)0.7338 0.055 0.6915 0.024 Tea-Graph[ 28 ] (2022)0.7541 0.021 0.7109 0.023 HEAT[ 3 ] (2023)0.7529 0.009 0.7011 0.018 BC-GraphSurv 0.7950 0.013 0.7458 0.020
Table 4. Second comparison experiments of different methods
View tableView in ArticleTable 4. Second comparison experiments of different methods
Method BRCA KIRC C-index SD C-index SD MLP 0.6176 0.027 0.5862 0.019 MIL-Transformer[ 29 ] (2021)0.6905 0.046 0.6642 0.029 SurvTRACE[ 30 ] (2022)0.7382 0.019 0.6974 0.027 BC-GraphSurv 0.7950 0.013 0.7458 0.020
Tools
Get Citation
Copy Citation Text
Shijie Ye, Yongxiong Wang. Graph neural network-based WSI cancer survival prediction method[J]. Opto-Electronic Engineering, 2024, 51(4): 240011-1
Paper Information
Category: Article
Received: Jan. 9, 2024
Accepted: Mar. 12, 2024
Published Online: Jul. 8, 2024
The Author Email: Wang Yongxiong (王永雄)
© Copyright 2018-2021 | Chinese Laser Press.
All Rights Reserved 沪ICP备15018463号-20