https://github.com/chengkai-liu/Mamba4Rec

[RelKD'24] Mamba4Rec: Towards Efficient Sequential Recommendation with Selective State Space Models
https://github.com/chengkai-liu/Mamba4Rec

Last synced: 3 months ago
JSON representation

[RelKD'24] Mamba4Rec: Towards Efficient Sequential Recommendation with Selective State Space Models

• Host: GitHub
• URL: https://github.com/chengkai-liu/Mamba4Rec
• Owner: chengkai-liu
• License: mit
• Created: 2024-03-07T04:57:58.000Z (11 months ago)
• Default Branch: main
• Last Pushed: 2024-08-24T15:10:25.000Z (5 months ago)
• Last Synced: 2024-08-24T16:27:37.831Z (5 months ago)
• Language: Python
• Homepage: https://arxiv.org/abs/2403.03900
• Size: 5.47 MB
• Stars: 72
• Watchers: 1
• Forks: 2
• Open Issues: 2
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

• Awesome-state-space-models - Recommendation

README

        
# Mamba4Rec

> **Mamba4Rec: Towards Efficient Sequential Recommendation with Selective State Space Models (RelKD@KDD 2024 Best Paper Award Award)**\

> Chengkai Liu, Jianghao Lin, Jianling Wang, Hanzhou Liu, James Caverlee\

> Paper: https://arxiv.org/abs/2403.03900

## Usage

### Requirements

* Python 3.7+

* PyTorch 1.12+

* CUDA 11.6+

* Install RecBole:

  * `pip install recbole`

* Install causal Conv1d and the core Mamba package:

  * `pip install causal-conv1d>=1.2.0`

  * `pip install mamba-ssm`

You can also refer to the required environment specifications in `environment.yaml`.

### Run

```python run.py```

Specifying the dataset in `config.yaml` will trigger an automatic download. Please set an appropriate maximum sequence length in `config.yaml` for each dataset before training.

## Citation

```bibtex

@article{liu2024mamba4rec,

  title={Mamba4rec: Towards efficient sequential recommendation with selective state space models},

  author={Liu, Chengkai and Lin, Jianghao and Wang, Jianling and Liu, Hanzhou and Caverlee, James},

  journal={arXiv preprint arXiv:2403.03900},

  year={2024}

}

```

## Acknowledgment

This project is based on [Mamba](https://github.com/state-spaces/mamba), [Causal-Conv1d](https://github.com/Dao-AILab/causal-conv1d), and [RecBole](https://github.com/RUCAIBox/RecBole). Thanks for their excellent works.