Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Deng, Yihe, Mineiro, Paul
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:https://arxiv.org/abs/2410.22304
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866929567062556672
author Deng, Yihe
Mineiro, Paul
author_facet Deng, Yihe
Mineiro, Paul
contents Mathematical reasoning is a crucial capability for Large Language Models (LLMs), yet generating detailed and accurate reasoning traces remains a significant challenge. This paper introduces a novel approach to produce high-quality reasoning traces for LLM fine-tuning using online learning \textbf{Flows}. Our method employs an incremental output production Flow, where component LLMs collaboratively construct solutions through iterative communication. We train the Flow using online Direct Preference Optimization (DPO) learning with rollouts, generating DPO pairs for each training example and updating models in real-time. We directly compare the quality of reasoning traces generated by our method with those produced through direct model inference, demonstrating the effectiveness of our approach in improving LLM performance in mathematical reasoning tasks.
format Preprint
id arxiv_https___arxiv_org_abs_2410_22304
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Flow-DPO: Improving LLM Mathematical Reasoning through Online Multi-Agent Learning
Deng, Yihe
Mineiro, Paul
Computation and Language
Machine Learning
Mathematical reasoning is a crucial capability for Large Language Models (LLMs), yet generating detailed and accurate reasoning traces remains a significant challenge. This paper introduces a novel approach to produce high-quality reasoning traces for LLM fine-tuning using online learning \textbf{Flows}. Our method employs an incremental output production Flow, where component LLMs collaboratively construct solutions through iterative communication. We train the Flow using online Direct Preference Optimization (DPO) learning with rollouts, generating DPO pairs for each training example and updating models in real-time. We directly compare the quality of reasoning traces generated by our method with those produced through direct model inference, demonstrating the effectiveness of our approach in improving LLM performance in mathematical reasoning tasks.
title Flow-DPO: Improving LLM Mathematical Reasoning through Online Multi-Agent Learning
topic Computation and Language
Machine Learning
url https://arxiv.org/abs/2410.22304