DAWN: Domain-Adaptive Weakly Supervised Nuclei Segmentation via Cross-Task Interactions
Weakly supervised segmentation methods have garnered considerable attention due to their potential to alleviate the need for labor-intensive pixel-level annotations during model training. Traditional weakly supervised nuclei segmentation approaches typically involve a two-stage process: pseudo-label generation followed by network training. The performance of these methods is highly dependent on the quality of the generated pseudo-labels, which can limit their effectiveness. In this paper, we propose a novel domain-adaptive weakly supervised nuclei segmentation framework that addresses the challenge of pseudo-label generation through cross-task interaction strategies. Specifically, our approach leverages weakly annotated data to train an auxiliary detection task, which facilitates domain adaptation of the segmentation network. To improve the efficiency of domain adaptation, we introduce a consistent feature constraint module that integrates prior knowledge from the source domain. Additionally, we develop methods for pseudo-label optimization and interactive training to enhance domain transfer capabilities. We validate the effectiveness of our proposed method through extensive comparative and ablation experiments conducted on six datasets. The results demonstrate that our approach outperforms existing weakly supervised methods and achieves performance comparable to or exceeding that of fully supervised methods. Our code is available at <uri xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">https://github.com/zhangye-zoe/DAWN</uri>.
Authors
- Hao Bian (ORCID: https://orcid.org/0000-0002-3809-9574)
- Zijie Fang (ORCID: https://orcid.org/0000-0002-6366-4379)
- Ziyue Wang
- Linghan Cai (ORCID: https://orcid.org/0000-0002-7931-7697)
- Yifeng Wang (ORCID: https://orcid.org/0000-0001-8317-8434)
- Yongbing Zhang (ORCID: https://orcid.org/0000-0003-3320-2904)
- Ye Zhang (ORCID: https://orcid.org/0009-0003-7254-5026)
Institutions
- Harbin Institute of Technology (CN)
- Tsinghua–Berkeley Shenzhen Institute (CN)
- Tsinghua University (CN)
Publication Details
- Journal
- IEEE Transactions on Circuits and Systems for Video Technology
- Published
- 2024-12-13
- DOI
- https://doi.org/10.1109/tcsvt.2024.3515467
- Citations
- 1
- Primary Topic
- Nuclear Physics and Applications
- Type
- article
- Field-Weighted Citation Impact
- 0.61
Funders
- National Natural Science Foundation of China
- Fundamental Research Funds for the Central Universities
- Science and Technology Planning Project of Shenzhen Municipality