AMARIS: A Memory-Augmented Rubric Improvement System for Rubric-Based Reinforcement Learning

Peilin Wu, Xinlu Zhang, Kun Wan, Wentian Zhao, Gang Wu, Xinya Du, Zhiyu Chen


Abstract
Rubric-based reward shaping provides interpretable and editable reward signals for fine-tuning LLMs via reinforcement learning (RL), but existing adaptive rubric methods typically update criteria from local evidence such as the current batch or instance-level comparisons. This local view discards diagnostic information produced during training, making it difficult to track recurring failures, evaluate previous rubric edits, or raise standards once earlier criteria become saturated. We introduce AMARIS, a Memory-Augmented Rubric Improvement System that grounds rubric updates in longitudinal training evidence. AMARIS stores rollout analyses, step-level summaries, and rubric update records in a persistent evaluation memory, then retrieves recent and semantically relevant history to revise rubrics. We evaluate AMARIS across science, medicine, instruction following, and creative writing under both global and instance-specific rubric settings, with policy models from the Qwen and Llama families trained by GRPO and PPO. AMARIS improves over static, local-adaptive, and memory-ablated baselines, such as +2.8 points on GPQA-Diamond and +2.2 points on IFBench over the strongest baselines, while analysis shows that memory reduces oscillatory rubric edits and supports a progression from early failure correction to later curriculum advancement. The final rubric set that AMARIS discovers is itself reusable, and the online trajectory contributes measurably beyond it. AMARIS runs asynchronously alongside the normal RL loop, reducing blocking latency relative to synchronous rubric updates. The source code is available at https://github.com/qualidea1217/AMARIS.
Anthology ID:
2026.findings-emnlp.2152
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
45981–46052
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.2152/
DOI:
Bibkey:
Cite (ACL):
Peilin Wu, Xinlu Zhang, Kun Wan, Wentian Zhao, Gang Wu, Xinya Du, and Zhiyu Chen. 2026. AMARIS: A Memory-Augmented Rubric Improvement System for Rubric-Based Reinforcement Learning. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 45981–46052, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
AMARIS: A Memory-Augmented Rubric Improvement System for Rubric-Based Reinforcement Learning (Wu et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.2152.pdf