REFACT: Adaptive Fact Restatement for Compact and Faithful Chain-of-Thought Reasoning

· Source: Takara TLDR - Daily AI Papers · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Natural Language Processing · Depth: Expert, quick

Summary

REFACT is an adaptive fact-restatement citation framework designed to enhance the faithfulness and compactness of large language models' long-form reasoning. It addresses the issue of reasoning traces drifting from context due to sparse, noisy, or conflicting evidence, a common problem with existing grounding methods that often fail to ensure cited content sufficiency. REFACT trains models to intelligently determine when and at what granularity a reasoning step requires contextual grounding, transforming citations into answer-supporting intermediate states. The framework is optimized via a two-stage SFT-to-RL pipeline, utilizing a citation-utility reward to ensure cited facts are well-formed, source-traceable, and sufficient for the answer. Experiments on LongBench, LV-Eval, and ConFiQA demonstrate REFACT's ability to improve long-context QA and counterfactual faithfulness while significantly reducing token consumption and producing denser, more evidence-rich reasoning traces.

Key takeaway

For Machine Learning Engineers developing LLM applications requiring robust long-form reasoning, REFACT offers a significant advancement. You should consider integrating this adaptive fact-restatement framework to mitigate reasoning drift and improve contextual grounding. This approach enhances both the faithfulness of your models' outputs and reduces token consumption, yielding denser, more reliable reasoning traces for complex tasks like long-context QA and counterfactual reasoning.

Key insights

REFACT adaptively grounds LLM reasoning steps with context, ensuring faithful and compact outputs by integrating citations as intermediate states.

Principles

Method

REFACT employs a two-stage SFT-to-RL pipeline. A citation-utility reward function encourages well-formed, source-traceable, and answer-sufficient facts, guiding models to adaptively restate context for grounding reasoning steps.

In practice

Topics

Code references

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

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Takara TLDR - Daily AI Papers.