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

YODA:面向语言模型的师生渐进式学习

YODA: Teacher-Student Progressive Learning for Language Models

The University of Hong Kong(香港大学);Sun Yat-sen University(中山大学)

Jianqiao Lu, Wanjun Zhong, Yufei Wang, Zhijiang Guo, Qi Zhu, Wenyong Huang, Yanlin Wang, Fei Mi, Baojun Wang, Yasheng Wang, Lifeng Shang, Xin Jiang, Qun Liu

AI总结:

针对大语言模型学习效率低于人类的问题,本文提出模拟师生教学的YODA渐进学习框架,通过基础-泛化-进阶的交互式循环生成流程数据训练模型,在数学推理任务上显著提升微调性能,结合课程学习可增强鲁棒性。

摘要:

尽管大语言模型(LLMs)已在诸多任务中展现出出色能力,但它们的学习效率仍落后于人类。这种差距通常与人类与生俱来的能力相关:人类能从基础示例中学习,逐步泛化并处理更复杂的问题,还能通过持续反馈优化技能。受此启发,本文提出YODA——一种新型师生渐进式学习框架,它模拟师生教学过程以提升模型微调的效果。该框架以交互式的“基础-泛化-进阶”循环运行。教师智能体针对学生的答案提供定制化反馈,并系统性地组织教学流程:先通过基础示例对学生进行教学,再通过泛化问题强化理解,随后提出复杂度逐步提升的问题以深化学习。在教师的引导下,学生学会借助反馈迭代优化自己的答案,并对所提出的问题形成稳固且全面的理解。随后,这些反映人类渐进学习过程的系统化流程数据被用于模型训练。以数学推理为测试场景,实验表明,使用YODA生成的数据训练LLaMA2,能显著提升有监督微调(SFT)的性能——在GSM8K上提升17.01%,在MATH上提升9.98%。此外,我们发现结合课程学习进行训练能进一步提升学习的鲁棒性。

英文摘要:

Although large language models (LLMs) have demonstrated adeptness in a range of tasks, they still lag behind human learning efficiency. This disparity is often linked to the inherent human capacity to learn from basic examples, gradually generalize and handle more complex problems, and refine their skills with continuous feedback. Inspired by this, this paper introduces YODA, a novel teacher-student progressive learning framework that emulates the teacher-student education process to improve the efficacy of model fine-tuning. The framework operates on an interactive \textit{basic-generalized-harder} loop. The teacher agent provides tailored feedback on the student's answers, and systematically organizes the education process. This process unfolds by teaching the student basic examples, reinforcing understanding through generalized questions, and then enhancing learning by posing questions with progressively enhanced complexity. With the teacher's guidance, the student learns to iteratively refine its answer with feedback, and forms a robust and comprehensive understanding of the posed questions. The systematic procedural data, which reflects the progressive learning process of humans, is then utilized for model training. Taking math reasoning as a testbed, experiments show that training LLaMA2 with data from YODA improves SFT with significant performance gain (+17.01\% on GSM8K and +9.98\% on MATH). In addition, we find that training with curriculum learning further improves learning robustness.

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

↑