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

偏好优化链:提升大语言模型的思维链推理能力

Chain of Preference Optimization: Improving Chain-of-Thought Reasoning in LLMs

Sea AI Lab(Sea AI实验室);School of Computing and Information Systems, Singapore Management University(新加坡管理大学计算与信息系统学院)

Xuan Zhang, Chao Du, Tianyu Pang, Qian Liu, Wei Gao, Min Lin

AI总结:

针对大语言模型思维链推理路径非最优、思维树方法推理复杂度高的问题,提出偏好优化链方法,借助思维树搜索树的偏好信息微调模型,让思维链推理达到更优性能且降低推理负担。

摘要:

思维链(Chain-of-Thought, CoT)解码技术的最新发展,使得大语言模型(Large Language Models, LLMs)能够为复杂问题求解生成显式的逻辑推理路径。但研究表明,这些路径并非总是经过审慎推导且最优的。思维树(Tree-of-Thought, ToT)方法采用树搜索来广泛探索推理空间,找到CoT解码可能遗漏的更优推理路径,然而这种审慎推导的代价是推理复杂度大幅提升。本研究表明,利用ToT构建的搜索树微调LLMs,可让CoT达到相近甚至更优的性能,从而避免沉重的推理负担。该效果通过偏好优化链(Chain of Preference Optimization, CPO)实现:借助树搜索过程中的内在偏好信息微调LLMs,使CoT推理路径的每一步都与ToT的推理路径对齐。大量实验结果显示,CPO显著提升了LLM在问答、事实验证、算术推理等各类复杂问题求解上的表现,证明了其有效性。我们的代码可在https://github.com/sail-sg/CPO获取。

英文摘要:

The recent development of chain-of-thought (CoT) decoding has enabled large language models (LLMs) to generate explicit logical reasoning paths for complex problem-solving. However, research indicates that these paths are not always deliberate and optimal. The tree-of-thought (ToT) method employs tree-searching to extensively explore the reasoning space and find better reasoning paths that CoT decoding might overlook. This deliberation, however, comes at the cost of significantly increased inference complexity. In this work, we demonstrate that fine-tuning LLMs leveraging the search tree constructed by ToT allows CoT to achieve similar or better performance, thereby avoiding the substantial inference burden. This is achieved through Chain of Preference Optimization (CPO), where LLMs are fine-tuned to align each step of the CoT reasoning paths with those of ToT using the inherent preference information in the tree-search process. Extensive experimental results show that CPO significantly improves LLM performance in solving a variety of complex problems, including question answering, fact verification, and arithmetic reasoning, demonstrating its effectiveness. Our code is available at https://github.com/sail-sg/CPO.

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

↑