A SwinTransformer-Based Segmentation Framework With Self-Supervised Strategy for Post-Operative Prostate Cancer Radiotherapy

IEEE J Biomed Health Inform. 2023 Nov 1:PP. doi: 10.1109/JBHI.2023.3329111. Online ahead of print.

Abstract

Radical prostatectomy (prostate removal) is a standard treatment for clinically localized prostate cancer and is often followed by postoperative radiotherapy. Postoperative radiotherapy requires accurate delineation of the clinical target volume (CTV) and lymph node drainage area (LNA) on computed tomography (CT) images. However, the CTV contour cannot be determined by the simple prostate expansion after resection of the prostate in the CT image. Constrained by this factor, the manual delineation process in postoperative radiotherapy is more time-consuming and challenging than in radical radiotherapy. In addition, CTV and LNA have no boundaries that can be distinguished by pixel values in CT images, and existing automatic segmentation models cannot get satisfactory results. Radiation oncologists generally determine CTV and LNA profiles according to clinical consensus and guidelines regarding surrounding organs at risk (OARs). In this work, we design a cascade segmentation block to explicitly establish correlations between CTV, LNA, and OARs, leveraging OARs features to guide CTV and LNA segmentation. Furthermore, inspired by the success of the self-attention mechanism and self-supervised learning, we adopt SwinTransformer as our backbone and propose a pure SwinTransformer-based segmentation network with self-supervised learning strategies. We performed extensive quantitative and qualitative evaluations of the proposed method. Compared to other competitive segmentation models, our model shows higher dice scores with minor standard deviations, and the detailed visualization results are more consistent with the ground truth. We believe this work can provide a feasible solution to this problem, making the postoperative radiotherapy process more efficient.