Grounded Token Initialization for New Vocabulary in LMs for Generative Recommendation
cs.CL, cs.AI, cs.LG
Submitted: 2026-04-02
Updated: 2026-08-27
License: http://creativecommons.org/licenses/by/4.0/
The gist: Language models (LMs) are increasingly extended with new learnable vocabulary tokens for domain-specific tasks, such as Semantic-ID tokens in generative recommendation.
Terminology
Abstract
Language models (LMs) are increasingly extended with new learnable vocabulary tokens for domain-specific tasks, such as Semantic-ID tokens in generative recommendation. The standard practice initializes these new tokens as the mean of existing vocabulary embeddings, then relies on supervised fine-tuning to learn their representations. We present a systematic analysis of this strategy: through spectral and geometric diagnostics, we show that mean initialization collapses all new tokens into a degenerate subspace, erasing inter-token distinctions that subsequent fine-tuning struggles to fully recover. These findings suggest that token initialization is a key bottleneck when extending LMs with new vocabularies. Motivated by this diagnosis, we propose the Grounded Token Initialization Hypothesis: linguistically grounding novel tokens in the pretrained embedding space before fine-tuning better enables the model to leverage its general-purpose knowledge for novel-token domains. We operationalize this hypothesis as GTI (Grounded Token Initialization), a lightweight grounding stage that, prior to fine-tuning, maps new tokens to distinct, semantically meaningful locations in the pretrained embedding space using only paired linguistic supervision. Despite its simplicity, GTI outperforms both mean initialization and existing auxiliary-task adaptation methods in the majority of evaluation settings across multiple generative recommendation benchmarks, including industry-scale and public datasets. Further analyses show that grounded embeddings produce richer inter-token structure that persists through fine-tuning, corroborating the hypothesis that initialization quality is a key bottleneck in vocabulary extension.
Sources
- Boosting Knowledge Graph-based Recommendations through Confidence-Aware Augmentation with Large Language Models
- OneSearch: A Preliminary Exploration of the Unified End-to-End Generative Framework for E-commerce Search
- A Review of Modern Recommender Systems Using Generative Models (Gen-RecSys)
- OneRec: Unifying Retrieve and Rank with Generative Recommender and Iterative Preference Alignment
- How Well Does Generative Recommendation Generalize?
- Recommendation as Language Processing (RLP): A Unified Pretrain, Personalized Prompt & Predict Paradigm (P5)
- MTGR: Industrial-Scale Generative Recommendation Framework in Meituan
- LightGCN: Simplifying and Powering Graph Convolution Network for Recommendation
- Understanding Dimensional Collapse in Contrastive Self-supervised Learning
- Yo'LLaVA: Your Personalized Language and Vision Assistant
- Recommender Systems with Generative Retrieval
- Neural Discrete Representation Learning
- Learnable Item Tokenization for Generative Recommendation
- LinkedOut: Linking World Knowledge Representation Out of Video LLM for Next-Generation Video Recommendation
- Adapting Large Language Models by Integrating Collaborative Semantics for Recommendation
- OneRec Technical Report
Related papers
- Exploring Solution Divergence and Its Effect on Large Language Model Problem Solving
- Ishigaki-IDS-Bench: A Benchmark for Generating Information Delivery Specification from BIM Information Requirements
- Subliminal Steering: Stronger Encoding of Hidden Signals
- MedStruct-S: A Benchmark for Key Discovery, Key-Conditioned QA and Semi-Structured Extraction from OCR Clinical Reports
- The End of Transformers? On Challenging Attention and the Rise of Sub-Quadratic Architectures
- Untangling the Mechanisms of Misleading Context in Medical Question Answering