arXiv:2503.14868cs.CVcs.AI2025-03CVPR被引 2

无需反向传播即可高效微调量化扩散模型,适合移动端个性化应用。

Efficient Personalization of Quantized Diffusion Model without Backpropagation

  • 用零阶优化直接更新个性化令牌,避免梯度存储
  • 通过历史令牌子空间降噪梯度,提升微调稳定性
  • 仅需前向计算,内存消耗降低8.2倍,适合边缘设备

扩散模型在图像生成中表现优异,但训练、微调和推理需要大量计算与内存资源。尽管先进量化技术已有效减少推理内存占用,但量化模型的训练与微调仍需大内存,通常因反向传播所需的梯度与激活值存储所致。针对移动端私有数据个性化需求,本文提出一种无需反向传播的量化扩散模型个性化方法:基于Textual Inversion,采用零阶优化更新个性化令牌,避免梯度与激活存储。为解决单张或少量图像下零阶梯度噪声大的问题,提出通过历史令牌构建子空间进行梯度投影,称为Subspace Gradient。此外,研究文本嵌入对生成的影响,提出部分均匀时间步采样(Partial Uniform Timestep Sampling),以更有效利用扩散时间步。实验表明,本方法仅使用前向传播,在保持与先前方法相当的图像-文本对齐分数的同时,将训练内存需求最高降低8.2倍。

原文摘要 · Abstract (English)

Diffusion models have shown remarkable performance in image synthesis, but they demand extensive computational and memory resources for training, fine-tuning and inference. Although advanced quantization techniques have successfully minimized memory usage for inference, training and fine-tuning these quantized models still require large memory possibly due to dequantization for accurate computation of gradients and/or backpropagation for gradient-based algorithms. However, memory-efficient fine-tuning is particularly desirable for applications such as personalization that often must be run on edge devices like mobile phones with private data. In this work, we address this challenge by quantizing a diffusion model with personalization via Textual Inversion and by leveraging a zeroth-order optimization on personalization tokens without dequantization so that it does not require gradient and activation storage for backpropagation that consumes considerable memory. Since a gradient estimation using zeroth-order optimization is quite noisy for a single or a few images in personalization, we propose to denoise the estimated gradient by projecting it onto a subspace that is constructed with the past history of the tokens, dubbed Subspace Gradient. In addition, we investigated the influence of text embedding in image generation, leading to our proposed time steps sampling, dubbed Partial Uniform Timestep Sampling for sampling with effective diffusion timesteps. Our method achieves comparable performance to prior methods in image and text alignment scores for personalizing Stable Diffusion with only forward passes while reducing training memory demand up to $8.2\times$.

扩散模型量化个性化零阶优化

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