OrbitQuant: Data-Agnostic Quantization for Image and Video Diffusion Transformers
OrbitQuant:图像与视频扩散 Transformer 的数据无关量化
🧭 扩散 Transformer PTQ · 数据无关旋转量化
Cantina Labs · University of Southern California · University of Illinois Urbana-Champaign
在归一化旋转基上量化,用固定码本适应所有时间步和模态,无需数据重校正,将 DiT PTQ 推进到 W2A4。
📌 工作参考价值(4 点)
- 对推荐系统中可能使用的 Transformer 或扩散生成模型,旋转基量化可消除校准集依赖,避免因用户行为分布漂移而反复重校。
- 权重旋转离线吸收进线性层,推理时仅对激活做前向旋转,工程实现简洁高效,可直接参考其算子融合技巧。
- 单一码本跨层复用,大幅降低推荐模型量化时的码本存储与设计复杂度,适合存储受限的线上环境。
- W2A4 极低比特可成倍压缩推理带宽,在电商直播、短视频内容生成等需高吞吐的 GPU 受限场景具备直接落地价值。