arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.02599cs.CLcs.AI
收藏

面向大语言模型对齐的渐进式标签增强

Progressively Label Enhancement for Large Language Model Alignment

School of Computer Science and Engineering(计算机科学与工程学院);Southeast University(东南大学)

Biao Liu, Ning Xu, Xin Geng

AI总结:

针对现有大语言模型对齐方法依赖高质量数据、训练与数据生成割裂导致数据利用效率低的问题,提出PLE渐进式标签增强框架,通过动态阈值匹配训练策略,实现更优的对齐效果。

摘要:

大语言模型(LLM)对齐旨在防止模型生成与人类预期不符的内容,这类内容可能引发伦理和法律问题。过去几年,基于人类反馈的强化学习(RLHF)一直是实现对齐的主流方法。由于多模型间复杂的交互导致RLHF各阶段在稳定性和可扩展性方面存在挑战,研究人员正在探索可达到与RLHF相当效果的替代方法。然而,这些方法往往依赖大规模高质量数据集。尽管部分方法考虑通过生成额外数据来扩充数据集,但它们通常将模型训练和数据生成视为相互独立的静态过程,忽略了二者高度相互依赖的事实,导致生成数据的利用效率低下。为解决这一问题,我们提出PLE,即Progressively Label Enhancement(渐进式标签增强)大语言模型对齐框架,该框架根据生成数据的质量演化动态调整模型的训练过程。具体而言,我们提示模型为原始查询以及由一组精心设计的原则引导的查询生成回复,随后利用动态阈值,根据两类回复对应的奖励分数为其确定合适的训练方式。实验结果表明,与现有大语言模型对齐方法相比,PLE具备更优的有效性。

英文摘要:

Large Language Models (LLM) alignment aims to prevent models from producing content that misaligns with human expectations, which can lead to ethical and legal concerns. In the last few years, Reinforcement Learning from Human Feedback (RLHF) has been the most prominent method for achieving alignment. Due to challenges in stability and scalability with RLHF stages, which arise from the complex interactions between multiple models, researchers are exploring alternative methods to achieve effects comparable to those of RLHF. However, these methods often rely on large high-quality datasets. Despite some methods considering the generation of additional data to expand datasets, they often treat model training and data generation as separate and static processes, overlooking the fact that these processes are highly interdependent, leading to inefficient utilization of the generated data. To deal with this problem, we propose PLE, i.e., Progressively Label Enhancement for LLM Alignment, a framework that dynamically adjusts the model's training process based on the evolving quality of the generated data. Specifically, we prompt the model to generate responses for both the original query and the query guided by a set of carefully designed principles, and then utilize a dynamic threshold to determine the appropriate training approach for both responses based on their corresponding reward scores. Experimental results demonstrate the effectiveness of PLE compared to existing LLM alignment methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑