提出分步优化的高效微调方法,提升扩散模型动作生成对齐效果。
EasyTune: Efficient Step-Aware Fine-Tuning for Diffusion-Based Motion Generation
- 在去噪每一步独立微调,打破步骤间递归依赖
- 相比基线提升8.2%对齐度,内存开销仅31.16%
- 自适应生成偏好对,缓解标注数据稀缺问题
近年来,动作生成模型取得显著进展,但在对齐下游目标方面仍面临挑战。已有研究通过可微奖励直接对齐扩散模型偏好,但存在优化效率低、粒度粗以及内存消耗高的问题。本文从理论上和实证上揭示其根源:去噪轨迹中各步骤间的递归依赖。为此,提出EasyTune,针对去噪过程中的每个步骤分别进行微调,从而解耦递归依赖,实现更密集、细粒度且内存高效的优化。此外,由于偏好动作对稀缺,限制了动作奖励模型训练。为此引入自重构偏好学习(SPL)机制,动态识别偏好对并开展偏好学习。大量实验表明,EasyTune在对齐度(MM-Dist)上比DRaFT-50提升8.2%,额外内存开销仅为31.16%,训练速度提升7.3倍。
原文摘要 · Abstract (English)
In recent years, motion generative models have undergone significant advancement, yet pose challenges in aligning with downstream objectives. Recent studies have shown that using differentiable rewards to directly align the preference of diffusion models yields promising results. However, these methods suffer from (1) inefficient and coarse-grained optimization with (2) high memory consumption. In this work, we first theoretically and empirically identify the key reason of these limitations: the recursive dependence between different steps in the denoising trajectory. Inspired by this insight, we propose EasyTune, which fine-tunes diffusion at each denoising step rather than over the entire trajectory. This decouples the recursive dependence, allowing us to perform (1) a dense and fine-grained, and (2) memory-efficient optimization. Furthermore, the scarcity of preference motion pairs restricts the availability of motion reward model training. To this end, we further introduce a Self-refinement Preference Learning (SPL) mechanism that dynamically identifies preference pairs and conducts preference learning. Extensive experiments demonstrate that EasyTune outperforms DRaFT-50 by 8.2% in alignment (MM-Dist) improvement while requiring only 31.16% of its additional memory overhead and achieving a 7.3x training speedup. The project page is available at this link {https://xiaofeng-tan.github.io/projects/EasyTune/index.html}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。