Survival Prediction of Lung Cancer Using Small-Size Clinical Data with a Multiple Task Variational Autoencoder

被引:9
|
作者
Thanh-Hung Vo [1 ]
Lee, Guee-Sang [1 ]
Yang, Hyung-Jeong [1 ]
Oh, In-Jae [2 ,3 ]
Kim, Soo-Hyung [1 ]
Kang, Sae-Ryung [3 ,4 ]
机构
[1] Chonnam Natl Univ, Dept Artificial Intelligence Convergence, Gwangju 61186, South Korea
[2] Chonnam Natl Univ Med Sch, Dept Internal Med, Jeonnam 58128, South Korea
[3] Hwasun Hosp, Jeonnam 58128, South Korea
[4] Chonnam Natl Univ Med Sch, Dept Nucl Med, Jeonnam 58128, South Korea
基金
新加坡国家研究基金会;
关键词
survival analysis; lung cancer; variational autoencoder; multiple tasks; prognosis;
D O I
10.3390/electronics10121396
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Due to the increase of lung cancer globally, and particularly in Korea, survival analysis for this type of cancer has gained prominence in recent years. For this task, mathematical and traditional machine learning approaches are commonly used by medical doctors. While the deep learning approach has had proven success in computer vision tasks, natural language processing and other AI techniques are also adopted for this task. Due to the privacy issues and management process, data in medicine are difficult to collect, which leads to a paucity of samples. The small number of samples makes it difficult to use deep learning and renders this approach unusable. In this investigation, we propose a network architecture that combines a variational autoencoder (VAE) with the typical DNN architecture to solve the survival analysis task. With a training size of n = 4107, MVAESA achieves a C-index of 0.722 while CoxCC, CoxPH, and CoxTime achieved scores of 0.713, 0.703, and 0.710, respectively. With a small training size of n = 379, MVAESA achieves a C-index of 0.707, compared with 0.689, 0.688 and 0.690 for CoxCC, CoxPH, and CoxTime, respectively. The results show that the combination of a VAE with a target task makes the network more stable and that the network could be trained using a small-sized sample.
引用
收藏
页数:13
相关论文
共 50 条
  • [41] The Presence of Small-Size Circulating Tumor Cells Predicts Worse Prognosis in Non-Small Cell Lung Cancer Patients
    Sun, Qianqian
    Li, Weiqing
    Yang, Donghua
    Lin, Peter Ping
    Zhang, Lina
    Guo, Huiqin
    ARCHIVES OF PATHOLOGY & LABORATORY MEDICINE, 2025, 149 (01) : 39 - 49
  • [42] Machine Learning and Deep Learning Approaches in Breast Cancer Survival Prediction Using Clinical Data
    Kalafi, E. Y.
    Nor, N. A. M.
    Taib, N. A.
    Ganggayah, M. D.
    Town, C.
    Dhillon, S. K.
    FOLIA BIOLOGICA, 2019, 65 (5-6) : 212 - 220
  • [43] Comparison of different approaches to modeling early tumor size dynamics for accurate prediction of survival in non-small cell lung cancer (NSCLC) clinical trials
    Kosinsky, Yuri
    Azarov, Ivan
    Chu, Lulu
    Helmlinger, Gabriel
    Peskov, Kirill
    Aksenov, Sergey
    JOURNAL OF PHARMACOKINETICS AND PHARMACODYNAMICS, 2018, 45 : S123 - S123
  • [44] Reliable channel reservoir characterization and uncertainty quantification using variational autoencoder and ensemble smoother with multiple data assimilation
    Ahn, Youngbin
    Choe, Jonggeun
    JOURNAL OF PETROLEUM SCIENCE AND ENGINEERING, 2022, 209
  • [45] A survival analysis using physique-adjusted tumor size of non-small cell lung cancer
    Ozeki, Naoki
    Fukui, Takayuki
    Kawaguchi, Koji
    Nakamura, Shota
    Hakiri, Shuhei
    Kato, Taketo
    Hirakawa, Akihiro
    Yokoi, Kohei
    INTERNATIONAL JOURNAL OF CLINICAL ONCOLOGY, 2018, 23 (02) : 266 - 274
  • [46] A survival analysis using physique-adjusted tumor size of non-small cell lung cancer
    Naoki Ozeki
    Takayuki Fukui
    Koji Kawaguchi
    Shota Nakamura
    Shuhei Hakiri
    Taketo Kato
    Akihiro Hirakawa
    Kohei Yokoi
    International Journal of Clinical Oncology, 2018, 23 : 266 - 274
  • [47] Metastatic Patterns of Mediastinal Lymph Nodes in Small-Size Non-small Cell Lung Cancer (T1b)
    Wu, Yijun
    Han, Chang
    Gong, Liang
    Wang, Zhile
    Liu, Jianghao
    Liu, Xinyu
    Chen, Xinyi
    Chong, Yuming
    Liang, Naixin
    Li, Shanqing
    FRONTIERS IN SURGERY, 2020, 7
  • [48] Prediction of Survival Rate from Non-Small Cell Lung Cancer using Improved Random Forest
    Nanda, Pranamita
    Duraipandian, N.
    PROCEEDINGS OF THE 5TH INTERNATIONAL CONFERENCE ON INVENTIVE COMPUTATION TECHNOLOGIES (ICICT-2020), 2020, : 93 - 97
  • [49] Original Article Prediction of Clinical Outcome for All Stages and Multiple Cell Types of Non-small Cell Lung Cancer in Five Countries Using Lung Cancer Prognostic Index
    Chen, Tiehua
    Chen, Luming
    EBIOMEDICINE, 2014, 1 (2-3): : 156 - 166
  • [50] Prediction of Survival in Thyroid Cancer Using Data Mining Technique
    Jajroudi, M.
    Baniasadi, T.
    Kamkar, L.
    Arbabi, F.
    Sanei, M.
    Ahmadzade, M.
    TECHNOLOGY IN CANCER RESEARCH & TREATMENT, 2014, 13 (04) : 353 - 359