Reliable segmentation of multiple lesions from medical images

Meng Wang,Kai Yu,Zhiwei Tan,Ke Zou,Rick Siow Mong Goh,Huazhu Fu
DOI: https://doi.org/10.1002/mp.17244
IF: 4.506
2024-06-12
Medical Physics
Abstract:Background Focusing on the complicated pathological features, such as blurred boundaries, severe scale differences between symptoms, and background noise interference, we aim to enhance the reliability of multiple lesions joint segmentation from medical images. Purpose Propose a novel reliable multi‐scale wavelet‐enhanced transformer network, which can provide accurate segmentation results with reliability assessment. Methods Focusing on enhancing the model's capability to capture intricate pathological features in medical images, this work introduces a novel segmentation backbone. The backbone integrates a wavelet‐enhanced feature extractor network and incorporates a multi‐scale transformer module developed within the scope of this work. Simultaneously, to enhance the reliability of segmentation outcomes, a novel uncertainty segmentation head is proposed. This segmentation head is rooted in the SL, contributing to the generation of final segmentation results along with an associated overall uncertainty evaluation score map. Results Comprehensive experiments are conducted on the public database of AI‐Challenge 2018 for retinal edema lesions segmentation and the segmentation of Thoracic Organs at Risk in CT images. The experimental results highlight the superior segmentation accuracy and heightened reliability achieved by the proposed method in comparison to other state‐of‐the‐art segmentation approaches. Conclusions Unlike previous segmentation methods, the proposed approach can produce reliable segmentation results with an estimated uncertainty and higher accuracy, enhancing the overall reliability of the model. The code will be release on https://github.com/LooKing9218/ReMultiSeg.
radiology, nuclear medicine & medical imaging
What problem does this paper attempt to address?