Mitigating Factual Hallucination in Large Reasoning Models via Mixed-Mode Advantage Regularization

· Source: Computation and Language · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Expert, quick

Summary

Large Reasoning Models (LRMs) enhance language model capabilities by generating explicit thinking traces for factuality-oriented question answering (QA). While often improving performance by recovering knowledge, this "thinking" can also cause "thinking-induced hallucination," where correct direct answers are overturned, leading to factual drift. Researchers explain this as a thinking residual that can either recover knowledge or introduce unsupported associations. To address this, they propose MARGO, Mixed-Mode Advantage Regularization for Grounded Optimization. MARGO is a reinforcement learning framework that utilizes non-thinking rollouts as same-model references for advantage estimation. By creating mixed-mode rollout groups with both thinking and non-thinking trajectories, MARGO assesses whether explicit thinking adds factual value beyond direct answering. This approach effectively suppresses hallucination-prone thinking while retaining beneficial reasoning behaviors. Experiments on multiple factuality-oriented QA benchmarks demonstrate MARGO's improved factual reliability over strong baselines, and evaluations on mathematical benchmarks confirm it preserves general reasoning ability.

Key takeaway

For AI Scientists and Machine Learning Engineers developing Large Reasoning Models for factuality-oriented QA, you should consider integrating MARGO to enhance factual reliability. This reinforcement learning framework helps mitigate "thinking-induced hallucination" by evaluating the factual value of explicit thinking against direct answers. It ensures your models maintain accuracy without sacrificing general reasoning capabilities on complex tasks. Implement MARGO to suppress detrimental thinking while preserving beneficial reasoning.

Key insights

Explicit thinking in LRMs can induce factual hallucinations; MARGO mitigates this by regularizing thinking based on non-thinking references.

Principles

Method

MARGO is an RL framework using non-thinking rollouts as same-model references in advantage estimation. It constructs mixed-mode rollout groups to evaluate thinking's factual value, suppressing hallucination-prone thinking.

In practice

Topics

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, NLP Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computation and Language.