通过锚点引导生成稳定伪标签,提升图像分割在域偏移下的自适应性能。
A3-TTA: Adaptive Anchor Alignment Test-Time Adaptation for Image Segmentation
- 用类别紧凑度识别可信目标图像作为锚点,指导伪标签生成。
- 在医学和自然图像上相比源模型提升10.40~17.68个百分点的Dice分数。
- 适合需要持续适应多个新领域的场景,抗遗忘能力强。
测试时自适应(TTA)为在域偏移下部署图像分割模型提供了一种无需访问源数据或重新训练的实用方案。现有基于伪标签的TTA方法常依赖扰动集成启发式策略(如丢弃采样、测试时增强、高斯噪声),缺乏分布基础,导致训练信号不稳定,易引发误差累积与灾难性遗忘。为此,本文提出A3-TTA框架,通过锚点引导监督构建可靠伪标签。具体而言,基于类别紧凑密度度量识别可信目标域图像作为锚点,假设高置信预测意味着分布接近源域。这些锚点作为稳定参考,指导伪标签生成,并通过语义一致性与边界感知熵最小化进一步正则化。此外,引入自适应指数移动平均策略以缓解标签噪声并稳定模型更新。在多域医学图像(心脏结构与前列腺分割)及自然图像上评估,A3-TTA相较源模型平均Dice分数提升10.40至17.68个百分点,优于多种先进TTA方法,且在连续TTA中表现优异,具备强抗遗忘能力。代码将公开于https://github.com/HiLab-git/A3-TTA。
原文摘要 · Abstract (English)
Test-Time Adaptation (TTA) offers a practical solution for deploying image segmentation models under domain shift without accessing source data or retraining. Among existing TTA strategies, pseudo-label-based methods have shown promising performance. However, they often rely on perturbation-ensemble heuristics (e.g., dropout sampling, test-time augmentation, Gaussian noise), which lack distributional grounding and yield unstable training signals. This can trigger error accumulation and catastrophic forgetting during adaptation. To address this, we propose \textbf{A3-TTA}, a TTA framework that constructs reliable pseudo-labels through anchor-guided supervision. Specifically, we identify well-predicted target domain images using a class compact density metric, under the assumption that confident predictions imply distributional proximity to the source domain. These anchors serve as stable references to guide pseudo-label generation, which is further regularized via semantic consistency and boundary-aware entropy minimization. Additionally, we introduce a self-adaptive exponential moving average strategy to mitigate label noise and stabilize model update during adaptation. Evaluated on both multi-domain medical images (heart structure and prostate segmentation) and natural images, A3-TTA significantly improves average Dice scores by 10.40 to 17.68 percentage points compared to the source model, outperforming several state-of-the-art TTA methods under different segmentation model architectures. A3-TTA also excels in continual TTA, maintaining high performance across sequential target domains with strong anti-forgetting ability. The code will be made publicly available at https://github.com/HiLab-git/A3-TTA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。