arXiv:2505.21591cs.LGcs.AI2025-05CVPR被引 6

首次实现扩散模型4比特浮点量化,性能超越现有方法。

Pioneering 4-Bit FP Quantization for Diffusion Models: Mixup-Sign Quantization and Timestep-Aware Fine-Tuning

  • 提出无符号浮点量化与时间步感知微调结合的新框架。
  • 在4比特浮点量化下实现优于整数量化的方法性能。
  • 适合关注高效扩散模型部署的研究者与工程师。

模型量化通过降低权重和激活的位宽,提升扩散模型的内存效率与推理速度。然而,实现4比特量化仍具挑战。现有方法主要基于整数量化与训练后量化微调,存在性能不稳定问题。受大语言模型中浮点量化成功的启发,本文探索扩散模型的低比特浮点量化,并发现三大关键挑战:有符号浮点量化难以处理非对称激活分布,微调时未充分考虑去噪过程的时间复杂性,以及微调损失与量化误差不匹配。为此,提出混合符号浮点量化(MSFP)框架,首次在模型量化中引入无符号浮点量化,结合时间步感知LoRA(TALoRA)与去噪因子损失对齐(DFA),实现精准稳定的微调。大量实验表明,本方法是首个在4比特浮点量化下取得优越性能的方案,显著优于现有4比特整数量化微调方法。

原文摘要 · Abstract (English)

Model quantization reduces the bit-width of weights and activations, improving memory efficiency and inference speed in diffusion models. However, achieving 4-bit quantization remains challenging. Existing methods, primarily based on integer quantization and post-training quantization fine-tuning, struggle with inconsistent performance. Inspired by the success of floating-point (FP) quantization in large language models, we explore low-bit FP quantization for diffusion models and identify key challenges: the failure of signed FP quantization to handle asymmetric activation distributions, the insufficient consideration of temporal complexity in the denoising process during fine-tuning, and the misalignment between fine-tuning loss and quantization error. To address these challenges, we propose the mixup-sign floating-point quantization (MSFP) framework, first introducing unsigned FP quantization in model quantization, along with timestep-aware LoRA (TALoRA) and denoising-factor loss alignment (DFA), which ensure precise and stable fine-tuning. Extensive experiments show that we are the first to achieve superior performance in 4-bit FP quantization for diffusion models, outperforming existing PTQ fine-tuning methods in 4-bit INT quantization.

扩散模型量化浮点量化高效推理

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