用扩散模型+对比学习,让乳腺超声诊断在少标注下也能跨数据集保持高精度。
ADAptation: Reconstruction-based Unsupervised Active Learning for Breast Ultrasound Diagnosis
- 先用扩散模型把目标域图像转成源域风格,缩小数据分布差异。
- 通过双评分机制选出最不确定又最具代表性的样本,提升标注效率。
- 在4个乳腺超声数据集上验证,比现有方法更稳定、适应性更强。
基于深度学习的诊断模型常因训练(源)与测试(目标)数据分布差异导致性能下降。尽管重新收集并标注目标域数据是理想方案,但受限于时间和资源。主动学习(AL)可降低标注成本,却难以应对不同数据集间的分布变化。本文提出一种新的无监督域适应主动学习框架ADAptation,能在有限标注预算下高效选择多源数据池中的信息量丰富的样本。首先,利用扩散模型的分布同质化能力,将目标域图像转换为源域风格以弥合跨数据集差异。随后引入两项创新:(a) 超球面约束的对比学习网络,实现特征紧凑聚类;(b) 双评分机制,量化并平衡样本的不确定性与代表性。在四个乳腺超声数据集(三个公开、一个自建/多中心)上,针对五种常见深度分类器的大量实验表明,本方法显著优于现有强基线主动学习方法,验证了其在临床域适应中的有效性与泛化能力。代码已开源:https://github.com/miccai25-966/ADAptation。
原文摘要 · Abstract (English)
Deep learning-based diagnostic models often suffer performance drops due to distribution shifts between training (source) and test (target) domains. Collecting and labeling sufficient target domain data for model retraining represents an optimal solution, yet is limited by time and scarce resources. Active learning (AL) offers an efficient approach to reduce annotation costs while maintaining performance, but struggles to handle the challenge posed by distribution variations across different datasets. In this study, we propose a novel unsupervised Active learning framework for Domain Adaptation, named ADAptation, which efficiently selects informative samples from multi-domain data pools under limited annotation budget. As a fundamental step, our method first utilizes the distribution homogenization capabilities of diffusion models to bridge cross-dataset gaps by translating target images into source-domain style. We then introduce two key innovations: (a) a hypersphere-constrained contrastive learning network for compact feature clustering, and (b) a dual-scoring mechanism that quantifies and balances sample uncertainty and representativeness. Extensive experiments on four breast ultrasound datasets (three public and one in-house/multi-center) across five common deep classifiers demonstrate that our method surpasses existing strong AL-based competitors, validating its effectiveness and generalization for clinical domain adaptation. The code is available at the anonymized link: https://github.com/miccai25-966/ADAptation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。