让不鲁棒的预训练模型在测试时自动提升抗干扰能力
Learning Robustness at Test-Time from a Non-Robust Teacher
- 用非鲁棒教师模型的预测做语义锚点,指导测试时自适应
- 在CIFAR-10和ImageNet上实现更稳定优化与更好鲁棒性-精度平衡
- 适合部署后无标签数据场景,对超参数不敏感
当前预训练模型常被用作通用骨干,在目标域测试时进行无监督自适应,以应对目标数据稀缺的问题。尽管该范式有效提升了干净准确率,但对抗鲁棒性仍鲜受关注,尤其当原始模型本身不具备鲁棒性时。本文探讨:能否在测试时通过适配使一个非鲁棒的预训练模型提升目标分布上的对抗鲁棒性?研究分析了经典对抗训练策略在无监督测试时适配场景下的表现,发现基于知识蒸馏的简单方法不稳定且对超参数敏感。为此,提出一种无需标签的框架,利用非鲁棒教师模型的预测作为清洁样本与对抗样本的目标语义锚点。理论分析表明,该方法比传统自一致性正则化更稳定。在CIFAR-10和ImageNet上的实验验证了其优势:优化更稳定、对参数选择更鲁棒,并在鲁棒性-精度权衡上优于现有基线。
原文摘要 · Abstract (English)
Nowadays, pretrained models are increasingly used as general-purpose backbones and adapted at test-time to downstream environments where target data are scarce and unlabeled. While this paradigm has proven effective for improving clean accuracy on the target domain, adversarial robustness has received far less attention, especially when the original pretrained model is not explicitly designed to be robust. This raises a practical question: \emph{can a pretrained, non-robust model be adapted at test-time to improve adversarial robustness on a target distribution?} To face this question, this work studies how adversarial training strategies behave when integrated into adaptation schemes for the unsupervised test-time setting, where only a small set of unlabeled target samples is available. It first analyzes how classical adversarial training formulations can be extended to this scenario, showing that straightforward distillation-based adaptations remain unstable and highly sensitive to hyperparameter tuning, particularly when the teacher itself is non-robust. To address these limitations, the work proposes a label-free framework that uses the predictions of a non-robust teacher model as a semantic anchor for both the clean and adversarial objectives during adaptation. We further provide theoretical insights showing that our formulation yields a more stable alternative to the self-consistency-based regularization commonly used in classical adversarial training. Experiments evaluate the proposed approach on CIFAR-10 and ImageNet under induced photometric transformations. The results support the theoretical insights by showing that the proposed approach achieves improved optimization stability, lower sensitivity to parameter choices, and a better robustness-accuracy trade-off than existing baselines in this post-deployment test-time setting.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。