Steven Gonsalvez

Software Engineer

Dispersion loss counteracts embedding condensation in small language models

Why CEREBRO kept it

LLM embedding behavior deep-dive, model mechanics.

The text below is an automated extraction of the article at https://chenliu-1996.github.io/projects/LM-Dispersion/, stored verbatim in the public cerebro-vault repository. Copyright remains with the original publisher (chenliu-1996.github.io).

More severe in smaller models than in larger counterparts (Figure 2). What makes LLMs better than small LMs? Data? Parameters? Geometry might play a role! What makes LLMs better than small LMs? Data? Parameters? Geometry might play a role! Every Transformer layer of a language model represents each input token as a vector in a high-dimensional embedding space. We notice that as those vectors progress through Transformer layers, they often behave as if they were confined to a narrow cone: they point to increasingly similar directions as measured by pairwise cosine similarity. We call this geome

Community take

Embedding anisotropy and representation collapse have been known transformer geometry issues since at least 2017–19, not novel findings.

Backlinks

Appeared in 1 briefing

Related

Shares tags: ai/llm-mechanics

Also from chenliu-1996.github.io

Only signal from chenliu-1996.github.io so far.