English

RewardFlow: Topology-Aware Reward Propagation on State Graphs for Agentic RL with Large Language Models

Artificial Intelligence 2026-05-29 v2 Computation and Language Machine Learning

Abstract

Reinforcement learning (RL) shows promise for enhancing LLM agentic reasoning, yet sparse terminal rewards hinder fine-grained optimization. Process reward modeling offers an alternative but incurs high computational costs, reward hacking risks, and annotation bottlenecks. We introduce RewardFlow, a lightweight method for estimating state-level rewards in agentic reasoning. By constructing state graphs that capture the intrinsic topological structure of trajectories, RewardFlow performs topology-aware propagation to estimate each state's contribution to success, yielding principled, annotation-free dense rewards. Used for RL optimization, RewardFlow substantially outperforms prior baselines across four agentic benchmarks: +6.2% average success rate on text-based tasks, +29.7% on visual reasoning over the strongest baseline across three model scales, and +10% accuracy on DeepResearch, with superior robustness and training efficiency. The implementation of RewardFlow is publicly available at https://github.com/tmlr-group/RewardFlow.

Keywords

Cite

@article{arxiv.2603.18859,
  title  = {RewardFlow: Topology-Aware Reward Propagation on State Graphs for Agentic RL with Large Language Models},
  author = {Xiao Feng and Bo Han and Zhanke Zhou and Jiaqi Fan and Jiangchao Yao and Ka Ho Li and Dahai Yu and Michael Kwok-Po Ng},
  journal= {arXiv preprint arXiv:2603.18859},
  year   = {2026}
}