chufangao / TTM-RE

ACL2024: TTM-RE Memory-Augmented Document-Level Relation Extraction
MIT License
9 stars 0 forks source link

TTM-RE

ACL2024: TTM-RE Memory-Augmented Document-Level Relation Extraction

Please see the scripts folder for example run files. Specifically, see run_roberta_rank.sh for the best results. Weights are released for roberta-large in the GitHub releases: https://github.com/chufangao/TTM-RE/releases/tag/v0.1

Please cite:

@article{gao2024ttm,
  title={TTM-RE: Memory-Augmented Document-Level Relation Extraction},
  author={Gao, Chufan and Wang, Xuan and Sun, Jimeng},
  journal={arXiv preprint arXiv:2406.05906},
  year={2024}
}

Much thanks to https://github.com/www-Ye/SSR-PU and https://github.com/wzhouad/ATLOP for making their code open source.