Continuous-Time Sequential Recommendation with State Space Models

Wei Xiao*, Huiying Wang*, Qifeng Zhou, Qing Wang

Paper

Requirements

s5-pytorch
mamba-ssm
causal-conv1d
recbole 1.0
torch

Usage

Firstly, replacing recbole.data.sequential_dataset.py with SS4Rec/sequential_dataset.py

python run.py

Other Related Projects

The code repository references [RecBole]https://github.com/RUCAIBox/RecBole and [Mamba4Rec]https://github.com/chengkai-liu/Mamba4Rec. Thanks a lot for their work!

If you find our work helpful, please kindly cite us

@article{xiao2025ss4rec,
  title={SS4Rec: Continuous-Time Sequential Recommendation with State Space Models},
  author={Xiao, Wei and Wang, Huiying and Zhou, Qifeng and Wang, Qing},
  journal={arXiv preprint arXiv:2502.08132},
  year={2025}
}