Grounded Token Initialization for New Vocabulary in LMs for Generative Recommendation
Organizations: University of Wisconsin-Madison · LinkedIn Corporation · Northeastern University · University of California, Davis
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.
Figures & tables
| Methodology | NDCG@K (Composite) | ||||
| @5 | @10 | @20 | @50 | @100 | |
| MI +Vanilla SFT ( Baseline ) | 0.00% | 0.00% | 0.00% | 0.00% | 0.00% |
| MI +Multi-task SFT ( LC-Rec ) | +6.94% | +4.38% | +1.94% | +1.95% | +1.01% |
| GTI +Multi-task SFT ( Ours ) | +17.88% | +12.03% | +6.90% | +4.99% | +2.89% |
| GTI : extra gain over LC-Rec ( ) | +10.94% | +7.65% | +4.96% | +3.04% | +1.88% |