A Comprehensive Deep Learning Framework for Automatic Target Volumes Segmentation in Post-Operative Stereotactic Partial Breast Irradiation (S-PBI)

Abstract

In S-PBI, accurate delineation of post-surgical tumor bed volume (TBV) and clinical target volume (CTV) are crucial tasks to achieve effective radiotherapy outcomes. However, manual contouring is labor intensive, time consuming, and largely relies on the experience of clinicians. We aimed to propose a deep learning (DL) approach which mimics physicians' contouring practice to accurately segment target volumes in post-operative breast CT images. Our approach incorporated domain knowledge into a 3D U-Net based DL model for breast target volumes (TBV and CTV) delineation. Our TBV segmentation approach was inspired by the marker-guidance procedure in manual delineation, where the visual clues provided by the markers assist physicians in defining TBV. For this purpose, a distance-transformation coupled with a Gaussian filter was adopted to convert markers' locations on the CT images to saliency maps. Subsequently, the CT images and the corresponding saliency maps formed a two-channel input for the segmentation model. For CTV segmentation, TBV was incorporated as an input in addition to the CT images, guiding the model to encode the location-related image features. The architecture allowed the network to emulate the oncologist's manual delineation where CTV is derived from TBV via a margin expansion, followed by correcting the extensions for anatomical barriers of tumor invasion (e.g., skin, chest wall). We retrospectively collected 175 prone CT images from 35 post-operative breast cancer patients who received 5-fraction partial breast irradiation (PBI) regimen on a Co-60 prone based S-PBI unit. The 35 patients were randomly split into 25, 5, and 5 for model training, validation, and testing respectively. We evaluated the performance of the developed DL model on the testing dataset by comparing the predicted volumes with the manually delineated contours (ground truth) using Dice similarity coefficient (DSC), 95th percentile Hausdorff distance (HD95), and average symmetric surface distance (ASD). For TBV segmentation, our model achieved mean (standard deviation) of 0.76 (±2.7), 6.76 (±1.83) mm, and 1.9 (±0.66) mm for DSC, HD95, and ASD respectively. For CTV segmentation, our model achieved 0.94 (±0.02), 2.46 (±0.5) mm, and 0.53 (±0.14) mm for DSC, HD95, and ASD respectively. The proposed auto-segmentation approach generated TBV and CTV masks in ∼11 seconds per CT volume, implying significantly improved efficiency compared to manual contouring. We developed a comprehensive DL framework mimicking clinical contouring practice for auto-segmentation of target volumes in S-PBI. The results demonstrated high levels of agreement between the predicted contours and physicians' manual contours. The approach is promising for improving the efficiency and accuracy of the on-line treatment planning workflow, such as adaptive based S-PBI.