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 条
  • [1] A Variational Approach for Small-Size Lung Nodule Segmentation
    Farag, Amal A.
    2013 IEEE 10TH INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING (ISBI), 2013, : 81 - 84
  • [2] Prediction of Survival in Patients With Esophageal Cancer After Immunotherapy Based on Small-Size Follow-Up Data
    Su, Yuhan
    Huang, Chaofeng
    Yang, Chen
    Lin, Qin
    Chen, Zhong
    IEEE OPEN JOURNAL OF ENGINEERING IN MEDICINE AND BIOLOGY, 2024, 5 : 769 - 782
  • [3] Small-Size Lung Nodule Modeling and Detection with Clinical Evaluation
    Farag, Amal
    Graham, James
    Abdelmunim, Hossam
    Elshazly, Salwa
    El-Mogy, Mohamed
    El-Mogy, Sabry
    Falk, Robert
    Farag, Aly A.
    2012 CAIRO INTERNATIONAL BIOMEDICAL ENGINEERING CONFERENCE (CIBEC), 2012, : 44 - 47
  • [4] Autoencoder-based multimodal prediction of non-small cell lung cancer survival
    Jacob G. Ellen
    Etai Jacob
    Nikos Nikolaou
    Natasha Markuzon
    Scientific Reports, 13
  • [5] Autoencoder-based multimodal prediction of survival for non-small cell lung cancer
    Ellen, Jacob G.
    Jacob, Etai
    Nikolaou, Nikos
    Markuzon, Natasha
    CANCER RESEARCH, 2023, 83 (07)
  • [6] Autoencoder-based multimodal prediction of non-small cell lung cancer survival
    Ellen, Jacob G.
    Jacob, Etai
    Nikolaou, Nikos
    Markuzon, Natasha
    SCIENTIFIC REPORTS, 2023, 13 (01)
  • [7] Surgical and nonsurgical approaches to small-size nonsmall cell lung cancer
    De Ruysscher, Dirk
    Nakagawa, Kazuo
    Asamura, Hisao
    EUROPEAN RESPIRATORY JOURNAL, 2014, 44 (02) : 483 - 494
  • [8] MULTIPLE IMPUTATION IN SMALL CLINICAL DATASETS: A PREDICTION MODEL FOR LUNG CANCER
    Deppen, S. A.
    Putnam, J. B.
    Speroff, T.
    Nesbitt, J.
    Lambright, E.
    Aldrich, M. C.
    AMERICAN JOURNAL OF EPIDEMIOLOGY, 2011, 173 : S87 - S87
  • [9] Lung cancer survival prediction using ensemble data mining on SEER data
    Agrawal, Ankit
    Misra, Sanchit
    Narayanan, Ramanathan
    Polepeddi, Lalith
    Choudhary, Alok
    SCIENTIFIC PROGRAMMING, 2012, 20 (01) : 29 - 42
  • [10] Acute Myocardial Infarction Associated with Small-size Lung Cancer in a Young Woman
    Mizumura, Kenji
    Sugane, Toshio
    Ozaki, Shunsuke
    Ohta, Hiroshi
    Kouzu, Yutaka
    Sekiyama, Akiko
    Hiranuma, Hisato
    Terakado, Masahiro
    Sato, Maki
    Shimizu, Tetsuo
    Kobayashi, Tomoko
    Takahashi, Noriaki
    Hashimoto, Shu
    INTERNAL MEDICINE, 2011, 50 (18) : 1997 - 2002