Position: Understanding LLMs Requires More Than Statistical Generalization

Patrik Reizinger, Szilvia Ujváry, Anna Mészáros, Anna Kerekes, Wieland Brendel +1 more
5/3/2024
stat.MLcs.LG

No official code found. Best implementation: rpatrik96/llm-non-identifiability (1 star, MIT, PyTorch, last commit 2024-05-19).

Abstract

The last decade has seen blossoming research in deep learning theory attempting to answer, "Why does deep learning generalize?" A powerful shift in perspective precipitated this progress: the study of overparametrized models in the interpolation regime. In this paper, we argue that another perspective shift is due, since some of the desirable qualities of LLMs are not a consequence of good statistical generalization and require a separate theoretical explanation. Our core argument relies on the observation that AR probabilistic models are inherently non-identifiable: models zero or near-zero KL divergence apart -- thus, equivalent test loss -- can exhibit markedly different behaviors. We support our position with mathematical examples and empirical observations, illustrating why non-identifiability has practical relevance through three case studies: (1) the non-identifiability of zero-shot rule extrapolation; (2) the approximate non-identifiability of in-context learning; and (3) the non-identifiability of fine-tunability. We review promising research directions focusing on LLM-relevant generalization measures, transferability, and inductive biases.

View on arXivView PDF

Code Implementations(1)

PyTorchMIT

Investigating the non-identifiability of Transformers

10Python, Jupyter NotebookDec 25, 20232 years agoMIT
llmllm-trainingmachine-learningnlp-machine-learningpytorch+3 more

Similar Papers

Cite this paper

@article{reizinger2024position,
  title  = {Position: Understanding LLMs Requires More Than Statistical Generalization},
  author = {Patrik Reizinger and Szilvia Ujváry and Anna Mészáros and Anna Kerekes and Wieland Brendel and Ferenc Huszár},
  year   = {2024},
  eprint = {2405.01964},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2405.01964v3},
  journal = {ICML 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2405.01964.svg)](https://codeofpaper.com/paper/2405.01964)

Discussion