RIGHT: Retrieval-augmented Generation for Mainstream Hashtag Recommendation

The source code of "RIGHT: Retrieval-augmented Generation for Mainstream Hashtag Recommendation (ECIR 2024)":

Datasets

Apply and download from:

https://github.com/OpenSUM/HashtagGen

Requerements

torch>=1.10.0
jsonargparse[signatures,urls]>=4.3.1
omegaconf
pytorch-lightning
rich
gpustat
matplotlib
sentencepiece
rouge
nltk
transformers
gensim==3.8.1
simcse
ipdb

Citation

Please cite the paper if it is helpful to you.

@inproceedings{fan2024right,
  title={RIGHT: Retrieval-Augmented Generation for Mainstream Hashtag Recommendation},
  author={Fan, Run-Ze and Fan, Yixing and Chen, Jiangui and Guo, Jiafeng and Zhang, Ruqing and Cheng, Xueqi},
  booktitle={European Conference on Information Retrieval},
  pages={39--55},
  year={2024},
  organization={Springer},
  url={https://link.springer.com/chapter/10.1007/978-3-031-56027-9_3}
}