English

Elastic ViTs from Pretrained Models without Retraining

Computer Vision and Pattern Recognition 2025-10-21 v1

Abstract

Vision foundation models achieve remarkable performance but are only available in a limited set of pre-determined sizes, forcing sub-optimal deployment choices under real-world constraints. We introduce SnapViT: Single-shot network approximation for pruned Vision Transformers, a new post-pretraining structured pruning method that enables elastic inference across a continuum of compute budgets. Our approach efficiently combines gradient information with cross-network structure correlations, approximated via an evolutionary algorithm, does not require labeled data, generalizes to models without a classification head, and is retraining-free. Experiments on DINO, SigLIPv2, DeIT, and AugReg models demonstrate superior performance over state-of-the-art methods across various sparsities, requiring less than five minutes on a single A100 GPU to generate elastic models that can be adjusted to any computational budget. Our key contributions include an efficient pruning strategy for pretrained Vision Transformers, a novel evolutionary approximation of Hessian off-diagonal structures, and a self-supervised importance scoring mechanism that maintains strong performance without requiring retraining or labels. Code and pruned models are available at: https://elastic.ashita.nl/

Keywords

Cite

@article{arxiv.2510.17700,
  title  = {Elastic ViTs from Pretrained Models without Retraining},
  author = {Walter Simoncini and Michael Dorkenwald and Tijmen Blankevoort and Cees G. M. Snoek and Yuki M. Asano},
  journal= {arXiv preprint arXiv:2510.17700},
  year   = {2025}
}

Comments

Accepted at NeurIPS 2025

R2 v1 2026-07-01T06:47:57.382Z