AgentRefine: Enhancing Agent Generalization through Refinement Tuning

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Fu, Dayuan, He, Keqing, Wang, Yejie, Hong, Wentao, Gongque, Zhuoma, Zeng, Weihao, Wang, Wei, Wang, Jingang, Cai, Xunliang, Xu, Weiran
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913704450195456
author Fu, Dayuan
He, Keqing
Wang, Yejie
Hong, Wentao
Gongque, Zhuoma
Zeng, Weihao
Wang, Wei
Wang, Jingang
Cai, Xunliang
Xu, Weiran
author_facet Fu, Dayuan
He, Keqing
Wang, Yejie
Hong, Wentao
Gongque, Zhuoma
Zeng, Weihao
Wang, Wei
Wang, Jingang
Cai, Xunliang
Xu, Weiran
contents Large Language Model (LLM) based agents have proved their ability to perform complex tasks like humans. However, there is still a large gap between open-sourced LLMs and commercial models like the GPT series. In this paper, we focus on improving the agent generalization capabilities of LLMs via instruction tuning. We first observe that the existing agent training corpus exhibits satisfactory results on held-in evaluation sets but fails to generalize to held-out sets. These agent-tuning works face severe formatting errors and are frequently stuck in the same mistake for a long while. We analyze that the poor generalization ability comes from overfitting to several manual agent environments and a lack of adaptation to new situations. They struggle with the wrong action steps and can not learn from the experience but just memorize existing observation-action relations. Inspired by the insight, we propose a novel AgentRefine framework for agent-tuning. The core idea is to enable the model to learn to correct its mistakes via observation in the trajectory. Specifically, we propose an agent synthesis framework to encompass a diverse array of environments and tasks and prompt a strong LLM to refine its error action according to the environment feedback. AgentRefine significantly outperforms state-of-the-art agent-tuning work in terms of generalization ability on diverse agent tasks. It also has better robustness facing perturbation and can generate diversified thought in inference. Our findings establish the correlation between agent generalization and self-refinement and provide a new paradigm for future research.
format Preprint
id arxiv_https___arxiv_org_abs_2501_01702
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle AgentRefine: Enhancing Agent Generalization through Refinement Tuning
Fu, Dayuan
He, Keqing
Wang, Yejie
Hong, Wentao
Gongque, Zhuoma
Zeng, Weihao
Wang, Wei
Wang, Jingang
Cai, Xunliang
Xu, Weiran
Artificial Intelligence
Computation and Language
Robotics
Large Language Model (LLM) based agents have proved their ability to perform complex tasks like humans. However, there is still a large gap between open-sourced LLMs and commercial models like the GPT series. In this paper, we focus on improving the agent generalization capabilities of LLMs via instruction tuning. We first observe that the existing agent training corpus exhibits satisfactory results on held-in evaluation sets but fails to generalize to held-out sets. These agent-tuning works face severe formatting errors and are frequently stuck in the same mistake for a long while. We analyze that the poor generalization ability comes from overfitting to several manual agent environments and a lack of adaptation to new situations. They struggle with the wrong action steps and can not learn from the experience but just memorize existing observation-action relations. Inspired by the insight, we propose a novel AgentRefine framework for agent-tuning. The core idea is to enable the model to learn to correct its mistakes via observation in the trajectory. Specifically, we propose an agent synthesis framework to encompass a diverse array of environments and tasks and prompt a strong LLM to refine its error action according to the environment feedback. AgentRefine significantly outperforms state-of-the-art agent-tuning work in terms of generalization ability on diverse agent tasks. It also has better robustness facing perturbation and can generate diversified thought in inference. Our findings establish the correlation between agent generalization and self-refinement and provide a new paradigm for future research.
title AgentRefine: Enhancing Agent Generalization through Refinement Tuning
topic Artificial Intelligence
Computation and Language
Robotics
url https://arxiv.org/abs/2501.01702