EPO: Entropy-regularized Policy Optimization for LLM Agents Reinforcement Learning

Wujiang Xu, Wentian Zhao, Zhenting Wang, Yu-Jhe Li, Can Jin +4 more
9/26/2025
cs.LGcs.CL

No official code found. Best implementation: WujiangXu/EPO (40 stars, Apache-2.0, last commit 2025-10-01).

Abstract

Training LLM agents in multi-turn environments with sparse rewards, where completing a single task requires 30+ turns of interaction within an episode, presents a fundamental challenge for reinforcement learning. We identify a critical failure mode unique to this setting: the exploration-exploitation cascade failure. This cascade begins with early-stage policy premature convergence, where sparse feedback causes agents to commit to flawed, low-entropy strategies. Subsequently, agents enter late-stage policy collapse, where conventional entropy regularization becomes counterproductive, promoting chaotic exploration that destabilizes training. We propose Entropy-regularized Policy Optimization (EPO), a general framework that breaks this failure cycle through three synergistic mechanisms: (1) adopting entropy regularization in multi-turn settings to enhance exploration, (2) an entropy smoothing regularizer that bounds policy entropy within historical averages to prevent abrupt fluctuations, and (3) adaptive phase-based weighting that balances exploration and exploitation across training. Our analysis justifies that EPO guarantees monotonically decreasing entropy variance while maintaining convergence. EPO achieves up to 152% performance improvement on ScienceWorld and up to 19.8% on ALFWorld. Our work demonstrates that multi-turn sparse-reward settings require fundamentally different entropy control than traditional RL, with broad implications for LLM agent training.

View on arXivView PDF

Code Implementations(1)

WujiangXu/EPOCommunity66%
Apache-2.0

The code for paper "EPO: Entropy-regularized Policy Optimization for LLM Agents Reinforcement Learning"

402C, CSSSep 20, 20251 years agoApache-2.0

Similar Papers

Cite this paper

@article{xu2025epo,
  title  = {EPO: Entropy-regularized Policy Optimization for LLM Agents Reinforcement Learning},
  author = {Wujiang Xu and Wentian Zhao and Zhenting Wang and Yu-Jhe Li and Can Jin and Mingyu Jin and Kai Mei and Kun Wan and Dimitris N. Metaxas},
  year   = {2025},
  eprint = {2509.22576},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2509.22576v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2509.22576.svg)](https://codeofpaper.com/paper/2509.22576)

Discussion