Skip to content

[KDD2025] PyTorch implementation of "Collaboration of Large Language Models and Small Recommendation Models for Device-Cloud Recommendation".

License

Notifications You must be signed in to change notification settings

HelloZicky/LSC4Rec

Repository files navigation

🚀 Collaborative LLMs and Small Recommender Models for Device-Cloud Recommendation (KDD 2025)

Static Badge Static Badge Static Badge Static Badge citation

This is the implementation of the "Collaborative LLMs and Small Recommender Models for Device-Cloud Recommendation"


Citation

@inproceedings{lv2025collaboration,
  author       = {Zheqi Lv and
                  Tianyu Zhan and
                  Wenjie Wang and
                  Xinyu Lin and
                  Shengyu Zhang and
                  Wenqiao Zhang and
                  Jiwei Li and
                  Kun Kuang and
                  Fei Wu},
  title        = {Collaboration of Large Language Models and Small Recommendation Models
                  for Device-Cloud Recommendation},
  booktitle    = {Proceedings of the 31st {ACM} {SIGKDD} Conference on Knowledge Discovery
                  and Data Mining, V.1, {KDD} 2025, Toronto, ON, Canada, August 3-7,
                  2025},
  pages        = {962--973},
  publisher    = {{ACM}},
  year         = {2025},
  url          = {https://doi.org/10.1145/3690624.3709335},
  doi          = {10.1145/3690624.3709335},
}

LLM&SRM Independent Training

LLM

bash train.sh

SRM

bash train_small.sh

LLM&SRM Collaborative Training

bash train_collaboration.sh


SRM Real-time Retraining

bash train_small_retrain.sh


LLM&SRM Collaborative Inference&Request

set ratio

bash colla_test.sh

Random Request

set ratio

python notebooks/test_filter_collaboration_dcc_t2.py

About

[KDD2025] PyTorch implementation of "Collaboration of Large Language Models and Small Recommendation Models for Device-Cloud Recommendation".

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published