Kailyn Stenhouse1,2, Michael Roumeliotis1,2,3, Robyn Banerjee3,4, Svetlana Yanushkevich5, Philip McGeachy1,2,3. 1. Department of Physics and Astronomy, University of Calgary, Calgary, AB, Canada. 2. Department of Medical Physics, Tom Baker Cancer Centre, Calgary, AB, Canada. 3. Department of Oncology, University of Calgary, Calgary, AB, Canada. 4. Department of Radiation Oncology, Tom Baker Cancer Centre, Calgary, AB, Canada. 5. Department of Electrical and Computer Engineering, University of Calgary, Calgary, AB, Canada.
Abstract
Purpose: To develop and validate a preliminary machine learning (ML) model aiding in the selection of intracavitary (IC) versus hybrid interstitial (IS) applicators for high-dose-rate (HDR) cervical brachytherapy. Methods: From a dataset of 233 treatments using IC or IS applicators, a set of geometric features of the structure set were extracted, including the volumes of OARs (bladder, rectum, sigmoid colon) and HR-CTV, proximity of OARs to the HR-CTV, mean and maximum lateral and vertical HR-CTV extent, and offset of the HR-CTV centre-of-mass from the applicator tandem axis. Feature selection using an ANOVA F-test and mutual information removed uninformative features from this set. Twelve classification algorithms were trained and tested over 100 iterations to determine the highest performing individual models through nested 5-fold cross-validation. Three models with the highest accuracy were combined using soft voting to form the final model. This model was trained and tested over 1,000 iterations, during which the relative importance of each feature in the applicator selection process was determined. Results: Feature selection indicated that the mean and maximum lateral and vertical extent, volume, and axis offset of the HR-CTV were the most informative features and were thus provided to the ML models. Relative feature importances indicated that the HR-CTV volume and mean lateral extent were most important for applicator selection. From the comparison of the individual classification algorithms, it was found that the highest performing algorithms were tree-based ensemble methods - AdaBoost Classifier (ABC), Gradient Boosting Classifier (GBC), and Random Forest Classifier (RFC). The accuracy of the individual models was compared to the voting model for 100 iterations (ABC = 91.6 ± 3.1%, GBC = 90.4 ± 4.1%, RFC = 89.5 ± 4.0%, Voting Model = 92.2 ± 1.8%) and the voting model was found to have superior accuracy. Over the final 1,000 evaluation iterations, the final voting model demonstrated a high predictive accuracy (91.5 ± 0.9%) and F1 Score (90.6 ± 1.1%). Conclusion: The presented model demonstrates high discriminative performance, highlighting the potential for utilization in informing applicator selection prospectively following further clinical validation.
Purpose: To develop and validate a preliminary machine learning (ML) model aiding in the selection of intracavitary (IC) versus hybrid interstitial (IS) applicators for high-dose-rate (HDR) cervical brachytherapy. Methods: From a dataset of 233 treatments using IC or IS applicators, a set of geometric features of the structure set were extracted, including the volumes of OARs (bladder, rectum, sigmoid colon) and HR-CTV, proximity of OARs to the HR-CTV, mean and maximum lateral and vertical HR-CTV extent, and offset of the HR-CTV centre-of-mass from the applicator tandem axis. Feature selection using an ANOVA F-test and mutual information removed uninformative features from this set. Twelve classification algorithms were trained and tested over 100 iterations to determine the highest performing individual models through nested 5-fold cross-validation. Three models with the highest accuracy were combined using soft voting to form the final model. This model was trained and tested over 1,000 iterations, during which the relative importance of each feature in the applicator selection process was determined. Results: Feature selection indicated that the mean and maximum lateral and vertical extent, volume, and axis offset of the HR-CTV were the most informative features and were thus provided to the ML models. Relative feature importances indicated that the HR-CTV volume and mean lateral extent were most important for applicator selection. From the comparison of the individual classification algorithms, it was found that the highest performing algorithms were tree-based ensemble methods - AdaBoost Classifier (ABC), Gradient Boosting Classifier (GBC), and Random Forest Classifier (RFC). The accuracy of the individual models was compared to the voting model for 100 iterations (ABC = 91.6 ± 3.1%, GBC = 90.4 ± 4.1%, RFC = 89.5 ± 4.0%, Voting Model = 92.2 ± 1.8%) and the voting model was found to have superior accuracy. Over the final 1,000 evaluation iterations, the final voting model demonstrated a high predictive accuracy (91.5 ± 0.9%) and F1 Score (90.6 ± 1.1%). Conclusion: The presented model demonstrates high discriminative performance, highlighting the potential for utilization in informing applicator selection prospectively following further clinical validation.
Authors: Akila N Viswanathan; Carien L Creutzberg; Peter Craighead; Mary McCormack; Takafumi Toita; Kailash Narayan; Nicholas Reed; Harry Long; Hak-Jae Kim; Christian Marth; Jacob C Lindegaard; Annmarie Cerrotta; William Small; Edward Trimble Journal: Int J Radiat Oncol Biol Phys Date: 2010-12-22 Impact factor: 7.038
Authors: Richard Pötter; Christine Haie-Meder; Erik Van Limbergen; Isabelle Barillot; Marisol De Brabandere; Johannes Dimopoulos; Isabelle Dumas; Beth Erickson; Stefan Lang; An Nulens; Peter Petrow; Jason Rownd; Christian Kirisits Journal: Radiother Oncol Date: 2006-01-05 Impact factor: 6.280
Authors: Christian Kirisits; Stefan Lang; Johannes Dimopoulos; Daniel Berger; Dietmar Georg; Richard Pötter Journal: Int J Radiat Oncol Biol Phys Date: 2006-06-01 Impact factor: 7.038
Authors: Christel N Nomden; Astrid A C de Leeuw; Marinus A Moerland; Judith M Roesink; Robbert J H A Tersteeg; Ina Maria Jürgenliemk-Schulz Journal: Int J Radiat Oncol Biol Phys Date: 2011-06-12 Impact factor: 7.038
Authors: Akila N Viswanathan; Sushil Beriwal; Jennifer F De Los Santos; D Jeffrey Demanes; David Gaffney; Jorgen Hansen; Ellen Jones; Christian Kirisits; Bruce Thomadsen; Beth Erickson Journal: Brachytherapy Date: 2012 Jan-Feb Impact factor: 2.362
Authors: Beth Erickson; Patricia Eifel; Jennifer Moughan; Jason Rownd; Thomas Iarocci; Jean Owen Journal: Int J Radiat Oncol Biol Phys Date: 2005-08-15 Impact factor: 7.038