Prediction of Radiotherapy-Induced Hematologic Toxicity in Cervical Cancer with Cohort-Aware Framework

Abstract

Hematologic toxicity (HT) is a major dose-limiting complication of pelvic radiotherapy for cervical cancer. Although radiomic and dosiomic features improve HT prediction beyond dosimetric metrics, their performance is highly sensitive to contour variability, limiting generalizability. We developed a cohort-aware representation-learning framework to address this challenge. We retrospectively analyzed 152 cervical cancer patients treated with pelvic radiotherapy without concurrent chemotherapy. Patients were divided into two cohorts based on the operators performing pelvic bone segmentation. HT prediction models were developed using cohort-specific training, pooled training, statistical harmonization, and a cohort-aware neural network (CANN) that learns shared and cohort-specific representations with contrastive regularization. Performance was evaluated using cross-validation and an independent test set. Cohort-specific models achieved test AUCs of 0.77 and 0.71, outperforming a dosimetry-only model (AUC=0.58). Directly pooling cohorts reduced performance (test AUC=0.64). Statistical harmonization provided limited benefit, while adversarial and correlation-based alignment further degraded performance. CANN achieved the best balance between robustness and generalizability (test AUC=0.72), with ablation studies confirming the importance of cohort-specific representations and contrastive alignment. These results demonstrate that cohort-aware representation learning effectively mitigates contour variability and improves the generalizability of radiomic and dosiomic models for HT prediction.