Large Language Models Can Self-Improve in Long-context Reasoning

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Li, Siheng, Yang, Cheng, Cheng, Zesen, Liu, Lemao, Yu, Mo, Yang, Yujiu, Lam, Wai
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910695861256192
author Li, Siheng
Yang, Cheng
Cheng, Zesen
Liu, Lemao
Yu, Mo
Yang, Yujiu
Lam, Wai
author_facet Li, Siheng
Yang, Cheng
Cheng, Zesen
Liu, Lemao
Yu, Mo
Yang, Yujiu
Lam, Wai
contents Large language models (LLMs) have achieved substantial progress in processing long contexts but still struggle with long-context reasoning. Existing approaches typically involve fine-tuning LLMs with synthetic data, which depends on annotations from human experts or advanced models like GPT-4, thus restricting further advancements. To address this issue, we investigate the potential for LLMs to self-improve in long-context reasoning and propose \ours, an approach specifically designed for this purpose. This approach is straightforward: we sample multiple outputs for each question, score them with Minimum Bayes Risk, and then apply supervised fine-tuning or preference optimization based on these outputs. Extensive experiments on several leading LLMs demonstrate the effectiveness of \ours, with an absolute improvement of $4.2$ points for Llama-3.1-8B-Instruct. Furthermore, \ours achieves superior performance compared to prior approaches that depend on data produced by human experts or advanced models. We anticipate that this work will open new avenues for self-improvement techniques in long-context scenarios, which are essential for the continual advancement of LLMs.
format Preprint
id arxiv_https___arxiv_org_abs_2411_08147
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Large Language Models Can Self-Improve in Long-context Reasoning
Li, Siheng
Yang, Cheng
Cheng, Zesen
Liu, Lemao
Yu, Mo
Yang, Yujiu
Lam, Wai
Computation and Language
Artificial Intelligence
Large language models (LLMs) have achieved substantial progress in processing long contexts but still struggle with long-context reasoning. Existing approaches typically involve fine-tuning LLMs with synthetic data, which depends on annotations from human experts or advanced models like GPT-4, thus restricting further advancements. To address this issue, we investigate the potential for LLMs to self-improve in long-context reasoning and propose \ours, an approach specifically designed for this purpose. This approach is straightforward: we sample multiple outputs for each question, score them with Minimum Bayes Risk, and then apply supervised fine-tuning or preference optimization based on these outputs. Extensive experiments on several leading LLMs demonstrate the effectiveness of \ours, with an absolute improvement of $4.2$ points for Llama-3.1-8B-Instruct. Furthermore, \ours achieves superior performance compared to prior approaches that depend on data produced by human experts or advanced models. We anticipate that this work will open new avenues for self-improvement techniques in long-context scenarios, which are essential for the continual advancement of LLMs.
title Large Language Models Can Self-Improve in Long-context Reasoning
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2411.08147