https://scholars.lib.ntu.edu.tw/handle/123456789/580600
標題: | Generative Adversarial Networks for Robust Breast Cancer Prognosis Prediction with Limited Data Size | 作者: | Hsu T.-C CHE LIN |
關鍵字: | Deep learning; Diseases; Forecasting; Medical applications; Network architecture; Patient treatment; Semi-supervised learning; Adversarial networks; Breast cancer prognosis; Concordance index; Data augmentation; Disease-specific survivals; Ensemble learning; Improve performance; Limited data size; Diagnosis | 公開日期: | 2020 | 卷: | 2020-July | 起(迄)頁: | 5669-5672 | 來源出版物: | Proceedings of the Annual International Conference of the IEEE Engineering in Medicine and Biology Society, EMBS | 摘要: | Accurate cancer patient prognosis stratification is essential for oncologists to recommend proper treatment plans. Deep learning models are capable of providing good prediction power for such stratification. The main challenge is that only a limited number of labeled patients are available for cancer prognosis. To overcome this, we proposed Wasserstein Generative Adversarial Network-based Deep Adversarial Data Augmentation (wDADA) that leverages generative adversarial networks to perform data augmentation and assist in model training. We used the proposed framework to train our model for predicting disease-specific survival (DSS) of breast cancer patients from the METABRIC dataset. We found that wDADA achieved 0.6726± 0.0278, 0.7538±0.0328, and 0.6507 ±0.0248 in terms of accuracy, AUC, and concordance index in predicting 5-year DSS, respectively, which is comparable to our previously proposed Bimodal model (accuracy: 0.6889±0.0159; AUC: 0.7546± 0.0183; concordance index: 0.6542±0.0120), which needs careful calibration and extensive search on pre-trained network architectures. The flexibility of the proposed wDADA allows us to incorporate it with ensemble learning and semi-supervised learning to further improve performance. Our results indicate that it is possible to utilize generative adversarial networks to train deep models in medical applications, wherein only limited data are available. ? 2020 IEEE. |
URI: | https://www.scopus.com/inward/record.uri?eid=2-s2.0-85091032251&doi=10.1109%2fEMBC44109.2020.9175736&partnerID=40&md5=d59d4ba4768406043c8d4b58532d25d2 https://scholars.lib.ntu.edu.tw/handle/123456789/580600 |
ISSN: | 1557170X | DOI: | 10.1109/EMBC44109.2020.9175736 | SDG/關鍵字: | Deep learning; Diseases; Forecasting; Medical applications; Network architecture; Patient treatment; Semi-supervised learning; Adversarial networks; Breast cancer prognosis; Concordance index; Data augmentation; Disease-specific survivals; Ensemble learning; Improve performance; Limited data size; Diagnosis |
顯示於: | 電機工程學系 |
在 IR 系統中的文件,除了特別指名其著作權條款之外,均受到著作權保護,並且保留所有的權利。