about
Why do small language models underperform? (arxiv.org)
4 points by tosh on Apr 15, 2024 | hide | past | pdf | 1 comment on HN

In plain words: Small models have too few internal numbers to describe the full spread of possible next words, so their final prediction layer cannot express it. Below 1000 internal numbers, late training collapses the model's representations and performance drops instead of improving.

Abstract · Why do small language models underperform? Studying Language Model Saturation via the Softmax Bottleneck

Recent advances in language modeling consist in pretraining highly parameterized neural networks on extremely large web-mined text corpora. Training and inference with such models can be costly in practice, which incentivizes the use of smaller counterparts. However, it has been observed that smaller models can suffer from saturation, characterized as a drop in performance at some advanced point in training followed by a plateau. In this paper, we find that such saturation can be explained by a mismatch between the hidden dimension of smaller models and the high rank of the target contextual probability distribution. This mismatch affects the performance of the linear prediction head used in such models through the well-known softmax bottleneck phenomenon. We measure the effect of the softmax bottleneck in various settings and find that models based on less than 1000 hidden dimensions tend to adopt degenerate latent representations in late pretraining, which leads to reduced evaluation performance.

Nathan Godey, Éric de la Clergerie, Benoît Sagot
arXiv:2404.07647 · cs.CL · submitted Apr 11, 2024
abstract · pdf · html

add comment on HN

Maybe not surprising, but it’s nice to have a definite, quantified result. Plus it may point the way for future research.