Development of learning-based predictive models for radiation-induced atrial fibrillation in non-small cell lung cancer patients by integrating patient-specific clinical, dosimetry, and diagnostic information

Background and purpose

Radiotherapy (RT) in non-small cell lung cancer (NSCLC) can induce cardiac adverse events, including atrial fibrillation (AF), despite advanced RT. This study integrates patient-specific information to develop learning-based models to predict the incidence of AF following NSCLC chemoradiotherapy (CRT) and evaluates these models using institutional and external datasets.

Materials and methods

Institutional and external patient cohorts consisted of 321 and 187 NSCLC datasets who received definitive CRT, including 17 and 6 AF incidences, respectively. The network input had 159 features with clinical, dosimetry, and diagnostic. The class imbalance was mitigated by synthetic minority oversampling technique. To handle various types of input features, machine learning-based model adopted an intervention technique that chose one feature with the largest weight at each dosimetry sub-group in feature selection process, while deep learning-based model employed a hybrid architecture assigning different types of networks to corresponding input paths. Performance was assessed by area under the curve (AUC). The key features were investigated for the machine and deep learning-based models.

Results

The hybrid deep learning model outperformed the machine learning-based algorithm in internal validation (AUC: 0.817 vs. 0.801) and produced more consistent performance in external validation (AUC: 0.806 vs. 0.776). Importantly, maximum dose to heart and sinoatrial node (SAN) were found to be the key features for both learning-based models in external and internal validations.

Conclusions

The learning-based predictive models showed consistent prediction performance across internal and external cohorts, identifying maximum heart and SAN dose as key features for the incidence of AF.