HSTU-BLaIR: Lightweight Contrastive Text Embedding for Generative Recommender

Yijun Liu
4/13/2025
cs.IRcs.LG

No official code found. Best implementation: snapfinger/HSTU-BLaIR (28 stars, Apache-2.0, last commit 2025-07-04).

Abstract

Recent advances in recommender systems have underscored the complementary strengths of generative modeling and pretrained language models. We propose HSTU-BLaIR, a hybrid framework that augments the Hierarchical Sequential Transduction Unit (HSTU)-based generative recommender with BLaIR, a lightweight contrastive text embedding model. This integration enriches item representations with semantic signals from textual metadata while preserving HSTU's powerful sequence modeling capabilities. We evaluate HSTU-BLaIR on two e-commerce datasets: three subsets from the Amazon Reviews 2023 dataset and the Steam dataset. We compare its performance against both the original HSTU-based recommender and a variant augmented with embeddings from OpenAI's state-of-the-art \texttt{text-embedding-3-large} model. Despite the latter being trained on a substantially larger corpus with significantly more parameters, our lightweight BLaIR-enhanced approach -- pretrained on domain-specific data -- achieves better performance in nearly all cases. Specifically, HSTU-BLaIR outperforms the OpenAI embedding-based variant on all but one metric, where it is marginally lower, and matches it on another. These findings highlight the effectiveness of contrastive text embeddings in compute-efficient recommendation settings.

View on arXivView PDF

Code Implementations(1)

Apache-2.0

HSTU-BLaIR: Lightweight Contrastive Text Embedding for Generative Recommender 🌱

282C, C++Mar 23, 20251 years agoApache-2.0
constrastive-learninggenerative-recommenderslightweighttext-embedding

Similar Papers

Cite this paper

@article{liu2025hstublair,
  title  = {HSTU-BLaIR: Lightweight Contrastive Text Embedding for Generative Recommender},
  author = {Yijun Liu},
  year   = {2025},
  eprint = {2504.10545},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2504.10545v3}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2504.10545.svg)](https://codeofpaper.com/paper/2504.10545)

Discussion