arXiv:2509.24416cs.CVcs.AI2025-09被引 2

提出CLQ量化方法,让扩散Transformer模型在边端设备上高效运行且质量损失极小。

CLQ: Cross-Layer Guided Orthogonal-based Quantization for Diffusion Transformers

  • 通过跨块校准获取更真实的激活分布,指导量化过程。
  • 用正交平滑技术抑制通道异常值,几乎无额外开销。
  • 跨层参数搜索提升压缩后模型性能,适合部署于资源受限设备。

视觉生成质量因扩散Transformer(DiTs)的规模与复杂度提升而显著提高,但这也阻碍了其在边缘设备上的实际部署。模型后训练量化(PTQ)作为高效的模型压缩技术,虽可降低内存占用并加速推理,但不可避免带来性能下降。为缓解此问题,我们提出CLQ——一种面向扩散Transformer的跨层引导正交量化方法。具体包括三项关键设计:首先,观察到现有PTQ方法使用的校准数据无法真实反映激活分布,因此提出跨块校准(CBC)以获取更准确的校准数据;其次,提出基于正交的平滑(OBS),量化各通道的离群程度,并利用块哈达玛矩阵进行平滑,开销极低;第三,引入跨层参数搜索(CLPS)优化量化配置。我们在图像与视频生成模型上评估了CLQ,成功将模型压缩至W4A4,视觉质量与指标几乎无损。相比原模型,实现3.98倍内存节省与3.95倍推理加速。代码已开源。

原文摘要 · Abstract (English)

Visual generation quality has been greatly promoted with the rapid advances in diffusion transformers (DiTs), which is attributed to the scaling of model size and complexity. However, these attributions also hinder the practical deployment of DiTs on edge devices, limiting their development and application. Serve as an efficient model compression technique, model post-training quantization (PTQ) can reduce the memory consumption and speed up the inference, with inevitable performance degradation. To alleviate the degradation, we propose CLQ, a cross-layer guided orthogonal-based quantization method for DiTs. To be specific, CLQ consists of three key designs. First, we observe that the calibration data used by most of the PTQ methods can not honestly represent the distribution of the activations. Therefore, we propose cross-block calibration (CBC) to obtain accurate calibration data, with which the quantization can be better guided. Second, we propose orthogonal-based smoothing (OBS), which quantifies the outlier score of each channel and leverages block Hadamard matrix to smooth the outliers with negligible overhead. Third, we propose cross-layer parameter searching (CLPS) to search. We evaluate CLQ with both image generation and video generation models and successfully compress the model into W4A4 with negligible degradation in visual quality and metrics. CLQ achieves 3.98x memory saving and 3.95x speedup. Our code is available at \hyperlink{https://github.com/Kai-Liu001/CLQ}{https://github.com/Kai-Liu001/CLQ}.

扩散模型量化边缘计算模型压缩

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。