arXiv Machine Learning
3d ago

RubricEM: Meta-RL with Rubric-guided Policy Decomposition beyond Verifiable Rewards

RubricEM is a unified reinforcement learning framework designed for deep research agents that plan, search, evaluate evidence, and synthesize long‑form reports. It introduces rubric‑guided policy decomposition, combining supervised fine‑tuning to scaffold agents with rubrics, stage‑structured GRPO for fine‑grained credit assignment, and a reflection meta‑policy for experience reuse. The resulting RubricEM‑8B model outperforms larger open models on four long‑form research benchmarks and approaches the performance of proprietary deep‑research systems.

By Gaotang Li, Bhavana Dalvi Mishra, Zifeng Wang, Jun Yan, Yanfei Chen, Chun-Liang Li, Long T. Le, Rujun Han, George Lee, Hanghang Tong, Chen-Yu Lee, Tomas Pfister
arXiv AI
6d ago

All-in-One Memory: Process-Rewarded Reinforcement Learning for Memory Agents

All-in-One Memory (AiO-Mem) is a reinforcement learning framework that unifies memory retrieval, writing, and stopping decisions into a single reward space. It compiles diverse histories into traceable multi‑view events, uses a unified policy actor to generate interaction trajectories, and employs a trajectory‑conditioned process reward model trained with supervised fine‑tuning and DPO. Experiments on long‑context memory benchmarks show AiO‑Mem consistently improves answer accuracy compared to static memory systems and other RL‑trained agents.

By Bingbing Wang, Jingjie Lin, Weiming Qiao, Jing Li, Ruifeng Xu
arXiv AI
Jul 16

Memory as a Controlled Process: Learned Adaptive Memory Management for LLM Agents

arXiv:2607. 13591v1 Announce Type: cross Abstract: Large Language Model (LLM) agents increasingly rely on external memory systems to accumulate experience across tasks.

By Eric Hanchen Jiang, Zhi Zhang, Yuchen Wu, Levina Li, Dong Liu, Xiao Liang, Rui Sun, Yubei Li, Edward Sun, Haozheng Luo, Zhaolu Kang, Aylin Caliskan, Kai-Wei Chang, Ying Nian Wu