GRACE: Generative Recommendation via Journey-Aware Sparse Attention on Chain-of-Thought Tokenization

Luyi Ma, Wanjia Zhang, Kai Zhao, Abhishek Kulkarni, Lalitesh Morishetti +13 more
7/19/2025
cs.CLcs.AIcs.IR

No confident code implementation found for this paper.

Abstract

Generative models have recently demonstrated strong potential in multi-behavior recommendation systems, leveraging the expressive power of transformers and tokenization to generate personalized item sequences. However, their adoption is hindered by (1) the lack of explicit information for token reasoning, (2) high computational costs due to quadratic attention complexity and dense sequence representations after tokenization, and (3) limited multi-scale modeling over user history. In this work, we propose GRACE (Generative Recommendation via journey-aware sparse Attention on Chain-of-thought tokEnization), a novel generative framework for multi-behavior sequential recommendation. GRACE introduces a hybrid Chain-of-Thought (CoT) tokenization method that encodes user-item interactions with explicit attributes from product knowledge graphs (e.g., category, brand, price) over semantic tokenization, enabling interpretable and behavior-aligned generation. To address the inefficiency of standard attention, we design a Journey-Aware Sparse Attention (JSA) mechanism, which selectively attends to compressed, intra-, inter-, and current-context segments in the tokenized sequence. Experiments on two real-world datasets show that GRACE significantly outperforms state-of-the-art baselines, achieving up to +106.9% HR@10 and +106.7% NDCG@10 improvement over the state-of-the-art baseline on the Home domain, and +22.1% HR@10 on the Electronics domain. GRACE also reduces attention computation by up to 48% with long sequences.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{ma2025grace,
  title  = {GRACE: Generative Recommendation via Journey-Aware Sparse Attention on Chain-of-Thought Tokenization},
  author = {Luyi Ma and Wanjia Zhang and Kai Zhao and Abhishek Kulkarni and Lalitesh Morishetti and Anjana Ganesh and Ashish Ranjan and Aashika Padmanabhan and Jianpeng Xu and Jason Cho and Praveen Kanumala and Kaushiki Nag and Sumit Dutta and Kamiya Motwani and Malay Patel and Evren Korpeoglu and Sushant Kumar and Kannan Achan},
  year   = {2025},
  eprint = {2507.14758},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2507.14758v1}
}

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/2507.14758.svg)](https://codeofpaper.com/paper/2507.14758)

Discussion