用低秩分解优化扩散模型的去噪路径,实现高效微调。
LoRA-Diffusion: Parameter-Efficient Fine-Tuning via Low-Rank Trajectory Decomposition

- 通过低秩分解去噪轨迹而非权重矩阵,实现参数高效微调。
- 在SST-2、QNLI、MRPC上均达到最优或强性能,五次实验均稳定。
- 支持多任务合并推理,适合需要轻量化部署的场景。
参数高效微调方法如LoRA已成功应用于大规模自回归语言模型,实现任务定制化仅需极少可训练参数。然而,此类方法尚未有效扩展至基于扩散的语言模型——这类模型通过迭代去噪生成文本,而非逐个预测词元。本文提出LoRA-Diffusion,一种将低秩分解应用于去噪轨迹而非模型权重的参数高效微调方法。与基于权重的LoRA不同,该方法学习从噪声到输出的完整去噪路径上的低秩扰动。我们引入轨迹级低秩适配器,对每个去噪步骤进行修改;设计跨扩散阶段的自适应秩分配策略;并实现组合式多任务学习,在推理时合并任务模块而无需重新训练。在SST-2、QNLI和MRPC数据集上,报告了五次随机种子下的词元级去噪验证准确率。结果表明,LoRA-Diffusion在SST-2上取得最高平均性能,在QNLI和MRPC上表现强劲。联合多任务训练进一步显示,其在所有评估方法中达到最高的词元级准确率。该方法显著降低每任务存储需求,为扩散语言模型建立了参数高效微调框架。
原文摘要 · Abstract (English)
Parameter-efficient fine-tuning methods such as LoRA have transformed the adaptation of large autoregressive language models, enabling task-specific customization with substantially fewer trainable parameters. However, these methods have not been successfully extended to diffusion-based language models, which generate text through iterative denoising rather than sequential token prediction. We propose LoRA-Diffusion, a parameter-efficient fine-tuning approach that applies low-rank decomposition to the denoising trajectory instead of model weights. Unlike weight-based LoRA, which modifies individual transformation matrices, our method learns low-rank perturbations to the entire diffusion path from noise to output. We introduce trajectory-level low-rank adapters that modify each denoising step, step-adaptive rank allocation across diffusion phases, and compositional multi-task learning that allows merging task-specific modules at inference without retraining. On SST-2, QNLI, and MRPC, we report token-level denoising validation accuracy over five random seeds. LoRA-Diffusion achieves the highest mean performance on SST-2 and strong performance on QNLI and MRPC. Joint multi-task training further shows that LoRA-Diffusion achieves the highest token-level accuracy among the evaluated methods. The approach reduces per-task storage compared with full fine-tuning and establishes a parameter-efficient fine-tuning framework for diffusion language models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。