Proceedings

Recall, Retrieve and Reason: Towards Better In-Context Relation Extraction

Download BibTeX Cite

Proceedings of the Thirty-Third International Joint Conference on Artificial Intelligence. Main Track. Pages 6368-6376.

https://doi.org/10.24963/ijcai.2024/704

Abstract

Relation extraction (RE) aims to identify relations between entities mentioned in texts. Although large language models (LLMs) have demonstrated impressive in-context learning (ICL) abilities in various tasks, they still suffer from poor performances compared to most supervised fine-tuned RE methods. Utilizing ICL for RE with LLMs encounters two challenges: (1) retrieving good demonstrations from training examples, and (2) enabling LLMs exhibit strong ICL abilities in RE. On the one hand, retrieving good demonstrations is a non-trivial process in RE, which easily results in low relevance regarding entities and relations. On the other hand, ICL with an LLM achieves poor performance in RE while RE is different from language modeling in nature or the LLM is not large enough. In this work, we propose a novel recall-retrieve-reason RE framework that synergizes LLMs with retrieval corpora (training examples) to enable relevant retrieving and reliable in-context reasoning. Specifically, we distill the consistently ontological knowledge from training datasets to let LLMs generate relevant entity pairs grounded by retrieval corpora as valid queries. These entity pairs are then used to retrieve relevant training examples from the retrieval corpora as demonstrations for LLMs to conduct better ICL via instruction tuning. Extensive experiments on different LLMs and RE datasets demonstrate that our method generates relevant and valid entity pairs and boosts ICL abilities of LLMs, achieving competitive or new state-of-the-art performance on sentence-level RE compared to previous supervised fine-tuning methods and ICL-based methods.

  1. [Agrawal et al., 2022] Monica Agrawal, Stefan Hegsel- mann, Hunter Lang, et al. Large language models are few- shot clinical information extractors. In EMNLP, 2022.
  2. [Arora et al., 2023] Simran Arora, Avanika Narayan, Mayee F Chen, et al. Ask me anything: A simple strategy for prompting language models. In ICLR, 2023.
  3. [Baldini Soares et al., 2019] Livio Baldini Soares, Nicholas FitzGerald, Jeffrey Ling, et al. Matching the blanks: Dis- tributional similarity for relation learning. In ACL, 2019.
  4. [Brown et al., 2020] Tom Brown, Benjamin Mann, Nick Ry- der, et al. Language models are few-shot learners. In NeurIPS, 2020.
  5. [Cabot and Navigli, 2021] Pere-Lluís Huguet Cabot and Roberto Navigli. Rebel: Relation extraction by end-to-end language generation. In Findings of EMNLP, 2021.
  6. [Chen et al., 2022a] Xiang Chen, Ningyu Zhang, Xin Xie, et al. Knowprompt: Knowledge-aware prompt-tuning with synergistic optimization for relation extraction. In WWW, 2022.
  7. [Chen et al., 2022b] Yanda Chen, Ruiqi Zhong, Sheng Zha, et al. Meta-learning via language model in-context tuning. In ACL, 2022.
  8. [Chen et al., 2023] Zeming Chen, Gail Weiss, Eric Mitchell, et al. Reckoning: Reasoning through dynamic knowledge encoding. In NeurIPS, 2023.
  9. [Coda-Forno et al., 2023] Julian Coda-Forno, Marcel Binz, Zeynep Akata, et al. Meta-in-context learning in large lan- guage models. In NeurIPS, 2023.
  10. [Devlin et al., 2019] Jacob Devlin, Ming-Wei Chang, Ken- ton Lee, et al. Bert: Pre-training of deep bidirectional transformers for language understanding. In NAACL-HLT, 2019.
  11. [Du et al., 2022] Zhengxiao Du, Yujie Qian, Xiao Liu, et al. GLM: General language model pretraining with autore- gressive blank infilling. In ACL, 2022.
  12. [Gao et al., 2021] Tianyu Gao, Xingcheng Yao, and Danqi Chen. Simcse: Simple contrastive learning of sentence embeddings. In EMNLP, 2021.
  13. [Hendrickx et al., 2019] Iris Hendrickx, Su Nam Kim, Zor- nitsa Kozareva, et al. Semeval-2010 task 8: Multi-way classification of semantic relations between pairs of nomi- nals. arXiv preprint arXiv:1911.10422, 2019.
  14. [Holtzman et al., 2021] Ari Holtzman, Peter West, Vered Shwartz, et al. Surface form competition: Why the highest probability answer isn't always right. In EMNLP, 2021.
  15. [Hu et al., 2022] Edward J Hu, Yelong Shen, Phillip Wallis, et al. Lora: Low-rank adaptation of large language models. In ICLR, 2022.
  16. [Ji et al., 2023] Ke Ji, Yixin Lian, Jingsheng Gao, and Baoyuan Wang. Hierarchical verbalizer for few-shot hi- erarchical text classification. In ACL, 2023.
  17. [Jimenez Gutierrez et al., 2022] Bernal Jimenez Gutierrez, Nikolas McNeal, Clayton Washington, et al. Thinking about GPT-3 in-context learning for biomedical IE? think again. In Findings of EMNLP, 2022.
  18. [Lewis et al., 2020] Mike Lewis, Yinhan Liu, Naman Goyal, et al. BART: Denoising sequence-to-sequence pre-training for natural language generation, translation, and compre- hension. In ACL, 2020.
  19. [Li et al., 2022] Guozheng Li, Xu Chen, Peng Wang, et al. Fastre: Towards fast relation extraction with convolutional encoder and improved cascade binary tagging framework. In IJCAI, 2022. IJCAI
  20. [Li et al., 2023a] Bo Li, Wei Ye, Jinglei Zhang, et al. Re- viewing labels: Label graph network with top-k prediction set for relation extraction. In AAAI, 2023.
  21. [Li et al., 2023b] Bo Li, Dingyao Yu, Wei Ye, et al. Se- quence generation with label augmentation for relation ex- traction. In AAAI, 2023.
  22. [Li et al., 2023c] Guozheng Li, Peng Wang, and Wenjun Ke. Revisiting large language models as zero-shot relation ex- tractors. In Findings of EMNLP, 2023.
  23. [Li et al., 2024] Guozheng Li, Wenjun Ke, Peng Wang, Zi- jie Xu, Ke Ji, Jiajun Liu, Ziyu Shang, and Qiqing Luo. Unlocking instructive in-context learning with tabular prompting for relational triple extraction. arXiv preprint arXiv:2402.13741, 2024.
  24. [Liu et al., 2019] Yinhan Liu, Myle Ott, Naman Goyal, et al. Roberta: A robustly optimized bert pretraining approach. arXiv preprint arXiv:1907.11692, 2019.
  25. [Liu et al., 2022] Jiachang Liu, Dinghan Shen, Yizhe Zhang, et al. What makes good in-context examples for GPT-3? In DeeLIO, 2022.
  26. [Liu et al., 2024] Jiajun Liu, Wenjun Ke, Peng Wang, Ziyu Shang, Jinhua Gao, Guozheng Li, Ke Ji, and Yanhe Liu. Towards continual knowledge graph embedding via incre- mental distillation. In AAAI, 2024.
  27. [Luan et al., 2018] Yi Luan, Luheng He, Mari Ostendorf, et al. Multi-task identification of entities, relations, and coreference for scientific knowledge graph construction. arXiv preprint arXiv:1808.09602, 2018.
  28. [Luo et al., 2023] Linhao Luo, Yuan-Fang Li, Gholamreza Haffari, et al. Reasoning on graphs: Faithful and inter- pretable large language model reasoning. arXiv preprint arXiv:2310.01061, 2023.
  29. [Ma et al., 2023] Yubo Ma, Yixin Cao, YongChing Hong, et al. Large language model is not a good few-shot in- formation extractor, but a good reranker for hard samples! In Findings of EMNLP, 2023.
  30. [Min et al., 2022a] Sewon Min, Mike Lewis, Hannaneh Ha- jishirzi, et al. Noisy channel language model prompting for few-shot text classification. In ACL, 2022.
  31. [Min et al., 2022b] Sewon Min, Mike Lewis, Luke Zettle- moyer, et al. Metaicl: Learning to learn in context. In NAACL-HLT, 2022.
  32. [OpenAI, 2022] OpenAI. Introducing chatgpt, 2022.
  33. [Paolini et al., 2021] Giovanni Paolini, Ben Athiwaratkun, Jason Krone, et al. Structured prediction as translation be- tween augmented natural languages. In ICLR, 2021.
  34. [Radford et al., 2019] Alec Radford, Jeffrey Wu, Rewon Child, et al. Language models are unsupervised multitask learners. OpenAI blog, 1(8):9, 2019.
  35. [Raffel et al., 2020] Colin Raffel, Noam Shazeer, Adam Roberts, et al. Exploring the limits of transfer learning with a unified text-to-text transformer. JMLR, 21(1):5485- 5551, 2020.
  36. [Reimers and Gurevych, 2019] Nils Reimers and Iryna Gurevych. Sentence-bert: Sentence embeddings using siamese bert-networks. In EMNLP, 2019.
  37. [Roche, 2003] Christophe Roche. Ontology: a survey. IFAC Proceedings Volumes, 36(22):187-192, 2003.
  38. [Sainz et al., 2021] Oscar Sainz, Oier Lopez de Lacalle, Gorka Labaka, et al. Label verbalization and entailment for effective zero-and few-shot relation extraction. In EMNLP, 2021.
  39. [Shang et al., 2024] Ziyu Shang, Wenjun Ke, Nana Xiu, Peng Wang, Jiajun Liu, Yanhui Li, Zhizhao Luo, and Ke Ji. Ontofact: Unveiling fantastic fact-skeleton of llms via ontology-driven reinforcement learning. In AAAI, 2024.
  40. [Shi et al., 2023] Freda Shi, Xinyun Chen, Kanishka Misra, et al. Large language models can be easily distracted by irrelevant context. In ICML, 2023.
  41. [Touvron et al., 2023] Hugo Touvron, Thibaut Lavril, Gau- tier Izacard, et al. Llama: Open and efficient foundation language models. arXiv preprint arXiv:2302.13971, 2023.
  42. [Vaswani et al., 2017] Ashish Vaswani, Noam Shazeer, Niki Parmar, et al. Attention is all you need. In NIPS, 2017.
  43. [Wan et al., 2023] Zhen Wan, Fei Cheng, Zhuoyuan Mao, et al. GPT-RE: In-context learning for relation extraction using large language models. In EMNLP, 2023.
  44. [Wang et al., 2022] Chenguang Wang, Xiao Liu, Zui Chen, et al. Deepstruct: Pretraining of language models for struc- ture prediction. In Findings of ACL, 2022.
  45. [Wang et al., 2023a] Peng Wang, Tong Shao, Ke Ji, Guozheng Li, and Wenjun Ke. fmlre: a low-resource rela- tion extraction model based on feature mapping similarity calculation. In AAAI, 2023.
  46. [Wang et al., 2023b] Peng Wang, Jiafeng Xie, Xiye Chen, et al. Pascore: a chinese overlapping relation extraction model based on global pointer annotation strategy. In IJ- CAI, 2023. IJCAI
  47. [Wei et al., 2022a] Jason Wei, Maarten Bosma, Vincent Y Zhao, et al. Finetuned language models are zero-shot learners. In ICLR, 2022.
  48. [Wei et al., 2022b] Jason Wei, Xuezhi Wang, Dale Schuur- mans, et al. Chain of thought prompting elicits reasoning in large language models. In NeurIPS, 2022.
  49. [Yamada et al., 2020] Ikuya Yamada, Akari Asai, Hiroyuki Shindo, et al. Luke: deep contextualized entity represen- tations with entity-aware self-attention. In EMNLP, 2020.
  50. [Zhang et al., 2017] Yuhao Zhang, Victor Zhong, Danqi Chen, et al. Position-aware attention and supervised data improve slot filling. In EMNLP, 2017.
  51. [Zhao et al., 2021] Zihao Zhao, Eric Wallace, Shi Feng, et al. Calibrate before use: Improving few-shot performance of language models. In ICML, 2021.
  52. [Zhong and Chen, 2021] Zexuan Zhong and Danqi Chen. A frustratingly easy approach for entity and relation extrac- tion. In NAACL-HLT, 2021.
  53. [Zhou and Chen, 2022] Wenxuan Zhou and Muhao Chen. An improved baseline for sentence-level relation extrac- tion. In AACL, 2022.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards