arXiv:2508.14079cs.LG2025-08被引 2

对比40种模型架构,发现监督预训练的卷积网络在鲁棒性上表现最佳。

A Guide to Robust Generalization: The Impact of Architecture, Pre-training, and Optimization Strategy

  • 系统测试6个数据集上1440种配置,评估不同架构与优化策略
  • 卷积网络在大模型+监督预训练下,对5类扰动鲁棒性最强
  • 为模型选择提供实证依据,适合想提升模型鲁棒性的研究者

深度学习图像模型对微小输入扰动敏感。过去常通过随机初始化训练并使用特殊损失函数来提升鲁棒性。近年,鲁棒微调成为更高效方案:在预训练模型基础上调整以兼顾性能与鲁棒性。设计包括模型更新方式(全量或部分)、损失函数、架构类型与大小、预训练表示等。这些因素影响模型在未见扰动下的泛化能力。本文开展大规模实证研究,覆盖6个数据集、40个预训练架构、2种专用损失和3种适配协议,共生成1,440种训练配置与7,200次鲁棒性测量,涵盖5类扰动。这是目前最全面的鲁棒微调基准。尽管基于注意力的架构和鲁棒预训练表示日益流行,我们发现监督预训练的卷积神经网络通常表现最优。研究验证并挑战了既有设计假设,揭示潜在方向并提供实用指导。

原文摘要 · Abstract (English)

Deep learning models operating in the image domain are vulnerable to small input perturbations. For years, robustness to such perturbations was pursued by training models from scratch (i.e., with random initializations) using specialized loss objectives. Recently, robust fine-tuning has emerged as a more efficient alternative: instead of training from scratch, pretrained models are adapted to maximize predictive performance and robustness. To conduct robust fine-tuning, practitioners design an optimization strategy that includes the model update protocol (e.g., full or partial) and the specialized loss objective. Additional design choices include the architecture type and size, and the pretrained representation. These design choices affect robust generalization, which is the model's ability to maintain performance when exposed to new and unseen perturbations at test time. Understanding how these design choices influence generalization remains an open question with significant practical implications. In response, we present an empirical study spanning 6 datasets, 40 pretrained architectures, 2 specialized losses, and 3 adaptation protocols, yielding 1,440 training configurations and 7,200 robustness measurements across five perturbation types. To our knowledge, this is the most diverse and comprehensive benchmark of robust fine-tuning to date. While attention-based architectures and robust pretrained representations are increasingly popular, we find that convolutional neural networks pretrained in a supervised manner on large datasets often perform best. Our analysis both confirms and challenges prior design assumptions, highlighting promising research directions and offering practical guidance.

模型鲁棒性微调策略卷积网络预训练

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。