When Mean CE Fails: Median CE Can Better Track Language Model Quality
Not on the map. Why?
Abstract (arXiv)
Mean cross-entropy is the standard validation metric for language models, but it can fail to track model quality during training. We examine this in two common scenarios. First, in Qwen2.5-1.5B SFT on synthetic fact-learning, we find that mean CE rises substantially after the initial learning phase while held-out fact-recall accuracy remains near its peak. Second, we find that in top-K distillation on TinyStories, decreasing K improves median CE while worsening mean CE; the Top-5 student attains the highest LLM-judge score and crosses below its teacher on median CE, despite having the worst mean CE. In both cases, median CE correlates much more closely with task performance than does mean CE. Analyzing how bulk and tail percentile CE move during training reveals that training reshapes the empirical per-token CE distribution. In top-K distillation, smaller K yields a distribution with more mass at both extremes, decreasing the median and increasing the mean. In Qwen SFT, the bulk saturates quickly while the tail extends in the latter half of training. In both, the task-evaluation metric appears more sensitive to the bulk than to the tail. Practically, we recommend reporting a small set of percentile CE summaries alongside the mean, and using concordance among them as a tool to keep track of distribution reshaping, as well as a low-cost diagnostic for when mean and median CE disagree on model selection.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 17 of its 20 references that are in the map. First posted on arXiv 23 May 2026.
Builds on
17 of 20 references.
References · 17
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2023Judging LLM-as-a-Judge with MT-Bench and Chatbot Arenacited by 8,626
- 2024Qwen2.5 Technical Reportcited by 4,942
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
- 2019The Curious Case of Neural Text Degenerationcited by 2,916
- 2019TinyBERT: Distilling BERT for Natural Language Understandingcited by 1,167
- 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,160
- 2018Generalized Cross Entropy Loss for Training Deep Neural Networks with Noisy Labelscited by 997
- 2023Retentive Network: A Successor to Transformer for Large Language Modelscited by 613
- 2021MAUVE: Measuring the Gap Between Neural Text and Human Text using Divergence Frontierscited by 470
- 2023TinyStories: How Small Can Language Models Be and Still Speak Coherent English?cited by 449
- 2019Mean estimation and regression under heavy-tailed distributions--a surveycited by 136
5 more
- 2024Observational Scaling Laws and the Predictability of Language Model Performancecited by 130
- 2022Locally Typical Samplingcited by 111
- 2015The Goldilocks Principle: Reading Children's Books with Explicit Memory Representationscited by 108
- 2024What is Wrong with Perplexity for Long-context Language Modeling?cited by 69
- 2026Don't Ignore the Tail: Decoupling top-K Probabilities for Efficient Language Model Distillationcited by 7
Built on by
Not cited yet.