TTT4Rec: A Test-Time Training Approach for Rapid Adaption in Sequential Recommendation

Zhaoqi Yang, Yanan Wang, Yong Ge
9/27/2024
cs.IRcs.AI

Official code: ZhaoqiZachYang/TTT4Rec (4 stars, MIT, last commit 2024-10-01).

Abstract

Sequential recommendation tasks, which aim to predict the next item a user will interact with, typically rely on models trained solely on historical data. However, in real-world scenarios, user behavior can fluctuate in the long interaction sequences, and training data may be limited to model this dynamics. To address this, Test-Time Training (TTT) offers a novel approach by using self-supervised learning during inference to dynamically update model parameters. This allows the model to adapt to new user interactions in real-time, leading to more accurate recommendations. In this paper, we propose TTT4Rec, a sequential recommendation framework that integrates TTT to better capture dynamic user behavior. By continuously updating model parameters during inference, TTT4Rec is particularly effective in scenarios where user interaction sequences are long, training data is limited, or user behavior is highly variable. We evaluate TTT4Rec on three widely-used recommendation datasets, demonstrating that it achieves performance on par with or exceeding state-of-the-art models. The codes are available at https://github.com/ZhaoqiZachYang/TTT4Rec.

View on arXivView PDF

Code Implementations(1)

MIT

A Test-Time Training Approach for Rapid Adaption in Sequential Recommendation

41PythonSep 27, 20242 years agoMIT

Similar Papers

Cite this paper

@article{yang2024tttrec,
  title  = {TTT4Rec: A Test-Time Training Approach for Rapid Adaption in Sequential Recommendation},
  author = {Zhaoqi Yang and Yanan Wang and Yong Ge},
  year   = {2024},
  eprint = {2409.19142},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2409.19142v1}
}

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

Discussion