CF-KAN is a recommendation system (collaborative filtering) based on the Kolmogorov-Arnold Network (KAN, https://arxiv.org/abs/2404.19756) approach. This project leverages and explores the power of KAN for CF-based recommendation.
(For KAN implementation, I've refered to this repo: https://github.com/Blealtan/efficient-kan)
To get started with CF-KAN, clone the repository and install the required dependencies (pytorch and scipy, etc)
If you are ready, test it with the notebook file (CF-KAN.ipynb)
ML1M
- Recall@20: 0.1791
- NDCG@20: 0.1117
Yelp
- Recall@20: 0.0710
- NDCG@20: 0.0353
We will keep update this project.
If you found this project is helpful for your research, please STAR this repository
If our work was helpful for your project, cite our work :)
** Once the paper is uploaded to arXiv soon, we will provide instructions on how to cite it.