Beyond Corrected Actions: Training LLM Agents on Textual Critiques
cs.CL, cs.AI
Submitted: 2026-10-14
Updated: 2026-10-14
License: http://creativecommons.org/licenses/by/4.0/
The gist: Large language models (LLMs) are increasingly used for multi-turn agentic tasks that require interactions with users and environments over extended sequences of actions.
Abstract
Large language models (LLMs) are increasingly used for multi-turn agentic tasks that require interactions with users and environments over extended sequences of actions. A common approach to improving LLM agents is to use more capable expert models to provide training supervision, either by behavior cloning the expert's trajectories or by using the expert to correct the actions in trajectories generated by the weaker student model. However, expert LLMs can provide richer supervision than actions alone: they can directly generate textual critique that explain why specific actions are incorrect. We study whether training on such feedback benefits LLM agents and propose Critique-Augmented Agentic Fine-Tuning (CAFT), a framework that elicits textual expert feedback on student-generated trajectories. For each incorrect action, the expert yields both a corrected action and a textual critique explaining the student's mistake--which we then use as the training target for fine-tuning. Across four student models from the Nemotron 3 and Gemma 4 families on the agentic domains τ squared-Bench Airline+Retail and WebShop, adding critiques to action-level supervision improves performance by 8% on average over action-only distillation. Additionally, CAFT checkpoints can also provide a better warm start for reinforcement learning than expert-trajectory distillation. Our results demonstrate that textual critiques can strengthen supervision for multi-turn LLM agents.
Sources
- Gemma 4 Technical Report
- Learning From Mistakes Makes LLM Better Reasoner
- NVIDIA Nemotron 3: Efficient and Open Intelligence
- Training LLM-Based Agents with Synthetic Self-Reflected Trajectories and Partial Masking
- DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning
- Revisiting DAgger in the Era of LLM-Agents
- Multi-Turn On-Policy Distillation with Prefix Replay
- ICRL: Learning to Internalize Self-Critique with Reinforcement Learning
- Agentic Critical Training
- Neglected Free Lunch from Post-training: Progress Advantage for LLM Agents
- Measuring Agents in Production
- CAST: Critique-Aware Supervision for Training Reliable Long-Horizon Tool-Calling Agents
- DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models
- T1-Bench: Benchmarking Multi-Scenario Agents in Real-World Domains
- Agent-R: Training Language Model Agents to Reflect via Iterative Self-Training
Related papers
- Exploring Solution Divergence and Its Effect on Large Language Model Problem Solving
- Ishigaki-IDS-Bench: A Benchmark for Generating Information Delivery Specification from BIM Information Requirements
- Subliminal Steering: Stronger Encoding of Hidden Signals
- MedStruct-S: A Benchmark for Key Discovery, Key-Conditioned QA and Semi-Structured Extraction from OCR Clinical Reports
- The End of Transformers? On Challenging Attention and the Rise of Sub-Quadratic Architectures
- Untangling the Mechanisms of Misleading Context in Medical Question Answering