Deep Learning for Automated Contouring of Primary Gross Tumor Volumes by MRI for Radiation Therapy of Brain Metastasis

Abstract

Radiotherapy is one of the most effective methods for the treatment of brain metastases (BMs). Traditional manual delineation of primary gross tumor volumes (GTV) of multiple BMs (especially small metastases) in radiotherapy practice is extremely labor intensive and highly dependent on oncologists' experience, achieving the precise and efficient automatic delineation of BMs is of great significance for efficient and homogeneous one-stop adaptive radiotherapy. We retrospectively collected 62 MRI (non-enhanced T1-weighted sequences) sequences of 50 patients with BMs from January 2020 to July 2021. An automatic model (BUC-Net) for automatic delineation BMs was proposed in this work, which was based on deep learning by combining 3D bottler layer module and the cascade architecture to improve the accuracy and efficient of BMs' automatic delineation, especially for small metastases with tiny size and relatively low contrast. The prosed method was compared with the existing 3D U-Net (U-Net) and 3D U-Net Cascade (U-Net Cascade). The performance of our proposed method was evaluated by Dice similarity coefficient (DSC), 95% Hausdorff distance (HD95) and average surface distance (ASD) with human experts. The automatic segmentation results of BUC-Net evaluated with 310 BMs in 13 test patients was summarized in Table 1. These BMs in each test patient were automatically delineated by two types of contours: as a whole tumor contour (Whole-delineation) and the multiple tumor contours (Multiple-delineation). BUC-Net performed the best mean DSC and HD95, which is significantly outperformed U-Net (Whole-delineation: 0.911 & 0.894 of DSC, Multiple-delineation: 0.794 & 0.754 of DSC, P < 0.05 for both) and U-Net cascade (Whole-delineation: 0.947 & 7.141 of HD95, Multiple-delineation: 0.902 & 1.171 of HD95, P < 0.05 for both); Additionally, BUC-Net achieved the best mean ASD for Whole-delineation and comparable ASD (0.290 & 0.277, P > 0) for Multiple-delineation with U-Net Cascade. Our results showed that the proposed approach is promising for the automatic delineation of BMs in MRI, which can be integrated into a radiotherapy workflow to significantly shorten segmentation time.