atlas

Research map / Large language models / Language, safety and interpretability

Improving LLMs for Recommendation with Out-Of-Vocabulary Tokens

ICML 2025 · Large language models › Language, safety and interpretability · approach: LLM unlearning and benchmarks · 1 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes META ID, which builds user and item identifiers from clustered meta-path skip-gram embeddings as out-of-vocabulary tokens added to an LLM's vocabulary, improving LLM-based recommendation with T5 and LLaMA2-7b.

Problem

LLM-based recommenders represent users and items with in-vocabulary numeric tokens, which lack expressive power and cause mapping conflicts, weakening recommendation performance after fine-tuning.

Approach

Learn user and item embeddings from meta-path random walks (U-I-U with equal ratings) via a skip-gram model, cluster them with K-Means into hierarchical OOV token groups, add these tokens to the LLM vocabulary, and initialize them with a linear layer from cluster centroids. The paper also introduces memorization and diversity scores to evaluate ID representations.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use META ID instead of CIDT5 sequential recommendation · SportsH@5: 0.0269 -> 0.0322Helped
use META ID instead of SIDT5 sequential recommendation · BeautyH@5: 0.0404 -> 0.0510Helped
use META ID instead of SASRecsequential recommendation · ToysH@5: 0.0463 -> 0.0503Helped
use META ID instead of CIDdirect recommendation · ToysH@5: 0.0076 -> 0.0564Helped
replace K-Means grouping with DBSCAN (ablation)META ID · Beauty sequential recommendationH@5: 0.0510 -> 0.0257Hurt
replace K-Means grouping with Spectral Clustering (ablation)META ID · Beauty sequential recommendationH@5: 0.0510 -> 0.0360Hurt
replace K-Means grouping with RQ-VAE (ablation)META ID · Beauty sequential recommendationH@5: 0.0510 -> 0.0368Hurt
use K-Means grouping rather than alternative grouping methods (ablation)META ID · sequential recommendationH@5Mixed
vary the number of OOV-token clusters (ablation)META ID · sequential recommendationrecommendation performanceMixed
index both users and items instead of only users or only items (ablation)META ID · Beauty sequential recommendationrecommendation performanceHelped
initialize META ID OOV token embeddings with cluster-centroid embedding augmentation instead of random initialization (ablation)META ID · recommendation performancerecommendation performanceHelped
randomly initialize numeric tokens instead of keeping T5's original embeddings (ablation)RID and SID · Sports and Toys recommendationrecommendation performanceHurt

Surprising findings

Limitations

Experimental setup

Datasets: Amazon Sports&Outdoors Amazon Beauty Amazon Toys&Games

Benchmarks: Amazon Sports Amazon Beauty Amazon Toys

Metrics: HR@K NDCG@K RMSE BLEU-1 BLEU-4 memorization score diversity score

Baselines: Caser HGN GRU4Rec BERT4Rec FDSA SASRec S3-Rec CL4SRec P5 with RID P5 with SID P5 with CID

Builds on: P5 RQ-VAE LoRA skip-gram

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy