Retraining-Free Depth Pruning of Vision Transformers: A Cross Domain Study of Layer Redundancy
cs.CV
Submitted: 2026-10-10
Updated: 2026-10-10
Code: https://github.com/zincalex/ViT-Drop
License: http://creativecommons.org/licenses/by/4.0/
The gist: Transformer architectures have achieved strong results in natural language processing and computer vision, but their deployment is limited by high computational, memory and storage costs, motivating
Terminology
Abstract
Transformer architectures have achieved strong results in natural language processing and computer vision, but their deployment is limited by high computational, memory and storage costs, motivating research into model compression techniques, such as pruning, while preserving performance. Prior work has shown substantial redundancy in large language models (LLMs), and redundancy in vision transformers has also been investigated, but it remains less systematically analyzed than in LLMs. In this work, we extend similarity-based redundancy analysis to vision models by measuring cosine similarity between layer inputs and outputs to identify transformations with minimal impact. We study redundancy patterns across several transformer architectures and find that deep vision transformers such as DINOv2 and SwinV2, like language models, exhibit significant redundancy. We then evaluate how this redundancy can be exploited to improve efficiency, analyzing the trade-off between inference speed and performance degradation under different pruning strategies. Our results show that moderate pruning can yield substantial acceleration with limited impact on accuracy, and that 20% of the DINOv2 blocks can be removed without any retraining, for a 1.24x throughput gain and a 20% lower energy per image at a relative accuracy loss below 1% on ImageNet-1K. We further propose a model selection pipeline that automatically identifies the best pruned configuration for a given dataset, reducing FLOPs by 13% to 27% on average across architectures with an average accuracy change between-0.3 and +1.0 points.
Sources
- SIMPLER: Efficient Foundation Model Adaptation via Similarity-Guided Layer Pruning for Earth Observation
- DAPlankton: Benchmark Dataset for Multi-instrument Plankton Recognition via Fine-grained Domain Adaptation
- A Comprehensive Survey on Hardware-Aware Neural Architecture Search
- Variance-Based Pruning for Accelerating and Compressing Trained Networks
- Token Merging: Your ViT But Faster
- Tree Species Identification from Bark Images Using Convolutional Neural Networks
- Chasing Sparsity in Vision Transformers: An End-to-End Exploration
- GPTQ: Accurate Post-Training Quantization for Generative Pre-trained Transformers
- The Unreasonable Ineffectiveness of the Deeper Layers
- Learning both Weights and Connections for Efficient Neural Networks
- What Matters in Transformers? Not All Attention is Needed
- Distilling the Knowledge in a Neural Network
- Pruning Filters for Efficient ConvNets
- AWQ: Activation-aware Weight Quantization for LLM Compression and Acceleration
- MLP Can Be A Good Transformer Learner
- GradMAP: Faster Layer Pruning with Gradient Metric and Projection Compensation
- Swin Transformer V2: Scaling Up Capacity and Resolution
- Preserving Deep Representations In One-Shot Pruning: A Hessian-Free Second-Order Optimization Framework
- ShortGPT: Layers in Large Language Models are More Redundant Than You Expect
- DINOv2: Learning Robust Visual Features without Supervision
Related papers
- Loss Knows Best: Detecting Annotation Errors in Videos via Loss Trajectories
- AnchorWeave: World-Consistent Video Generation with Retrieved Local Spatial Memories
- Benchmarking the Robustness of Foundation Models for Mammography under Domain Shift
- MambaX-Net: Dual-Input Mamba-Enhanced Cross-Attention Network for Longitudinal MRI Segmentation
- TeleOCR: Navigating Document Parsing Across Digital and Camera-Captured Documents
- A Survey on Efficient Vision-Language-Action Models