arXiv:2605.09503cs.CV2026-05被引 1

通过重排通道降低扩散模型量化误差,提升低比特部署效率。

PermuQuant: Lowering Per-Group Quantization Error by Reordering Channels for Diffusion Models

论文配图:PermuQuant: Lowering Per-Group Quantization Error by Reordering Channels for Diffusion Models
图 1 · 摘自论文原文
  • 按激活与权重统计相似性重排通道,优化分组量化
  • 在W4A4 NVFP4下实现3.5倍内存压缩和1.7倍加速
  • 无需运行时操作,适合资源受限场景部署

大规模视觉生成模型性能卓越,但其高计算与内存开销限制了在交互应用和单GPU设备上的部署。后训练量化(PTQ)可在不重新训练的情况下压缩模型,但现有方法在极低比特设置下仍存在严重质量下降。本文发现通道排序是分组量化中被忽视的关键因素:当统计差异大的通道共处一组时,量化尺度易受异常值主导,导致大误差。为此提出PermuQuant,一种简单高效的低比特扩散模型PTQ框架。该方法在分组量化前,依据联合二阶矩准则对通道进行排序,使统计相近的通道归入同组,并采用校准数据驱动的接受规则,仅在能降低量化误差时才应用重排。重排结果可嵌入相邻模块或离线应用于权值,避免运行时操作。大量实验表明,PermuQuant持续降低量化误差,优于现有基线。在FLUX.1-dev上使用RTX 5090,W4A4 NVFP4量化下实现最高1.7倍单步加速,且DiT内存占用减少3.5倍。代码将开源于https://github.com/yscheng04/PermuQuant。

原文摘要 · Abstract (English)

Large-scale visual generative models have achieved remarkable performance. However, their high computational and memory costs make deployment challenging in resource-constrained scenarios, such as interactive applications and personal single-GPU usage. Post-training quantization (PTQ) offers a practical solution by compressing pretrained models without expensive retraining. However, existing PTQ methods still suffer from severe quality degradation under extremely low-bit settings. In this paper, we identify channel ordering as an important but underexplored factor in per-group quantization. In this setting, each contiguous group shares one quantization scale. When channels with very different statistics are placed in the same group, the scale can be dominated by outliers and cause large quantization errors. Based on this observation, we propose PermuQuant, a simple and effective PTQ framework for low-bit diffusion models. PermuQuant sorts channels by a joint second-moment criterion before per-group quantization, placing channels with similar activation and weight statistics into the same group. It further uses a calibration-based acceptance rule to apply reordering only when the selected permutation reduces quantization error on calibration data. The selected permutations are absorbed into adjacent modules or applied to weights offline, avoiding explicit runtime permutation operations. Extensive experiments on multiple large diffusion models show that PermuQuant consistently reduces quantization error and outperforms existing PTQ baselines. On FLUX.1-dev with an RTX 5090, PermuQuant achieves up to a 1.7$\times$ single step speedup and reduces the DiT memory footprint by 3.5$\times$ under W4A4 NVFP4 quantization. Code will be available at https://github.com/yscheng04/PermuQuant.

量化扩散模型低比特

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