首页|Structure-aware single-source generalization with pixel-level disentanglement for joint optic disc and cup segmentation

Structure-aware single-source generalization with pixel-level disentanglement for joint optic disc and cup segmentation

扫码查看
原文链接
  • NETL
  • NSTL
  • Elsevier
© 2024 Elsevier LtdDeploying deep segmentation models in new medical centers poses a significant challenge due to statistical disparities between source and unknown domains. Recent advancements in domain generalization (DG) have shown improved generalization performance by leveraging disentanglement techniques on domain-specific and domain-invariant features. However, existing DG methods face challenges in achieving optimal feature segregation. To address this, we introduce a pixel-level contrastive single domain generalization (PCSDG) framework and a structure-aware brightness augmentation (SABA) technique for joint optic disc and cup segmentation. First, a disentanglement module captures content and style-related maps, pixel-wise multiplied with the original image to generate saliency-based attention maps, resulting in distinct structure and style representations. Second, using a contrastive loss in the latent space enhances segregation. Finally, SABA introduces random brightness variations, preserving anatomical information and diversifying sample styles. Experimental validation on two public fundus image datasets with two source domains and five target domains demonstrates the superior performance of PCSDG and SABA across diverse domains when compared to state-of-the-art methods. Our code and models are made public at: https://github.com/HopkinsKwong/PCSDG.

Contrastive learningDisentanglementMedical image segmentationSingle domain generalization

Jiang J.-X.、Li Y.、Wang Z.

展开 >

School of Information Science and Engineering Lanzhou University||Key Laboratory of AI and Information Processing Education Department of Guangxi Zhuang Autonomous Region Hechi University

School of Information Science and Engineering Lanzhou University

School of Information Science and Engineering Lanzhou University||School of Computer Science and Engineering Lanzhou UniversitySchool of Computer Science and Engineering Lanzhou University||Key Laboratory of AI and Information Processing Education Department of Guangxi Zhuang Autonomous Region Hechi University||

2025

10.1016/j.bspc.2024.106801

Biomedical signal processing and control

Biomedical signal processing and control

SCI
ISSN：1746-8094
年,卷(期)：2025.99(Jan.)
  • 54