cs.LGOct 10, 2026

The Optimal Decoder is Tilted Temperature

Authors: Nikolaos Papagiannis, Ananth Grama, Wojciech Szpankowski

Organizations: Purdue University

Abstract

Larger sample budgets benefit from higher sampling temperatures in LLMs, but what temperature tuning optimizes and why it works remain poorly understood. We seek to answer these questions: we pose decoding as pass@kk maximization subject to two constraints that encode critical decoding failure modes: an entropy floor preventing diversity collapse, and a cross-entropy ceiling preventing implausibility. We prove that this maximization takes the form of a single closed-form conditional sampling rule, the tilted temperature q⋆∝(peηV)1/Tq^\star \propto \bigl(p e^{ηV}\bigr)^{1/T}. Here, the tilt comes from a verifier potential VV that quantifies how likely a partial generation is to eventually pass. Ordinary temperature sampling, verifier-greedy decoding, and reward-tilted policies represent special cases. We further show that, under a natural rule for scaling the diversity budget with sample size kk, the resulting optimal temperature increases with kk, turning empirical choice into a theorem. The optimum still needs the unknown verifier potential, which we estimate using two design approaches. In the online setting, our J-EOS algorithm jointly tracks the two constraint multipliers and learns the verifier potential from binary pass/ fail feedback alone. In the offline setting, we show that when the model is calibrated, its log-probability tracks correctness up to an affine map. Under this condition, the tilt collapses into the exponent and the optimum reduces to ordinary temperature sampling. Calibrated Temperature Decoding (CTD) then fits one scalar offline and samples with zero additional inference-time overhead relative to temperature sampling. Building on these decision rules, we design a practical decoding algorithm for code generation and evaluate it alongside CTD on benchmarks including HumanEval and MBPP, showing improvements over standard and adaptive decoding baselines.

Explore similar work

CardsList
  1. Decoding as Optimisation on the Probability Simplex: From Top-K to Top-P (Nucleus) to Best-of-K Samplers

    Feb 20, 2026Xiaotong Ji, Rasul Tutunov, Matthieu Zimmer +1Language Model DecodingBest-of-N Sampling

  2. Min-kk Sampling: Decoupling Truncation from Temperature Scaling via Relative Logit Dynamics

    Apr 13, 2026Yuanhao Ding, Meimingwei Li, Esteban Garces Arias +3Language Model DecodingAdaptive Inference