Automatic Classification of Uveal Melanoma Regression Patterns Following Ruthenium-106 Plaque Brachytherapy Using Ultrasound Images and Deep Convolutional Neural Network

Following uveal melanoma (UM) affected treatment using ruthenium-106 brachytherapy, tumor thickness patterns fall into one of four categories: decrease (regression), increase (recurrence), stop (stable), or other, which are assessed in follow-up A-mode and B-mode images. These patterns are critical indicators of the tumor’s response to therapy. This study aims to apply deep learning (DL) models for predicting post-brachytherapy tumor regression patterns. A cohort of 192 patients participated in this study. B-Mode images taken at the time of diagnosis were collected, and the ophthalmologists labeled the images into four regression patterns based on the results of the treatment. DenseNet121 and ResNet34 models were trained and evaluated using performance metrics. DenseNet121 achieved a macro-average AUC of 0.933, compared to 0.916 for the ResNet34. The per-class evaluation showed that DenseNet121 excelled in predicting all categories, providing superior predictive accuracy. The ablation study revealed that the best performance was achieved without pretrained weights, using dropout layers and a batch size of 32. Both models demonstrated strong classification capabilities, with DenseNet121 providing the highest overall accuracy. This study highlights the potential of DL models in predicting regression patterns in UM patients undergoing brachytherapy. Further validation and exploration of their integration into clinical practice are warranted.