onóna: twin embeddings with tied unembeddings cut validation loss across scales

Cheng Luo · hf · 2026-11-11

Cheng Luo et al. release onóna, a two-table weight-sharing method for Transformer LMs that preserves input-output weight sharing while boosting output expressiveness: the model averages two embedding tables as input, and the output head combines the softmax distributions defined by both tables, with a gate conditioned only on the current token.

Experiments across several architectures and scales show reduced validation loss and perplexity, at the cost of an extra embedding table and vocabulary projection.

Original post →

More from Research

Research channel →