Skip to content

Cross-modal Reinforced Prompting for Graph and Language Tasks, KDD 2024.

License

Notifications You must be signed in to change notification settings

JohnJiang12138/CMRP

Repository files navigation


⭐️CMRP: Cross-modal Reinforced Prompting for Graph and Language Tasks⭐️

Quick Start

Package Dependencies

  • Python 3.10.13
  • PyTorch 2.0.1

for detailed information, please see requirements.txt

git clone https://github.com/JohnJiang12138/CMRP.git
cd CMRP
conda create -n CMRP python=3.10.13
conda activate CMRP
pip install -r requirements.txt

Preparations

During preprocessing, we use T5-Small for embedding.

git lfs install
git clone https://huggingface.co/google-t5/t5-small

Please replace the api_key and the requests detail with your customization. In fileLLM_tasks/t1step2_llm_evaluate.py, line 9 and line 73.

Preprocessing

Here we preprocess the datasets, prepare embeddings and calculate similarities.

cd ./preprocess
sh preprocess_WN18.sh

Start Training

cd ../REINFORCE
sh run_kg_WN18.sh

🌹Please Cite Our Work If Helpful:

Thanks! / 谢谢! / ありがとう! / merci! / 감사! / Danke! / спасибо! / gracias! ...

@inproceedings{jiang2024killing,
  title={Killing Two Birds with One Stone: Cross-modal Reinforced Prompting for Graph and Language Tasks},
  author={Jiang, Wenyuan and Wu, Wenwei and Zhang, Le and Yuan, Zixuan and Xiang, Jian and Zhou, Jingbo and Xiong, Hui},
  booktitle={Proceedings of the 30th ACM SIGKDD Conference on Knowledge Discovery and Data Mining},
  pages={1301--1312},
  year={2024}
}

About

Cross-modal Reinforced Prompting for Graph and Language Tasks, KDD 2024.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published