Support Vector Machine Model Predicts Dose for Organs at Risk in High-Dose Rate Brachytherapy of Cervical Cancer.

INTRODUCTION

This study aimed to establish a support vector machine (SVM) model to predict the dose for organs at risk (OARs) in intracavitary brachytherapy planning for cervical cancer with tandem and ovoid treatments.

METHODS

Fifty patients with loco-regionally advanced cervical cancer treated with 200 CT-based tandem and ovoid brachytherapy plans were included. The brachytherapy plans were randomly divided into the training (N = 160) and verification groups (N = 40). The bladder, rectum, sigmoid colon, and small intestine were divided into sub-OARs. The SVM model was established using MATLAB software based on the sub-OAR volume to predict the bladder, rectum, sigmoid colon, and small intestine D 2 c m 3 . Model performance was quantified by mean squared error (MSE) and δ ( δ = | D 2 c m 3 / D prescription ( actual ) - D 2 c m 3 / D prescription ( predicted ) | ) . The goodness of fit of the model was quantified by the coefficient of determination (R2). The accuracy and validity of the SVM model were verified using the validation group.

RESULTS

The D 2 c m 3 value of the bladder, rectum, sigmoid colon, and small intestine correlated with the volume of the corresponding sub-OARs in the training group. The mean squared error (MSE) in the SVM model training group was <0.05; the R2 of each OAR was >0.9. There was no significant difference between the D 2 c m 3 -predicted and actual values in the validation group (all P > 0.05): bladder δ = 0.024 ± 0.022, rectum δ = 0.026 ± 0.014, sigmoid colon δ = 0.035 ± 0.023, and small intestine δ = 0.032 ± 0.025.

CONCLUSION

The SVM model established in this study can effectively predict the D 2 c m 3 for the bladder, rectum, sigmoid colon, and small intestine in cervical cancer brachytherapy.