针对大模型3D重建的量化难题,提出首个专用压缩框架。
Quantized Visual Geometry Grounded Transformer
- 设计双平滑细粒度量化,缓解激活分布不均问题。
- 通过噪声过滤采样构建稳定校准集,提升量化精度。
- 4比特量化实现3.7倍内存压缩与2.5倍加速,精度超98%。
基于学习的3D重建模型(如视觉几何引导变压器,VGGT)虽借助大规模Transformer取得显著进展,但其高昂的计算与内存开销严重阻碍实际部署。后训练量化(PTQ)已成为压缩和加速模型的常用手段。然而我们发现,将PTQ应用于数十亿参数的VGGT时面临独特挑战:数据无关的特殊标记导致激活分布呈重尾特征,而3D数据的多视角特性使校准样本选择极不稳定。本文提出首个专用于VGGT的量化框架QuantVGGT,包含两项关键技术:第一,引入双平滑细粒度量化,结合预全局Hadamard旋转与后局部通道平滑,鲁棒地缓解重尾分布与跨通道方差;第二,设计噪声过滤多样化采样,通过深层统计过滤异常值,并构建帧感知的多样化校准簇,确保量化范围稳定。大量实验表明,QuantVGGT在多个基准与比特位宽下均达当前最优,显著超越此前通用量化方法。特别地,4比特量化版本在真实硬件上实现3.7倍内存减少与2.5倍加速,同时保持超过全精度模型98%的重建精度,充分展现其在资源受限场景下的巨大优势与实用性。代码已开源。
原文摘要 · Abstract (English)
Learning-based 3D reconstruction models, represented by Visual Geometry Grounded Transformers (VGGTs), have made remarkable progress with the use of large-scale transformers. Their prohibitive computational and memory costs severely hinder real-world deployment. Post-Training Quantization (PTQ) has become a common practice for compressing and accelerating models. However, we empirically observe that PTQ faces unique obstacles when compressing billion-scale VGGTs: the data-independent special tokens induce heavy-tailed activation distributions, while the multi-view nature of 3D data makes calibration sample selection highly unstable. This paper proposes the first Quantization framework for VGGTs, namely QuantVGGT. This mainly relies on two technical contributions: First, we introduce Dual-Smoothed Fine-Grained Quantization, which integrates pre-global Hadamard rotation and post-local channel smoothing to mitigate heavy-tailed distributions and inter-channel variance robustly. Second, we design Noise-Filtered Diverse Sampling, which filters outliers via deep-layer statistics and constructs frame-aware diverse calibration clusters to ensure stable quantization ranges. Comprehensive experiments demonstrate that QuantVGGT achieves the state-of-the-art results across different benchmarks and bit-width, surpassing the previous state-of-the-art generic quantization method with a great margin. We highlight that our 4-bit QuantVGGT can deliver a 3.7$\times$ memory reduction and 2.5$\times$ acceleration in real-hardware inference, while maintaining reconstruction accuracy above 98\% of its full-precision counterpart. This demonstrates the vast advantages and practicality of QuantVGGT in resource-constrained scenarios. Our code is released in https://github.com/wlfeng0509/QuantVGGT.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。