TRACE: Trajectory Correction from Cross-layer Evidence for Hallucination Reduction
cs.AI, cs.CL
Submitted: 2026-05-18
Updated: 2026-09-05
Comments: 24 pages, 8 figures, 4 tables
License: http://creativecommons.org/licenses/by/4.0/
The gist: Hallucination correction is not a one-direction problem.
Terminology
Abstract
Hallucination correction is not a one-direction problem. We show that intermediate layers are neither uniformly more truthful than final layers nor uniformly less trustworthy. Yet hallucination reduction is usually instantiated through one fixed intervention form: contrast one layer against another, steer along a truthfulness direction, or defer to external evidence. This framing is structurally incomplete. Cross-layer factual evidence does not evolve uniformly: in some failures truthful support is present internally and later suppressed, whereas in others candidate competition remains genuinely multi-directional across depth, so no single signed scalar family is generally sufficient. We introduce Trajectory Correction from Cross-layer Evidence for Hallucination Reduction (TRACE), a deterministic, training-free algorithm which corrects hallucinations at inference time by deriving both the corrective layer and the appropriate correction operator from each input's cross-layer candidate trajectory inside the LLM's own forward pass. Under one frozen hyperparameter setting, TRACE selects among scalar reversal, earlier-state recovery, and candidate-space correction using only model-internal evidence. Evaluated as a single universal algorithm across 15 models, 8 model families, and 3 factuality benchmarks, TRACE improves every evaluation cell, yielding mean gains of +12.26 MC1 points and +8.65 MC2-style points with no regressions, with gains reaching +47.20 MC1 and +43.38 MC2-style points. The method uses no labels, retrieval, pretraining, finetuning, or per-model calibration.
Sources
- Phi-4 Technical Report
- Phi-4-reasoning Technical Report
- Layer Normalization
- DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning
- LLM.int8(): 8-bit Matrix Multiplication for Transformers at Scale
- The Llama 3 Herd of Models
- Gemma 3 Technical Report
- Mistral 7B
- Mixtral of Experts
- Ministral 3
- gpt-oss-120b & gpt-oss-20b Model Card
- PyTorch: An Imperative Style, High-Performance Deep Learning Library
- Qwen3 Technical Report
- LLaMA: Open and Efficient Foundation Language Models
- Adaptive Activation Steering: A Tuning-Free LLM Truthfulness Improvement Method for Diverse Hallucinations Categories
- Root Mean Square Layer Normalization
- PoLLMgraph: Unraveling Hallucinations in Large Language Models via State Transition Dynamics
- Representation Engineering: A Top-Down Approach to AI Transparency
Related papers
- MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
- Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models
- The Clinician's Veto: Navigating Trust, Liability, and Uncertainty in Autonomous AI Prescribing
- MindHelper: Closed-Loop Embodied Mental-State Reasoning for Precision Intervention
- Incumbent Advantage: Brand Bias and Cognitive Manipulation Dynamics in LLM Recommendation Systems
- VSAL: A Vision Solver with Adaptive Layouts for Graph Property Detection