Skip to content

Breaking the Likelihood Trap: Variance-Calibrated Modulation for Large Language Model Decoding

Jun 2026 · arXiv.org · Vol abs/2606.22511 · 2 citations · 58 references
Computer Science Mathematics

TL;DR

In experiments across open-ended generation, factual QA, and mathematical reasoning, it is shown that VCM consistently mitigates the likelihood trap, and integrates with existing decoding strategies, improving diversity and coherence and, particularly at higher decoding temperatures, reasoning accuracy.

Abstract

In open-ended generation, LLMs frequently fall into the"likelihood trap", characterized by repetitive degeneration and vocabulary dullness, resulting in a discrepancy between machine-generated and human-written text. While post-hoc tail truncation (e.g., Top-p, Min-p) avoids sampling from the unreliable tail, it can misalign generation with human lexical preferences by over-sampling from the uncalibrated head; fixed scalar repetition penalties, in turn, ignore how the scale of the logit distribution varies across inference steps, which can disrupt semantic coherence. To address both shortcomings, we propose Variance-Calibrated Modulation (VCM), a training-free pre-decoding intervention. VCM directly reshapes the probability distribution prior to truncation via two dynamic mechanisms: (1) Contextual Searchlight via PMI, which naturally suppresses global stopwords and elevates context-evoked tokens, and (2) Adaptive Self-Debiasing, which utilizes real-time logit standard deviation to provide scale-invariant penalization. In experiments across open-ended generation, factual QA, and mathematical reasoning, we show that VCM consistently mitigates the likelihood trap. With negligible computational overhead, VCM integrates with existing decoding strategies, improving diversity and coherence and, particularly at higher decoding temperatures, reasoning accuracy. Our code is publicly available on GitHub: https://github.com/AetherDing/VCM

View source

Similar papers

Preprint Aug 2026

XTC: Head-Aware Sampling by Excluding Top Choices

This work introduces XTC (Exclude Top Choices), a lightweight head-aware decoding operator that targets this regime of open-ended generation in which several continuations are plausible but too much probability mass remains concentrated on the most generic choice.

Philippe Weidmann, Allen G. Roush, Judah Goldfeder et al. · 0 citations
#artificial intelligence Preprint Sep 2026

Greedy Decoding Is Not Precision-Invariant: Cross-Precision Output Divergence in LLM Inference

An empirical error-propagation analysis is developed and finds that 22 layers of accumulated body error do not distinguish flipping from non-flipping steps; the outcome depends primarily on the top-two logit margin at the LM head relative to the directional perturbation between the top-two candidates.

Gao-Yuan Du, A. Khan, Rex Zhou et al. · 1 citation
#natural language process... Preprint Oct 2026

HeadEdit: Calibrating Language Model Behavior Through the Frozen Unembedding Matrix

Alignment does not eliminate behavioral errors in language models. Models may still refuse benign requests, call unnecessary tools, or yield to false user claims. Current methods mitigate such errors as a computation problem, and rarely explore if the desired behavior is already encoded in the model's representation. M...

Zirui He, Hai-Yan Zhao, Jing-Yu Hu et al. · 0 citations
#machine learning Preprint Sep 2026

Fork-dLLM: Avoiding the Flexibility Trap in Diffusion Language Models

Masked diffusion language models (dLLMs) have shown strong potential for faster inference through parallel token generation when combined with confidence-based samplers. However, recent work has shown that such methods can defer unmasking high-entropy fork positions at which multiple plausible continuations exist. This...

S. Frkovic, Metod Jazbec, C. Naesseth · 0 citations
#natural language process... Preprint Sep 2026

To Copy or Not to Copy: Controlling Speculative Decoding via Intrinsic Model Signals

SwitchSD is introduced, an adaptive framework that treats copying as a latent control signal of the LLM that allows the system to dynamically switch between neural drafting and context-based copying, effectively turning copying from a noisy heuristic into a principled, model-aware decoding regime.

Roy Eisenstadt, Ido Cohen, Edo Cohen-Karlik et al. · 1 citation
#artificial intelligence Preprint Sep 2026

FLIP: Final Layer Inference-Time Probing for Vision-Language Models

FLIP is a validation step for intervention-based mechanistic interpretability, not a steering method, for testing whether a logit-facing intervention site in an open-weight vision-language model (VLM) supports structured, task-linked computation rather than generic perturbation.

Drandreb Earl O. Juanico, Rowel O. Atienza · 0 citations

Related blog posts

Microsoft Research Blog Oct 22, 2026

Flint-interactive: A language for creating interactive data visualizations

Flint-interactive gives AI agents a language for creating interactive visualizations. Click a chart, filter data, highlight related content, and explore beyond the view the author originally chose. The post Flint-interactive: A language for creating interactive data visualizations appeared first on Microsoft Research.

Microsoft Research Blog Oct 15, 2026

Proof-based general-purpose computing with Magnetar

Zero-knowledge proofs can verify computing results without rerunning workloads or exposing sensitive data. Magnetar helps bring them into the mainstream with support for standard programming languages, major speed gains, and quantum-resistant security. The post Proof-based general-purpose computing with Magnetar appeared first on Microsoft Research.

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.