English

Revisiting Mixout: An Overlooked Path to Robust Finetuning

Machine Learning 2025-10-09 v1 Computer Vision and Pattern Recognition

Abstract

Finetuning vision foundation models often improves in-domain accuracy but comes at the cost of robustness under distribution shift. We revisit Mixout, a stochastic regularizer that intermittently replaces finetuned weights with their pretrained reference, through the lens of a single-run, weight-sharing implicit ensemble. This perspective reveals three key levers that govern robustness: the \emph{masking anchor}, \emph{resampling frequency}, and \emph{mask sparsity}. Guided by this analysis, we introduce GMixout, which (i) replaces the fixed anchor with an exponential moving-average snapshot that adapts during training, and (ii) regulates masking period via an explicit resampling-frequency hyperparameter. Our sparse-kernel implementation updates only a small fraction of parameters with no inference-time overhead, enabling training on consumer-grade GPUs. Experiments on benchmarks covering covariate shift, corruption, and class imbalance, ImageNet / ImageNet-LT, DomainNet, iWildCam, and CIFAR100-C, GMixout consistently improves in-domain accuracy beyond zero-shot performance while surpassing both Model Soups and strong parameter-efficient finetuning baselines under distribution shift.

Cite

@article{arxiv.2510.06982,
  title  = {Revisiting Mixout: An Overlooked Path to Robust Finetuning},
  author = {Masih Aminbeidokhti and Heitor Rapela Medeiros and Eric Granger and Marco Pedersoli},
  journal= {arXiv preprint arXiv:2510.06982},
  year   = {2025}
}
R2 v1 2026-07-01T06:23:46.612Z