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.
It gives inference-serving teams a principled ceiling on achievable decode-length prediction accuracy and a concrete protocol for evaluating predictors fairly.
Comments & Questions (0)
No comments yet
Be the first to comment!