Robust and interpretable deep learning system for prognostic stratification of extranodal natural killer/T-cell lymphoma

Chong Jiang; Zekun Jiang; Xinyu Zhang; Linhao Qu; Kexue Fu; Yue Teng; Ruihe Lai; Rui Guo; Chongyang Ding; Kang Li; Rong Tian

doi:10.1007/s00259-024-07024-x

Robust and interpretable deep learning system for prognostic stratification of extranodal natural killer/T-cell lymphoma

Eur J Nucl Med Mol Imaging. 2024 Dec 23. doi: 10.1007/s00259-024-07024-x. Online ahead of print.

Authors

Chong Jiang^#¹, Zekun Jiang^#^{2

3}, Xinyu Zhang⁴, Linhao Qu⁵, Kexue Fu⁵, Yue Teng⁶, Ruihe Lai⁶, Rui Guo⁷, Chongyang Ding⁸, Kang Li^{9

10}, Rong Tian¹¹

Affiliations

¹ Department of Nuclear Medicine, West China Hospital, Sichuan University, No.37, Guoxue Alley, Chengdu, Sichuan, 610041, China.
² West China Biomedical Big Data Center, West China Hospital, Sichuan University, Chengdu, China.
³ Med-X Center for Informatics, Sichuan University, Chengdu, China.
⁴ Department of Nuclear Medicine, Ruijin Hospital, Shanghai Jiao Tong University School of Medicine, No. 197 Ruijin Second Road, Shanghai, 200025, China.
⁵ School of Basic Medical Sciences, Fudan University, Shanghai, China.
⁶ Department of Nuclear Medicine, Nanjing Drum Tower Hospital, The Affiliated Hospital of Nanjing University Medical School, Nanjing, China.
⁷ Department of Nuclear Medicine, Ruijin Hospital, Shanghai Jiao Tong University School of Medicine, No. 197 Ruijin Second Road, Shanghai, 200025, China. gr11734@rjh.com.cn.
⁸ Department of Nuclear Medicine, The First Affiliated Hospital of Nanjing Medical University, Jiangsu Province Hospital, No.321, Zhongshan Road, Nanjing, Jiangsu, 210008, China. chongyangding@163.com.
⁹ West China Biomedical Big Data Center, West China Hospital, Sichuan University, Chengdu, China. likang@wchscu.cn.
¹⁰ Med-X Center for Informatics, Sichuan University, Chengdu, China. likang@wchscu.cn.
¹¹ Department of Nuclear Medicine, West China Hospital, Sichuan University, No.37, Guoxue Alley, Chengdu, Sichuan, 610041, China. rongtiannuclear@126.com.

^# Contributed equally.

PMID: 39714634
DOI: 10.1007/s00259-024-07024-x

Abstract

Purpose: Extranodal natural killer/T-cell lymphoma (ENKTCL) is an hematologic malignancy with prognostic heterogeneity. We aimed to develop and validate DeepENKTCL, an interpretable deep learning prediction system for prognosis risk stratification in ENKTCL.

Methods: A total of 562 patients from four centers were divided into the training cohort, validation cohort and test cohort. DeepENKTCL combined a tumor segmentation model, a PET/CT fusion model, and prognostic prediction models. RadScore and TopoScore were constructed using radiomics and topological features derived from fused images, with SHapley Additive exPlanations (SHAP) analysis enhancing interpretability. The final prognostic models, termed FusionScore, were developed for predicting progression-free survival (PFS) and overall survival (OS). Performance was assessed using area under the receiver operator characteristic curve (AUC), time-dependent C-index, clinical decision curves (DCA), and Kaplan-Meier (KM) curves.

Results: The tumor segmentation model accurately delineated the tumor lesions. RadScore (AUC: 0.908 for PFS, 0.922 for OS in validation; 0.822 for PFS, 0.867 for OS in test) and TopoScore (AUC: 0.756 for PFS, 0.805 for OS in validation; 0.689 for PFS, 0.769 for OS in test) both exhibited potential prognostic capability. The time-dependent C-index (0.897 for PFS, 0.928 for OS in validation; 0.894 for PFS, 0.868 for OS in test) and DCA indicated that FusionScore offers significant prognostic performance and superior net clinical benefits compared to existing models. KM survival analysis showed that higher FusionScores correlated with poorer PFS and OS across all cohorts.

Conclusion: DeepENKTCL provided a robust and interpretable framework for ENKTCL prognosis, with the potential to improve patient outcomes and guide personalized treatment.

Keywords: Deep learning; Extranodal NK/T-cell lymphoma; Machine learning; PET/CT; Radiomics; Survival analysis; Topological data analysis.

Abstract

Grants and funding