Learning from Mistakes via Cooperative Study Assistant for Large Language Models
Not on the map. Why?
Abstract (arXiv)
Large language models (LLMs) have demonstrated their potential to refine their generation based on their own feedback. However, the feedback from LLM itself is often inaccurate, thereby limiting its benefits. In this paper, we propose Study Assistant for Large LAnguage Model (SALAM), a novel framework with an auxiliary agent to assist the main LLM in learning from mistakes through interactive cooperation. In the gathering phase, the student assistant agent probes the main LLM, analyzes its errors, and collects the interaction in a mistake memory. During the examination phase, the study assistant provides guidelines by retrieving relevant cases to help the main LLM anticipate and avoid similar errors. We first investigate the effectiveness of a general study assistant and then customize it to provide LLM-specific guidance through imitation learning from successful guidance experiences. Our experiments on three LLMs using two challenging frameworks demonstrate that SALAM can significantly boost LLMs by an accuracy margin of up to 6.6 on BBH and 12.6 on BBQ.
Key facts. Cited by 9 papers in the map (data through 7 Oct 2026). Builds on 32 of its 39 references that are in the map. First posted on arXiv 23 May 2023.
Builds on
32 of 39 references.
References · 32
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2019Sentence-BERT: Sentence Embeddings using Siamese BERT-Networkscited by 9,049
- 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,355
- 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,304
- 2022Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedbackcited by 3,634
- 2023Self-Refine: Iterative Refinement with Self-Feedbackcited by 3,317
- 2022Scaling Instruction-Finetuned Language Modelscited by 2,997
- 2022Constitutional AI: Harmlessness from AI Feedbackcited by 2,950
20 more
- 2020Learning to summarize from human feedbackcited by 2,562
- 2022Beyond the Imitation Game: Quantifying and extrapolating the capabilities of language modelscited by 1,970
- 2022Challenging BIG-Bench Tasks and Whether Chain-of-Thought Can Solve Themcited by 1,767
- 2020Knowledge Distillation: A Surveycited by 1,578
- 2022Solving math word problems with process- and outcome-based feedbackcited by 871
- 2022GPT-NeoX-20B: An Open-Source Autoregressive Language Modelcited by 711
- 2021BBQ: A Hand-Built Bias Benchmark for Question Answeringcited by 683
- 2022Large Language Models Can Self-Improvecited by 656
- 2019Improved Knowledge Distillation via Teacher Assistantcited by 567
- 2023Large Language Models Are State-of-the-Art Evaluators of Translation Qualitycited by 416
- 2022Self-critiquing models for assisting human evaluatorscited by 372
- 2023Check Your Facts and Try Again: Improving Large Language Models with External Knowledge and Automated Feedbackcited by 351
- 2022Teaching Small Language Models to Reasoncited by 279
- 2022Generating Sequences by Learning to Self-Correctcited by 271
- 2021Symbolic Knowledge Distillation: from General Language Models to Commonsense Modelscited by 270
- 2022Quark: Controllable Text Generation with Reinforced Unlearningcited by 236
- 2022Re3: Generating Longer Stories With Recursive Reprompting and Revisioncited by 193
- 2023Training Language Models with Language Feedback at Scalecited by 125
- 2023The Wisdom of Hindsight Makes Language Models Better Instruction Followerscited by 59
- 2022I2D2: Inductive Knowledge Distillation with NeuroLogic and Self-Imitationcited by 21
Built on by
Citing papers per year: 2024: 3; 2025: 2; 2026: 4.
9 papers.
- 2024Recover: A Neuro-Symbolic Framework for Failure Detection and Recoverycited by 15
- 2024Beyond Imitation: Learning Key Reasoning Steps from Dual Chain-of-Thoughts in Reasoning Distillationcited by 13
- 2024TPD: Enhancing Student Language Model Reasoning via Principle Discovery and Guidancecited by 5
- 2026Reward Prediction with Factorized World Statescited by 2
- 2026Learning-Zone Energy: Online Data Selection for Efficient RL Post-Trainingcited by 2
- 2025It Helps to Take a Second Opinion: Teaching Smaller LLMs to Deliberate Mutually via Selective Rationale Optimisationcited by 1
- 2025Error-Driven Scene Editing for 3D Grounding in Large Language Modelscited by 0
- 2026GRPO and Reflection Reward for Mathematical Reasoning in Large Language Modelscited by 0
- 2026Smarter by the Moment: Environment-Driven Dynamic Policies for Continual LLM Improvementcited by 0
- 2026Smarter by the Moment: Environment-Driven Dynamic Policies for Continual LLM Improvement15 Sep 2026
- 2026Learning-Zone Energy: Online Data Selection for Efficient RL Post-Training16 May 2026
- 2026GRPO and Reflection Reward for Mathematical Reasoning in Large Language Models14 Mar 2026
- 2026Reward Prediction with Factorized World States10 Mar 2026
- 2025Error-Driven Scene Editing for 3D Grounding in Large Language Models18 Nov 2025
- 2025It Helps to Take a Second Opinion: Teaching Smaller LLMs to Deliberate Mutually via Selective Rationale Optimisation4 Mar 2025
- 2024Beyond Imitation: Learning Key Reasoning Steps from Dual Chain-of-Thoughts in Reasoning Distillation30 May 2024
- 2024Recover: A Neuro-Symbolic Framework for Failure Detection and Recovery31 Mar 2024
- 2024TPD: Enhancing Student Language Model Reasoning via Principle Discovery and Guidance24 Jan 2024