https://scholars.lib.ntu.edu.tw/handle/123456789/632214
Title: | Training with Small Medical Data: Robust Bayesian Neural Networks for Colon Cancer Overall Survival Prediction | Authors: | Hsu T.-C CHE LIN |
Issue Date: | 2021 | Start page/Pages: | 2030-2033 | Source: | Proceedings of the Annual International Conference of the IEEE Engineering in Medicine and Biology Society, EMBS | Abstract: | Fast and accurate cancer prognosis stratification models are essential for treatment designs. Large labeled patient data can power advanced deep learning models to obtain precise predictions. However, since fully labeled patient data are hard to acquire in practical scenarios, deep models are prone to make non-robust predictions biased toward data partition and model hyper-parameter selection. Given a small training set, we applied the systems biology feature selector in our previous study to avoid over-fitting and select 18 prognostic biomarkers. Combined with three other clinical features, we trained Bayesian binary classifiers to predict the 5-year overall survival (OS) of colon cancer patients in this study. Results showed that Bayesian models could provide better and more robust predictions compared to their non-Bayesian counterparts. Specifically, in terms of the area under the receiver operating characteristic curve (AUC), macro F1-score (maF 1 ), and concordance index (CI), we found that the Bayesian bimodal neural network (late fusion) classifier (B-Bimodal) achieved the best results (AUC: 0.8083 ± 0.0736; maF 1 : 0.7300 ± 0.0659; CI: 0.7238 ± 0.0440). The single modal Bayesian neural network classifier (B-Concat) fed with concatenated patient data (early fusion) achieved slightly worse but more robust performance in terms of AUC and CI (AUC: 0.7105 ± 0.0692; maF 1 : 0.7156 ± 0.0690; CI: 0.6627 ± 0.0558). Such robustness is essential to training learning models with small medical data. © 2021 IEEE. |
URI: | https://www.scopus.com/inward/record.uri?eid=2-s2.0-85122534073&doi=10.1109%2fEMBC46164.2021.9630698&partnerID=40&md5=293c3463e4efcbd9085b25aa48c4a910 https://scholars.lib.ntu.edu.tw/handle/123456789/632214 |
ISSN: | 1557170X | DOI: | 10.1109/EMBC46164.2021.9630698 | SDG/Keyword: | Bayesian networks; Deep learning; Diagnosis; Forecasting; Hospital data processing; Neural networks; Bayesian; Bayesian neural networks; Colon cancer; Concordance index; F1 scores; Learning models; Medical data; Overall survival; Robust predictions; Survival prediction; Diseases; Bayes theorem; colon tumor; human; receiver operating characteristic; systems biology; Bayes Theorem; Colonic Neoplasms; Humans; Neural Networks, Computer; ROC Curve; Systems Biology |
Appears in Collections: | 電機工程學系 |
Items in DSpace are protected by copyright, with all rights reserved, unless otherwise indicated.