高率混入法让预训练模型更抗分布偏移,且省下大量计算资源。
High-Rate Mixout: Revisiting Mixout for Robust Domain Generalization
- 用高概率随机替换权重,保持预训练知识的同时适应新域。
- 在多个数据集上达到与集成方法相当的泛化性能。
- 大幅降低梯度计算和内存占用,最高节省90%资源。
微调多个从强大预训练权重初始化的模型是提升分布偏移下鲁棒性的常见策略,但需训练并存储多个模型,带来巨大计算开销。Dropout虽轻量,但在预训练模型上易过度正则化,破坏关键表征。本文研究了混入(Mixout)这一随机正则化技术,用于领域泛化。与丢弃神经元不同,混入通过概率性地将部分微调权重替换为预训练权重来缓解过拟合,平衡适应与知识保留。研究发现,在领域泛化基准上取得强性能需设定较高掩码率:ViT为0.9,ResNet为0.8。这虽看似简单调整,却带来两大优势:(1) 高掩码率更强地惩罚对预训练参数的偏离,促进未见域泛化;(2) 高率掩码显著降低计算开销,梯度计算最多减少45%,梯度内存使用最多减少90%。在五个领域泛化基准(PACS、VLCS、OfficeHome、TerraIncognita、DomainNet)上,使用ResNet和ViT架构的实验表明,高率混入法在保持与集成方法相当的域外准确率的同时,大幅降低训练成本。
原文摘要 · Abstract (English)
Ensembling fine-tuned models initialized from powerful pre-trained weights is a common strategy to improve robustness under distribution shifts, but it comes with substantial computational costs due to the need to train and store multiple models. Dropout offers a lightweight alternative by simulating ensembles through random neuron deactivation; however, when applied to pre-trained models, it tends to over-regularize and disrupt critical representations necessary for generalization. In this work, we investigate Mixout, a stochastic regularization technique that provides an alternative to Dropout for domain generalization. Rather than deactivating neurons, Mixout mitigates overfitting by probabilistically swapping a subset of fine-tuned weights with their pre-trained counterparts during training, thereby maintaining a balance between adaptation and retention of prior knowledge. Our study reveals that achieving strong performance with Mixout on domain generalization benchmarks requires a notably high masking probability of 0.9 for ViTs and 0.8 for ResNets. While this may seem like a simple adjustment, it yields two key advantages for domain generalization: (1) higher masking rates more strongly penalize deviations from the pre-trained parameters, promoting better generalization to unseen domains; and (2) high-rate masking substantially reduces computational overhead, cutting gradient computation by up to 45% and gradient memory usage by up to 90%. Experiments across five domain generalization benchmarks, PACS, VLCS, OfficeHome, TerraIncognita, and DomainNet, using ResNet and ViT architectures, show that our approach, High-rate Mixout, achieves out-of-domain accuracy comparable to ensemble-based methods while significantly reducing training costs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。