Prospective validation of a machine learning model for applicator and hybrid interstitial needle selection in high-dose-rate (HDR) cervical brachytherapy

PURPOSE To Demonstrate the clinical validation of a machine learning (ML) model for applicator and interstitial needle prediction in gynecologic brachytherapy through a prospective clinical study in a single institution. METHODS The study included cervical cancer patients receiving high-dose-rate brachytherapy using intracavitary (IC) or hybrid interstitial (IC/IS) applicators. For each patient, the primary radiation oncologist contoured the high-risk clinical target volume on a pre-brachytherapy MRI, indicated the approximate applicator location, and made a clinical determination of the first fraction applicator. A pre-trained ML model predicted the applicator and IC/IS needle arrangement using tumor geometry. Following the first fraction, ML and radiation oncologist predictions were compared and a replanning study determined the applicator providing optimal organ-at-risk (OAR) dosimetry. The ML-predicted applicator and needle arrangement and the clinical determination were compared to this dosimetric ground truth. RESULTS Ten patients were accrued from December 2020 to October 2022. Compared to the dosimetrically optimal applicator, both the radiation oncologist and ML had an accuracy of 70%. ML demonstrated better identification of patients requiring IC/IS applicators and provided balanced IC and IC/IS predictions. The needle selection model achieved an average accuracy of 82.5%. ML-predicted needle arrangements matched or improved plan quality when compared to clinically selected arrangements. Overall, ML predictions led to an average total improvement of 2.0 Gy to OAR doses over three treatment fractions when compared to clinical predictions. CONCLUSION In the context of a single institution study, the presented ML model demonstrates valuable decision-support for the applicator and needle selection process with the potential to provide improved dosimetry. Future work will include a multi-center study to assess generalizability.