LOAD: Enabling Low-Bit Group Quantization for Vision Transformers via Adaptive Data Formats
Vision Transformers (ViTs) achieve state-of-the-art performance across various vision tasks but pose significant deployment challenges due to their high computational complexity and memory footprint. Post-training quantization (PTQ) offers an efficient, retraining-free compression strategy; however, existing solutions typically enforce a uniform quantization scheme (fixed bit-width and data format) across all layers, constraining the potential accuracy-efficiency trade-off. We propose LOAD-ViT , a novel PTQ framework that leverages both mixed-precision and mixed-data-format quantization to allocate resources at a fine granularity based on layer sensitivity. LOAD-ViT optimizes per-layer configurations via a calibration-based sensitivity metric and a constrained search algorithm, minimizing accuracy loss under strict hardware budgets. To support this scheme, we design a dedicated hardware accelerator featuring heterogeneous compute engines and lightweight format-conversion units, facilitating the efficient execution of mixed formats. Extensive evaluations on various ViT models and benchmarks demonstrate that LOAD-ViT achieves up to 5.3 × speedup and 4.2 × energy savings compared to state-of-the-art quantization accelerators, with negligible accuracy degradation.
Authors
- Yuang Peng (ORCID: https://orcid.org/0000-0002-1448-5489)
- Fangxin Liu (ORCID: https://orcid.org/0000-0002-8769-293X)
- Songwen Pei (ORCID: https://orcid.org/0000-0003-0810-1458)
- Ning Yang (ORCID: https://orcid.org/0009-0004-6964-8910)
- Yongbiao Chen (ORCID: https://orcid.org/0000-0002-7727-5489)
- Haibing Guan (ORCID: https://orcid.org/0000-0002-4714-7400)
- Chenyang Guan (ORCID: https://orcid.org/0009-0008-6422-7729)
- Junjie Wang (ORCID: https://orcid.org/0009-0003-8754-7948)
- Haoming Li (ORCID: https://orcid.org/0000-0002-2939-6534)
- Li Jiang (ORCID: https://orcid.org/0000-0002-7353-8798)
- Xinran Liang (ORCID: https://orcid.org/0009-0004-8442-873X)
Institutions
- University of Shanghai for Science and Technology (CN)
- National University of Defense Technology (CN)
- China Mobile (China) (CN)
- Shanghai Jiao Tong University (CN)
- United Imaging Intelligence (China) (CN)
- Northeastern University (CN)
Publication Details
- Journal
- ACM Transactions on Architecture and Code Optimization
- Published
- 2026-10-15
- DOI
- https://doi.org/10.1145/3857350
- Primary Topic
- Advanced Neural Network Applications
- Type
- article
- Field-Weighted Citation Impact
- 0.00