Writer-R1: Enhancing Generative Writing in LLMs via Memory-augmented Replay Policy Optimization
On the map: LLM alignment / RLHF / preference optimisation → Human values and AI feedback alignment
Abstract (arXiv)
As a typical open-ended generation task, creative writing lacks verifiable reference answers, which has long constrained reward modeling and automatic evaluation due to high human annotation costs, evaluative bias, and coarse feedback signals. To address these challenges, this paper first designs a multi-agent collaborative workflow based on Grounded Theory, performing dimensional decomposition and hierarchical induction of the problem to dynamically produce interpretable and reusable fine-grained criteria. Furthermore, we propose the Memory-augmented Replay Policy Optimization (MRPO) algorithm: on the one hand, without additional training, MRPO guides models to engage in self-reflection based on dynamic criteria, enabling controlled iterative improvement; on the other hand, we adopt the training paradigm that combines supervised fine-tuning with reinforcement learning to convert evaluation criteria into reward signals, achieving end-to-end optimization. Experimental results demonstrate that the automatically constructed criteria achieve performance gains comparable to human annotations. Writer-R1-4B models trained with this approach outperform baselines across multiple creative writing tasks and surpass some 100B+ parameter open-source models.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in Human values and AI feedback alignment, one of 10 directions in LLM alignment / RLHF / preference optimisation. Builds on 16 of its 28 references that are in the map. First posted on arXiv 16 Mar 2026.
Builds on
16 of 28 references.
References · 16
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
- 2023Reflexion: Language Agents with Verbal Reinforcement Learningcited by 4,064
- 2024A Survey on LLM-as-a-Judgecited by 1,384
- 2023RLAIF vs. RLHF: Scaling Reinforcement Learning from Human Feedback with AI Feedbackcited by 615
- 2025Agent Laboratory: Using LLM Agents as Research Assistantscited by 405
- 2017Why We Need New Evaluation Metrics for NLGcited by 232
- 2018Plan-And-Write: Towards Better Automatic Storytellingcited by 232
- 2024LongWriter: Unleashing 10,000+ Word Generation from Long Context LLMscited by 142
- 2023RLTF: Reinforcement Learning from Unit Test Feedbackcited by 110
4 more
- 2025ACECODER: Acing Coder RL via Automated Test-Case Synthesiscited by 89
- 2024StepCoder: Improve Code Generation with Reinforcement Learning from Compiler Feedbackcited by 74
- 2022Of Human Criteria and Automatic Metrics: A Benchmark of the Evaluation of Story Generationcited by 73
- 2024Language Models can Self-Lengthen to Generate Long Textscited by 24
Built on by
Citing papers per year: 2026: 1.
1 paper.
- 2026POLARIS: Guiding Small Models to Write Long Storiescited by 0
- 2026POLARIS: Guiding Small Models to Write Long Stories2 Jun 2026
Similar, not cited
Close in wording, no citation either way
- 2026Generating Constructive Feedback on Stories via Reinforcement Learning
- 2025RLMR: Reinforcement Learning with Mixed Rewards for Creative Writing
- 2025Igniting Creative Writing in Small Language Models: LLM-as-a-Judge versus Multi-Agent Refined Rewards
- 2026UniCreative: Unifying Long-form Logic and Short-form Sparkle via Reference-Free Reinforcement Learning
- 2026From Coarse to Fine: Benchmarking and Reward Modeling for Writing-Centric Generation Tasks
- 2026DPWriter: Reinforcement Learning with Diverse Planning Branching for Creative Writing
- 2026Rewarding Creativity: A Human-Aligned Generative Reward Model for Reinforcement Learning in Storytelling
- 2025LongWriter-Zero: Mastering Ultra-Long Text Generation via Reinforcement Learning