A deep learning framework for lesion-level treatment response prediction in hodgkin lymphoma using PET/CT tensor
Mahdie Jajroudi1,2, Hossein Jamalirad2, Milad Enferadi3
1Pharmaceutical Research Center, Pharmaceutical Technology Institute, Mashhad University of Medical Sciences, Mashhad, Iran.
Background:
Accurate prediction of treatment response in Hodgkin lymphoma (HL) is crucial for personalized therapy. The Tensor Radiomics (TR) paradigm advances traditional radiomics by producing and analyzing diverse feature variations, employing tensors computed across multiple parameter combinations to optimize radiomics feature extraction and improve tumor characterization. However, the complexity and variability of these features pose significant challenges for analysis and clinical application in HL research. This study introduces the Tensor Radiomics Network (TR-NET), a deep learning framework that automates feature selection from tensor radiomics data and predicts treatment response. It also incorporates explainable AI techniques to identify the most influential features.
Methods:
We analyzed 420 lesions from 70 HL patients using 18F-FDG PET/CT. Tensor radiomic feature sets were generated by varying bin widths for PET (0.2, 0.4, 0.6, 0.8) and CT (10, 25, 50) images. Extracted features encompassed first-order, shape, texture, and filter classes (Laplacian of Gaussian, Wavelet, Square, Square Root, Logarithm, Exponential, Gradient, Local Binary Pattern (LBP) 2D, LBP 3D for each lesion, totaling 13,790 features. We utilized machine learning models, including Random Forest (RF), XGBoost, Support Vector Machine (SVM), and the deep learning model TR-NET for analysis. SHAP (SHapley Additive exPlanations), an explainable AI method, was applied to interpret the predictions of the best-performing model.
Results:
TR-NET achieved a predictive performance, with an AUC-ROC of 0.8470 (95% CI: 0.7669-0.9185), a sensitivity of 82.2%, and a specificity of 78.2%, offering a balanced classification characteristic. In comparison, XGBoost achieved an AUC of 0.8160 (95% CI: 0.7457-0.8599), characterized by high sensitivity (93.3%) but lower specificity (66.2%), indicating a tendency towards false positives. Random Forest and SVM also showed high sensitivity (92% and 91.1%), but both suffered from significantly reduced specificity (29.5% and 51.3%).
Conclusion:
TR-NET demonstrates enhanced predictive accuracy and clinical relevance compared to classic machine learning, supporting personalized treatment strategies and the integration of explainable AI in oncology.