Deep learning model integrating positron emission tomography and clinical data for prognosis prediction in non-small cell lung cancer patients

BMC Bioinformatics. 2023 Feb 6;24(1):39. doi: 10.1186/s12859-023-05160-z.

Abstract

Background: Lung cancer is the leading cause of cancer-related deaths worldwide. The majority of lung cancers are non-small cell lung cancer (NSCLC), accounting for approximately 85% of all lung cancer types. The Cox proportional hazards model (CPH), which is the standard method for survival analysis, has several limitations. The purpose of our study was to improve survival prediction in patients with NSCLC by incorporating prognostic information from F-18 fluorodeoxyglucose positron emission tomography (FDG PET) images into a traditional survival prediction model using clinical data.

Results: The multimodal deep learning model showed the best performance, with a C-index and mean absolute error of 0.756 and 399 days under a five-fold cross-validation, respectively, followed by ResNet3D for PET (0.749 and 405 days) and CPH for clinical data (0.747 and 583 days).

Conclusion: The proposed deep learning-based integrative model combining the two modalities improved the survival prediction in patients with NSCLC.

Keywords: Deep learning; FDG PET; Lung cancer; Multimodal learning; Survival prediction.

MeSH terms

  • Carcinoma, Non-Small-Cell Lung* / diagnostic imaging
  • Deep Learning*
  • Fluorodeoxyglucose F18
  • Humans
  • Lung Neoplasms* / diagnostic imaging
  • Positron-Emission Tomography
  • Radiopharmaceuticals
  • Retrospective Studies

Substances

  • Fluorodeoxyglucose F18
  • Radiopharmaceuticals