Back to AI News
arxiv

Study Derives Fundamental Limits of Prefill-Based Decode-Length Prediction

A new arXiv paper argues that existing benchmarks for predicting how many tokens an LLM will decode from prefill-phase hidden states conflate predictor quality with task difficulty, causing reported accuracy to collapse when the model, dataset or sampling configuration changes. The authors trace this instability to the inherent stochasticity of autoregressive decoding and model decoding as a stopped stochastic process whose per-step stopping probability ramps up over hundreds of steps, yielding per-prompt normality and an affine variance law sigma = alpha*mu + b. Empirically the normal law reproduces per-prompt MAE and the central mass of the decode-length distribution to within a few percent, with residual skew and kurtosis matching the predicted sampling-noise correction. Across MLP, EGTP, FiLM, soft-label and Mixture-of-Experts predictors, performance converges near the derived bound under fixed conditions, which the authors read as system error dominating model error, and they propose a benchmarking protocol that fixes model, data and sampling parameters and reports relative alongside absolute error.

AI Brief

It gives inference-serving teams a principled ceiling on achievable decode-length prediction accuracy and a concrete protocol for evaluating predictors fairly.

arXiv:2610.40541v1 Announce Type: new Abstract: Predicting how many tokens a large language model (LLM) will decode, using only prefill-phase hidden states, is a key primitive for inference scheduling. We show that current benchmarks conflate predictor quality with task hardness: when the target model, dataset, or sampling configuration changes, reported accuracy collapses even though the predictor is unchanged. We trace this instability to an overlooked error source---the inherent stochasticity of autoregressive decoding---and derive bounds that every prefill-side method must obey. Our analysis treats decoding as a stopped stochastic process. The per-step stopping probability of a response is not constant but ramps up over hundreds of steps, so the decode length inherits a central limit theorem from the many local decisions that determine where a response ends; this yields per-prompt normality with an explicit error term, together with an affine variance law $\sigma=\alpha\mu+b$ whose slope is the coefficient of variation of the number of response segments. Empirically, the normal law reproduces the per-prompt MAE and the central mass of the decode-length distribution to within a few percent, and the residual departures from normality (mild right skew, slight excess kurtosis) are the ones predicted by the sampling-noise correction. Under fixed model, data, and sampling conditions, a broad design space of predictors (MLP, EGTP, FiLM, soft-label, and Mixture-of-Experts variants) converges to similar performance close to the bound, confirming that \emph{system error dominates model error}. We argue for a benchmarking protocol that fixes the model, data, and sampling parameters and that reports relative-error metrics alongside absolute ones. Code and analysis tools are available at https://github.com/ShwStone/DecodingPredict.
Be the first to rate this article.

Comments & Questions (0)

Captcha
Please be respectful — let's keep the conversation friendly.

No comments yet

Be the first to comment!