Rotation Is Token-Blind: A Structural Criterion for Post-Training Quantization of Feed-Forward Geometry Transformers
Huanfei He, Bohao Zhang, Jiaxin Zhang, Wei Sui, John See, Zeyd Boukhers, Cong Yang
Abstract
Feed-forward 3D models use Transformers to predict camera poses, depth, and structure in one pass, avoiding iterative multi-view optimization. Long sequences and deep attention incur high compute and memory costs, motivating Post-Training Quantization (PTQ) for deployment. Existing PTQ methods mainly suppress channel-wise outliers by smoothing or orthogonal rotation. Geometry Transformers introduce a different challenge: heterogeneous token types share one activation quantization grid. A few special tokens may exert high leverage over global outputs and be updated by only a subset of blocks. Token-blind transforms, which apply the same transformation to every token, reduce overall quantization error but cannot remove differentiated degradation across token types. We propose QMapAnything, a structure-aware PTQ framework. We cast floating-point-preserving rotation and smoothing as a token-blind reparameterization group, and prove that they yield common-mode gains while leaving relative token SNR unchanged. We introduce a statistic Γ that combines a special token's quantization disadvantage and output leverage to predict whether differentiated bias correction is needed beyond rotation. Experiments on geometric benchmarks show that W4A8 closely matches full-precision performance. Under W8A8, peak memory falls by 2.2--3.1× versus FP32, with up to 3.1× speedup on real hardware. These results provide a principled and efficient approach to deploying feed-forward 3D reconstruction models in resource-constrained edges.
Create a lesson
Related papers
Deformable Contact-Aware 3D Object Placement
Songhe Wang, David Miller, Charles A. Kamhoua
When GeoFMs Meet VHR: Benchmarking Foundation Models on Very High Resolution Earth Observation Imagery
Kshitij Raj Sharma, Pierre Adorni, Minh-Tan Pham et al.
Reviving the Discarded High-Resolution Feature for Transformer-Based Tiny Object Detection
Zehao Wang, Si Wu, Guiping Cao et al.
Bri3D: Bridging the Object-Scene Gap in Single-Image 3D Generation via Scene-Anchor
Hongye Hou, Shuai Zhang, Hongjie Li et al.
IDPrior: Decoupling Identity Generation from Facial Image Synthesis
Fadi Boutros, Meriç Demirörs, Darian Tomašević et al.
WAM-Sim0.1: World Anchored Memory Meets Online RL for Consistent and Realistic Driving Simulation
Jiahao Cui, Jiaye Li, Wenhao Zhang et al.