Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Learnable Item Tokenization for Generative Recommendation

About

Utilizing powerful Large Language Models (LLMs) for generative recommendation has attracted much attention. Nevertheless, a crucial challenge is transforming recommendation data into the language space of LLMs through effective item tokenization. Current approaches, such as ID, textual, and codebook-based identifiers, exhibit shortcomings in encoding semantic information, incorporating collaborative signals, or handling code assignment bias. To address these limitations, we propose LETTER (a LEarnable Tokenizer for generaTivE Recommendation), which integrates hierarchical semantics, collaborative signals, and code assignment diversity to satisfy the essential requirements of identifiers. LETTER incorporates Residual Quantized VAE for semantic regularization, a contrastive alignment loss for collaborative regularization, and a diversity loss to mitigate code assignment bias. We instantiate LETTER on two models and propose a ranking-guided generation loss to augment their ranking ability theoretically. Experiments on three datasets validate the superiority of LETTER, advancing the state-of-the-art in the field of LLM-based generative recommendation.

Wenjie Wang, Honghui Bao, Xinyu Lin, Jizhi Zhang, Yongqi Li, Fuli Feng, See-Kiong Ng, Tat-Seng Chua• 2024

Related benchmarks

TaskDatasetResultRank
Sequential RecommendationAmazon Beauty (test)
NDCG@103.4
107
Sequential RecommendationSports
Recall@103.91
62
Sequential RecommendationSports
Recall@50.0141
43
Sequential RecommendationBeauty
Recall@106.16
42
RecommendationBeauty
Recall@106.72
39
RecommendationQilin
Hit Rate@50.2116
30
Generative RecommendationBeauty
R@106.72
28
Sequential RecommendationBeauty
Recall@54.44
24
Sequential RecommendationAmazon Toys (test)
NDCG@103.08
24
Generative RecommendationToys
Recall@100.0673
23
Showing 10 of 31 rows

Other info

Follow for update