Despite that deep learning models have achieved remarkable performance in medical image segmentation, their performance is often limited on testing images from new centers with a domain shift. To achieve Domain Generalization (DG) for medical image segmentation, we propose a Domain Composition and Attention-based Network (DCA-Net) combined with structure- and style-based data augmentation that generates unlabeled synthetic images for training. First, DCA-Net represents features in one certain domain by a linear combination of a set of basis representations that are learned by parallel domain preceptors with a divergence constraint. The linear combination is used to calibrate the feature maps of an input image, which enables the model to generalize to unseen domains. Second, considering the number of domains and images for training is limited, we employ generative models to synthesize images with a higher structure diversity, and to leverage the unlabeled synthetic images, we introduce a consistency constraint for their predictions under style augmentation based on frequency amplitude mixture. Additionally, a Test-Time Frequency Augmentation (TTFA) is proposed to neutralize the domain shift from the target to source domains. Experimental results on two multi-domain datasets for fundus structure and nasopharyngeal carcinoma segmentation showed that: (1) our method significantly outperformed several existing DG methods, and (2) the model's generalizability was largely improved by domain composition and attention modules; (3) by leveraging the unlabeled synthetic images and the TTFA, the model could better deal with images from unseen domains.
基金:
National Key Research&Development Program of China [2022ZD0160705]; National Natural Science Foun-dation of China [62271115]; Science and Technology Department of Sichuan Province, China [2022YFSY0055]; Radiation Oncology Key Laboratory of Sichuan Province Open Fund [2022ROKF04]; Sichuan Province International Science, Technology and Innovation Coopera-tion Foundation [2022YFH0004]
第一作者机构:[1]Univ Elect Sci & Technol China, Sch Mech & Elect Engn, Chengdu, Peoples R China
共同第一作者:
通讯作者:
通讯机构:[1]Univ Elect Sci & Technol China, Sch Mech & Elect Engn, Chengdu, Peoples R China[3]Shanghai Artificial Intelligence Lab, Shanghai, Peoples R China
推荐引用方式(GB/T 7714):
Lu Jiangshan,Gu Ran,Liao Wenjun,et al.Domain composition and attention network trained with synthesized unlabeled images for generalizable medical image segmentation[J].NEUROCOMPUTING.2024,599:doi:10.1016/j.neucom.2024.128115.
APA:
Lu, Jiangshan,Gu, Ran,Liao, Wenjun,Zhang, Shichuan,Yu, Huijun...&Wang, Guotai.(2024).Domain composition and attention network trained with synthesized unlabeled images for generalizable medical image segmentation.NEUROCOMPUTING,599,
MLA:
Lu, Jiangshan,et al."Domain composition and attention network trained with synthesized unlabeled images for generalizable medical image segmentation".NEUROCOMPUTING 599.(2024)