Colinearity Decay: Training Quantization-Friendly ViTs with Outlier Decay

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Tong, Jin, Liang, Guang, Sun, Peilin, Wu, Jianxin
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913083506556928
author Tong, Jin
Liang, Guang
Sun, Peilin
Wu, Jianxin
author_facet Tong, Jin
Liang, Guang
Sun, Peilin
Wu, Jianxin
contents Low-bit quantization is a practical route for efficiently deploying vision Transformers, yet activation outliers complicate fully quantized deployment. Existing methods either handle quantization post-training or suppress large activations during training; however, aggressively restricting outliers in vision models can lead to a poorer trade-off between full-precision and quantized accuracy. We argue that rather than simply suppressing outliers, the training objective should control the structural amplification that makes them harmful. To this end, we introduce Colinearity-Decay (CD), a structural regularizer for ordered matrix pairs within Transformer blocks. CD penalizes detrimental cross-matrix alignment and mitigates extreme activations without altering the architecture or task loss. Applied as a decoupled update, CD is non-invasive and introduces minimal training overhead. Across ImageNet-1K pre-training, COCO detection, and downstream fine-tuning, CD consistently boosts quantized accuracy across multiple pipelines while preserving, or even improving, full-precision performance. Ultimately, our results demonstrate that structural regularization effectively prepares vision Transformers for low-bit deployment with zero inference-time overhead.
format Preprint
id arxiv_https___arxiv_org_abs_2605_01330
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Colinearity Decay: Training Quantization-Friendly ViTs with Outlier Decay
Tong, Jin
Liang, Guang
Sun, Peilin
Wu, Jianxin
Computer Vision and Pattern Recognition
Low-bit quantization is a practical route for efficiently deploying vision Transformers, yet activation outliers complicate fully quantized deployment. Existing methods either handle quantization post-training or suppress large activations during training; however, aggressively restricting outliers in vision models can lead to a poorer trade-off between full-precision and quantized accuracy. We argue that rather than simply suppressing outliers, the training objective should control the structural amplification that makes them harmful. To this end, we introduce Colinearity-Decay (CD), a structural regularizer for ordered matrix pairs within Transformer blocks. CD penalizes detrimental cross-matrix alignment and mitigates extreme activations without altering the architecture or task loss. Applied as a decoupled update, CD is non-invasive and introduces minimal training overhead. Across ImageNet-1K pre-training, COCO detection, and downstream fine-tuning, CD consistently boosts quantized accuracy across multiple pipelines while preserving, or even improving, full-precision performance. Ultimately, our results demonstrate that structural regularization effectively prepares vision Transformers for low-bit deployment with zero inference-time overhead.
title Colinearity Decay: Training Quantization-Friendly ViTs with Outlier Decay
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2605.01330