提出无需采样的能量模型,实现高效可靠测试时自适应。
Contrastive Residual Energy Test-time Adaptation
- 通过对比损失重构能量函数,避免采样与近似误差。
- 在多个数据集上达到高精度且预测置信度更可信。
- 适合对推理速度与可靠性要求高的实际部署场景。
测试时自适应(TTA)通过调整模型以应对与训练分布不同的目标分布,提升实际应用中的泛化能力。然而,现有方法多关注条件分布调整,依赖无标签预测,导致校准性差。基于能量的TTA框架通过建模目标数据的边缘分布提供替代方案,但其依赖昂贵采样,难以满足低延迟需求。本文提出对比残差能量测试时自适应(CreTTA),理论上将边缘分布适应重构为学习残差能量函数。该形式导出一个对比目标,使不可计算的归一化常数在数学上抵消,彻底消除采样与近似误差。关键分析表明,该设计通过利用相对能量差异实现自适应梯度重加权,防止过拟合并规避熵最小化带来的自证实偏差。大量实验验证,CreTTA在真实计算约束下实现了可扩展且校准良好的适应性能。
原文摘要 · Abstract (English)
Test-time adaptation (TTA) enhances model robustness by enabling adaptation to target distributions that differ from training distributions, improving real-world generalizability. However, most existing TTA approaches focus on adjusting the conditional distribution and therefore exhibit poor calibration, as they rely on uncertain predictions in the absence of labels. Energy-based TTA frameworks provide an alternative by modeling the marginal distribution of target data without depending on label predictions, but their reliance on costly sampling hinders scalability in real-world scenarios where decisions must be made without latency. In this work, we propose Contrastive Residual Energy Test-time Adaptation (CreTTA), a practical solution for reliable adaptation. We theoretically reformulate the marginal distribution adaptation as learning a residual energy function. This formulation leads to a contrastive objective where the intractable partition function mathematically cancels out, removing sampling and approximation error.Crucially, our analysis reveals that this design prevents overfitting through an adaptive gradient reweighting mechanism that leverages relative energy differences, avoiding the self-confirming bias of entropy minimization. Extensive experiments demonstrate that CreTTA achieves scalable and well-calibrated adaptation under real-world computational constraints.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。