PET and CT based DenseNet outperforms advanced deep learning models for outcome prediction of oropharyngeal cancer.
Authors
Affiliations (5)
Affiliations (5)
- Department of Radiation Oncology, University of Groningen, University Medical Center Groningen, Groningen, Netherlands. Electronic address: [email protected].
- Department of Radiation Oncology, University of Groningen, University Medical Center Groningen, Groningen, Netherlands; Machine Learning Lab, Data Science Center in Health (DASH), Groningen, Netherlands; Bernoulli Institute for Mathematics, Computer Science and Artificial Intelligence, University of Groningen, Groningen, Netherlands.
- Department of Radiation Oncology, University of Groningen, University Medical Center Groningen, Groningen, Netherlands; Department of Radiation Oncology, University of Texas MD Anderson Cancer Center, Houston, TX USA.
- Department of Radiation Oncology, University of Groningen, University Medical Center Groningen, Groningen, Netherlands.
- Department of Radiation Oncology, University of Groningen, University Medical Center Groningen, Groningen, Netherlands; Machine Learning Lab, Data Science Center in Health (DASH), Groningen, Netherlands.
Abstract
In the HECKTOR 2022 challenge set [1], several state-of-the-art (SOTA, achieving best performance) deep learning models were introduced for predicting recurrence-free period (RFP) in head and neck cancer patients using PET and CT images. This study investigates whether a conventional DenseNet architecture, with optimized numbers of layers and image-fusion strategies, could achieve comparable performance as SOTA models. The HECKTOR 2022 dataset comprises 489 oropharyngeal cancer (OPC) patients from seven distinct centers. It was randomly divided into a training set (n = 369) and an independent test set (n = 120). Furthermore, an additional dataset of 400 OPC patients, who underwent chemo(radiotherapy) at our center, was employed for external testing. Each patients' data included pre-treatment CT- and PET-scans, manually generated GTV (Gross tumour volume) contours for primary tumors and lymph nodes, and RFP information. The present study compared the performance of DenseNet against three SOTA models developed on the HECKTOR 2022 dataset. When inputting CT, PET and GTV using the early fusion (considering them as different channels of input) approach, DenseNet81 (with 81 layers) obtained an internal test C-index of 0.69, a performance metric comparable with SOTA models. Notably, the removal of GTV from the input data yielded the same internal test C-index of 0.69 while improving the external test C-index from 0.59 to 0.63. Furthermore, compared to PET-only models, when utilizing the late fusion (concatenation of extracted features) with CT and PET, DenseNet81 demonstrated superior C-index values of 0.68 and 0.66 in both internal and external test sets, while using early fusion was better in only the internal test set. The basic DenseNet architecture with 81 layers demonstrated a predictive performance on par with SOTA models featuring more intricate architectures in the internal test set, and better performance in the external test. The late fusion of CT and PET imaging data yielded superior performance in the external test.