arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2406.08657cs.CL
收藏

Mistral-C2F:用于增强RLHF与高效合并LLM中分析和推理能力的由粗到细Actor

Mistral-C2F: Coarse to Fine Actor for Analytical and Reasoning Enhancement in RLHF and Effective-Merged LLMs

Bytedance(字节跳动)

Chen Zheng, Ke Sun, Xun Zhou

AI总结:

本文提出两步式由粗到细Actor框架Mistral-C2F,通过Continuous Maximization增强RLHF分析生成,并用Knowledge Residue Merger合并精炼,显著提升小型Mistral的对话与推理能力。

摘要:

尽管以GPT-4和Claude等模型为代表的大型语言模型(LLM)取得了进展,但Llama和Mistral等较小规模的LLM往往难以生成深入且连贯的对话。本文提出了一种新颖的两步式由粗到细Actor模型,以解决小型LLM在对话和分析能力方面的固有局限。我们的方法从基于策略的Coarse Actor开始,采用了一种我们称为“Continuous Maximization”的技术。Coarse Actor建立了一个增强的、知识丰富的池,善于契合人类在分析和推理方面的偏好风格。在RLHF过程中,它采用Continuous Maximization这一策略,动态且自适应地延长输出长度限制,从而能够生成更详细、更具分析性的内容。随后,Fine Actor对这些分析内容进行精炼,解决Coarse Actor生成过度冗余信息的问题。我们引入了一种“Knowledge Residue Merger”方法,对来自Coarse Actor的内容进行精炼,并将其与现有的Instruction模型合并,以提升质量和正确性并减少冗余。我们将该方法应用于流行的Mistral模型,创建了Mistral-C2F;该模型在11项通用语言任务和MT-Bench Dialogue任务上展现出卓越性能,优于相似规模模型,甚至优于具有13B和30B参数的更大模型。我们的模型显著提升了对话与分析推理能力。

英文摘要:

Despite the advances in Large Language Models (LLMs), exemplified by models like GPT-4 and Claude, smaller-scale LLMs such as Llama and Mistral often struggle with generating in-depth and coherent dialogues. This paper presents a novel two-step Coarse-to-Fine Actor model to address the inherent limitations in conversational and analytical capabilities of small-sized LLMs. Our approach begins with the Policy-based Coarse Actor, employing a technique we term "Continuous Maximization". The Coarse Actor establishes an enhanced, knowledge-rich pool adept at aligning with human preference styles in analysis and reasoning. Through the RLHF process, it employs Continuous Maximization, a strategy that dynamically and adaptively extends the output length limit, enabling the generation of more detailed and analytical content. Subsequently, the Fine Actor refines this analytical content, addressing the generation of excessively redundant information from the Coarse Actor. We introduce a "Knowledge Residue Merger" approach, refining the content from the Coarse Actor and merging it with an existing Instruction model to improve quality, correctness, and reduce redundancies. We applied our methodology to the popular Mistral model, creating Mistral-C2F, which has demonstrated exceptional performance across 11 general language tasks and the MT-Bench Dialogue task, outperforming similar-scale models and even larger models with 13B and 30B parameters. Our model has significantly improved conversational and analytical reasoning abilities.

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

↑