Improved brain metastases segmentation using generative adversarial network and conditional random field optimization mask RвҖҗCNN
Yiren Wang,Zhongjian Wen,Lei Su,Hairui Deng,Jiali Gong,Hongli Xiang,Yongcheng He,Huaiwen Zhang,Ping Zhou,Haowen Pang
DOI: https://doi.org/10.1002/mp.17176
IF: 4.506
2024-05-24
Medical Physics
Abstract:Background In radiotherapy, the delineation of the gross tumor volume (GTV) in brain metastases using computed tomography (CT) simulation localization is very important. However, despite the criticality of this process, a pronounced gap exists in the availability of tools tailored for the automatic segmentation of the GTV based on CT simulation localization images. Purpose This study aims to fill this gap by devising an effective tool specifically for the automatic segmentation of the GTV using CT simulation localization images. Methods A dualвҖҗnetwork generative adversarial network (GAN) architecture was developed, wherein the generator focused on refining CT images for more precise delineation, and the discriminator differentiated between real and augmented images. This architecture was coupled with the Mask RвҖҗCNN model to achieve meticulous GTV segmentation. An endвҖҗtoвҖҗend training process facilitated the integration between the GAN and Mask RвҖҗCNN functionalities. Furthermore, a conditional random field (CRF) was incorporated to refine the initial masks generated by the Mask RвҖҗCNN model to ensure optimal segmentation accuracy. The performance was assessed using key metrics, namely, the Dice coefficient (DSC), intersection over union (IoU), accuracy, specificity, and sensitivity. Results The GAN+Mask RвҖҗCNN+CRF integration method in this study performs well in GTV segmentation. In particular, the model has an overall average DSC of 0.819В ВұВ 0.102 and an IoU of 0.712В ВұВ 0.111 in the internal validation. The overall average DSC in the external validation data is 0.726В ВұВ 0.128 and the IoU is 0.640В ВұВ 0.136. It demonstrates favorable generalization ability. Conclusion The integration of the GAN, Mask RвҖҗCNN, and CRF optimization provides a pioneering tool for the sophisticated segmentation of the GTV in brain metastases using CT simulation localization images. The method proposed in this study can provide a robust automatic segmentation approach for brain metastases in the absence of MRI.
radiology, nuclear medicine & medical imaging