Deep segmentation networks predict survival of non-small cell lung cancer.

Journal: Scientific reports
Published Date:

Abstract

Non-small-cell lung cancer (NSCLC) represents approximately 80-85% of lung cancer diagnoses and is the leading cause of cancer-related death worldwide. Recent studies indicate that image-based radiomics features from positron emission tomography/computed tomography (PET/CT) images have predictive power for NSCLC outcomes. To this end, easily calculated functional features such as the maximum and the mean of standard uptake value (SUV) and total lesion glycolysis (TLG) are most commonly used for NSCLC prognostication, but their prognostic value remains controversial. Meanwhile, convolutional neural networks (CNN) are rapidly emerging as a new method for cancer image analysis, with significantly enhanced predictive power compared to hand-crafted radiomics features. Here we show that CNNs trained to perform the tumor segmentation task, with no other information than physician contours, identify a rich set of survival-related image features with remarkable prognostic value. In a retrospective study on pre-treatment PET-CT images of 96 NSCLC patients before stereotactic-body radiotherapy (SBRT), we found that the CNN segmentation algorithm (U-Net) trained for tumor segmentation in PET and CT images, contained features having strong correlation with 2- and 5-year overall and disease-specific survivals. The U-Net algorithm has not seen any other clinical information (e.g. survival, age, smoking history, etc.) than the images and the corresponding tumor contours provided by physicians. In addition, we observed the same trend by validating the U-Net features against an extramural data set provided by Stanford Cancer Institute. Furthermore, through visualization of the U-Net, we also found convincing evidence that the regions of metastasis and recurrence appear to match with the regions where the U-Net features identified patterns that predicted higher likelihoods of death. We anticipate our findings will be a starting point for more sophisticated non-intrusive patient specific cancer prognosis determination. For example, the deep learned PET/CT features can not only predict survival but also visualize high-risk regions within or adjacent to the primary tumor and hence potentially impact therapeutic outcomes by optimal selection of therapeutic strategy or first-line therapy adjustment.

Authors

  • Stephen Baek
    Department of Mechanical and Industrial Engineering, The University of Iowa, Iowa City, IA 52242, USA.
  • Yusen He
    University of Iowa, Department of Industrial and Systems Engineering, Iowa City, IA, 52242, United States.
  • Bryan G Allen
    Department of Radiation Oncology, University of Iowa Hospitals and Clinics, Iowa City, IA, 52242, USA.
  • John M Buatti
    Department of Radiation Oncology, Carver College of Medicine, University of Iowa Carver College of Medicine, LL-W Pomerantz Family Pavilion, 200 Hawkins Drive, Iowa City, IA, 52242-1089, USA.
  • Brian J Smith
    Department of Biostatistics, University of Iowa, 145 N. Riverside Drive, 100 CPHB, Iowa City, IA, 52242, USA.
  • Ling Tong
  • Zhiyu Sun
    University of Iowa, Department of Industrial and Systems Engineering, Iowa City, IA, 52242, United States.
  • Jia Wu
  • Maximilian Diehn
    Department of Radiation Oncology, Stanford University School of Medicine, Stanford, California.
  • Billy W Loo
    Department of Radiation Oncology, Stanford University School of Medicine, Stanford, California.
  • Kristin A Plichta
    University of Iowa, Department of Radiation Oncology, Iowa City, IA, 52242, United States.
  • Steven N Seyedin
    University of Iowa, Department of Radiation Oncology, Iowa City, IA, 52242, United States.
  • Maggie Gannon
    University of Iowa, Department of Radiation Oncology, Iowa City, IA, 52242, United States.
  • Katherine R Cabel
    University of Iowa, Department of Radiation Oncology, Iowa City, IA, 52242, United States.
  • Yusung Kim
    Department of Radiation Oncology, University of Iowa Hospitals and Clinics, Iowa City, IA, 52242, USA.
  • Xiaodong Wu
    Department of Electrical and Computer Engineering, The University of Iowa, Iowa City, IA, 52242, USA.