用自动筛选方法,用不到0.5%数据提升文生图模型对齐效果。
Automated Filtering of Human Feedback Data for Aligning Text-to-Image Diffusion Models
- 通过优化偏好边界、文本质量和多样性,自动筛选高价值反馈数据。
- 仅用0.5%数据即获得17%更高人类偏好,训练耗时仅为1%。
- 无需人工标注,适用于任意大规模反馈数据集,适合高效对齐研究。
使用人类反馈微调文生图扩散模型是实现模型行为与人类意图对齐的有效方法,但该过程常因数据集规模大且噪声多导致收敛缓慢。本文提出FiFA,一种基于直接偏好优化(DPO)的自动化数据过滤算法。该方法通过求解优化问题,最大化三个目标:偏好边界(利用代理奖励模型计算)、文本质量(由大语言模型评估以避免有害内容)和文本多样性(通过k近邻熵估计器衡量)。将三者整合进统一优化框架,通过为每个数据对分配重要性得分并选择最高分项,实现自动高效的数据筛选。实验表明,FiFA显著提升训练稳定性,仅使用全量数据的0.5%,即可使人类偏好提升17%,训练所需GPU时长降低至1%。
原文摘要 · Abstract (English)
Fine-tuning text-to-image diffusion models with human feedback is an effective method for aligning model behavior with human intentions. However, this alignment process often suffers from slow convergence due to the large size and noise present in human feedback datasets. In this work, we propose FiFA, a novel automated data filtering algorithm designed to enhance the fine-tuning of diffusion models using human feedback datasets with direct preference optimization (DPO). Specifically, our approach selects data by solving an optimization problem to maximize three components: preference margin, text quality, and text diversity. The concept of preference margin is used to identify samples that are highly informative in addressing the noisy nature of feedback dataset, which is calculated using a proxy reward model. Additionally, we incorporate text quality, assessed by large language models to prevent harmful contents, and consider text diversity through a k-nearest neighbor entropy estimator to improve generalization. Finally, we integrate all these components into an optimization process, with approximating the solution by assigning importance score to each data pair and selecting the most important ones. As a result, our method efficiently filters data automatically, without the need for manual intervention, and can be applied to any large-scale dataset. Experimental results show that FiFA significantly enhances training stability and achieves better performance, being preferred by humans 17% more, while using less than 0.5% of the full data and thus 1% of the GPU hours compared to utilizing full human feedback datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。