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 条
  • [21] Nomogram prediction for the survival of the patients with small cell lung cancer
    Pan, Hui
    Shi, Xiaoshun
    Xiao, Dakai
    He, Jiaxi
    Zhang, Yalei
    Liang, Wenhua
    Zhao, Zhi
    Guo, Zhihua
    Zou, Xusen
    Zhang, Jinxin
    He, Jianxing
    JOURNAL OF THORACIC DISEASE, 2017, 9 (03) : 507 - 518
  • [22] Optimal resection strategies for small-size lung cancer: Is a wedge enough? Is lobectomy too much?
    Na, Kwon Joong
    Kim, Young Tae
    JTCVS OPEN, 2023, 16 : 17 - 21
  • [23] A Novel Aggregated Multiple Imputation Approach for Enhanced Survival Prediction and Classification on Breast Cancer and Lung Cancer Data
    Deepa, P.
    Gunavathi, C.
    IEEE ACCESS, 2024, 12 : 189102 - 189121
  • [24] Survival prediction of non-small cell lung cancer by deep learning model integrating clinical and positron emission tomography data
    Kang, Sae-Ryung
    Oh, Seungwon
    Oh, In-Jae
    Min, Jung-Joon
    Bom, Hee-Seung
    Yang, Hyung-Jeong
    Lee, Guee-Sang
    Kim, Soo-Hyung
    Kim, Min Soo
    CLINICAL CANCER RESEARCH, 2021, 27 (05)
  • [25] Prognosis prediction of patients with malignant pleural mesothelioma using conditional variational autoencoder on 3D PET images and clinical data
    Matsuo, Hidetoshi
    Kitajima, Kazuhiro
    Kono, Atsushi K.
    Kuribayashi, Kozo
    Kijima, Takashi
    Hashimoto, Masaki
    Hasegawa, Seiki
    Yamakado, Koichiro
    Murakami, Takamichi
    MEDICAL PHYSICS, 2023, 50 (12) : 7548 - 7557
  • [26] Gene Profiling of Clinical Routine Biopsies and Prediction of Survival in Non-Small Cell Lung Cancer
    Baty, Florent
    Facompre, Michael
    Kaiser, Sergio
    Schumacher, Martin
    Pless, Miklos
    Bubenclorf, Lukas
    Savic, Spasenija
    Marrer, Estelle
    Budach, Wolfgang
    Buess, Martin
    Kehren, Jeanne
    Tamm, Michael
    Brutsche, Martin H.
    AMERICAN JOURNAL OF RESPIRATORY AND CRITICAL CARE MEDICINE, 2010, 181 (02) : 181 - 188
  • [27] Multiparameter analysis using cell cycle biomarkers for small-size lung adenocarcinoma: Prognostic implications
    Haruki, Tomohiro
    Shomori, Kohei
    Shiomi, Tatsushi
    Taniguchi, Yuji
    Nakamura, Hiroshige
    Ito, Hisao
    ONCOLOGY REPORTS, 2012, 28 (03) : 915 - 922
  • [28] Preoperative CT findings of small-size lung cancer with recurrence during 1 year after surgical lung resection
    Kondo, Tetsuro
    Murakami, Shuuji
    Saito, Haruhiro
    Oshita, Hirofumi
    Suga, Yasuhiro
    Ito, Hiroyuki
    Tsuboi, Masahiro
    Nakayama, Haruhiko
    Yokose, Tomoyuki
    Kameda, Youichi
    Noda, Kazumasa
    Yamada, Kouzo
    JOURNAL OF THORACIC ONCOLOGY, 2009, 4 (09) : S482 - S482
  • [29] Immunohistochemistry and Radiomic Features for Survival Prediction in Small Cell Lung Cancer
    Gkika, Eleni
    Benndorf, Matthias
    Oerther, Benedict
    Mohammad, Farid
    Beitinger, Susanne
    Adebahr, Sonja
    Carles, Montserrat
    Schimek-Jasch, Tanja
    Zamboglou, Constantinos
    Frye, Bjorn C.
    Bamberg, Fabian
    Waller, Cornelius F.
    Werner, Martin
    Grosu, Anca L.
    Nestle, Ursula
    Kayser, Gian
    FRONTIERS IN ONCOLOGY, 2020, 10
  • [30] Immunohistochemistry and Radiomics Features for Survival Prediction in Small Cell Lung Cancer
    Gkika, E.
    Benndorf, M.
    Oerther, B.
    Mohammad, F.
    Beitinger, S.
    Adebahr, S.
    Carles, M.
    Schimek-Jasch, T.
    Zamboglou, C.
    Frye, B. C.
    Bamberg, F.
    Waller, C.
    Grosu, A.
    Nestle, U.
    Kayser, G.
    INTERNATIONAL JOURNAL OF RADIATION ONCOLOGY BIOLOGY PHYSICS, 2020, 108 (03): : E154 - E154