Proceedings

Meta In-Context Learning Makes Large Language Models Better Zero and Few-Shot Relation Extractors

Download BibTeX Cite

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

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

Abstract

Relation extraction (RE) is an important task that aims to identify the relationships between entities in texts. While large language models (LLMs) have revealed remarkable in-context learning (ICL) capability for general zero and few-shot learning, recent studies indicate that current LLMs still struggle with zero and few-shot RE. Previous studies are mainly dedicated to design prompt formats and select good examples for improving ICL-based RE. Although both factors are vital for ICL, if one can fundamentally boost the ICL capability of LLMs in RE, the zero and few-shot RE performance via ICL would be significantly improved. To this end, we introduce Micre (Meta In-Context learning of LLMs for Relation Extraction), a new meta-training framework for zero and few-shot RE where an LLM is tuned to do ICL on a diverse collection of RE datasets (i.e., learning to learn in context for RE). Through meta-training, the model becomes more effectively to learn a new RE task in context by conditioning on a few training examples with no parameter updates or task-specific templates at inference time, enabling better zero and few-shot task generalization. We experiment Micre on various LLMs with different model scales and 12 public RE datasets, and then evaluate it on unseen RE benchmarks under zero and few-shot settings. Micre delivers comparable or superior performance compared to a range of baselines including supervised fine-tuning and typical in-context learning methods. We find that the gains are particular significant for larger model scales, and using a diverse set of the meta-training RE datasets is key to improvements. Empirically, we show that Micre can transfer the relation semantic knowledge via relation label name during inference on target RE datasets.

  1. [Agrawal et al., 2022] Monica Agrawal, Stefan Hegsel- mann, et al. Large language models are few-shot clinical information extractors. In EMNLP, 2022.
  2. [Brown et al., 2020] Tom Brown, Benjamin Mann, et al. Language models are few-shot learners. In NeurIPS, 2020.
  3. [Chen and Li, 2021] Chih-Yao Chen and Cheng-Te Li. ZS- BERT: Towards zero-shot relation extraction with attribute representation learning. In NAACL-HLT, 2021.
  4. [Chen et al., 2022] Yanda Chen, Ruiqi Zhong, et al. Meta- learning via language model in-context tuning. In ACL, 2022.
  5. [Chia et al., 2022] Yew Ken Chia, Lidong Bing, et al. Re- lationprompt: Leveraging prompts to generate synthetic data for zero-shot relation triplet extraction. In Findings of ACL, 2022.
  6. [Cong et al., 2021] Xin Cong, Shiyao Cui, et al. Few-shot event detection with prototypical amortized conditional random field. In Findings of ACL, 2021.
  7. [Cong et al., 2022] Xin Cong, Jiawei Sheng, et al. Relation- guided few-shot relational triple extraction. In SIGIR, 2022.
  8. [Devlin et al., 2019] Jacob Devlin, Ming-Wei Chang, et al. Bert: Pre-training of deep bidirectional transformers for language understanding. In NAACL-HLT, 2019.
  9. [Doddington et al., 2004] George R Doddington, Alexis Mitchell, et al. The automatic content extraction (ace) program-tasks, data, and evaluation. In LREC, 2004.
  10. [Du et al., 2022] Zhengxiao Du, Yujie Qian, et al. GLM: General language model pretraining with autoregressive blank infilling. In ACL, 2022.
  11. [Finn et al., 2017] Chelsea Finn, Pieter Abbeel, and Sergey Levine. Model-agnostic meta-learning for fast adaptation of deep networks. In ICML, 2017.
  12. [Gardent et al., 2017] Claire Gardent, Anastasia Shimorina, et al. The webnlg challenge: Generating text from rdf data. In INLG, 2017.
  13. [Gurulingappa et al., 2012] Harsha Gurulingappa, Ab- dul Mateen Rajput, et al. Development of a benchmark corpus to support the automatic extraction of drug-related adverse effects from medical case reports. Journal of biomedical informatics, 45(5):885-892, 2012.
  14. [Han et al., 2018] Xu Han, Hao Zhu, et al. FewRel: A large- scale supervised few-shot relation classification dataset with state-of-the-art evaluation. In EMNLP, 2018.
  15. [Han et al., 2021] Jiale Han, Bo Cheng, and Wei Lu. Ex- ploring task difficulty for few-shot relation extraction. In EMNLP, 2021.
  16. [Hendrickx et al., 2019] Iris Hendrickx, Su Nam Kim, et al. Semeval-2010 task 8: Multi-way classification of seman- tic relations between pairs of nominals. arXiv preprint arXiv:1911.10422, 2019.
  17. [Hu et al., 2022] Edward J Hu, Yelong Shen, et al. Lora: Low-rank adaptation of large language models. In ICLR, 2022.
  18. [Jat et al., 2018] Sharmistha Jat, Siddhesh Khandelwal, and Partha Talukdar. Improving distantly supervised relation extraction using word and entity based attention. arXiv preprint arXiv:1804.06987, 2018.
  19. [Ji et al., 2023] Ke Ji, Yixin Lian, et al. Hierarchical verbal- izer for few-shot hierarchical text classification. In ACL, 2023.
  20. [Kim et al., 2023] Bosung Kim, Hayate Iso, et al. Zero-shot triplet extraction by template infilling. In AACL, 2023.
  21. [Kojima et al., 2022] Takeshi Kojima, Shixiang Shane Gu, et al. Large language models are zero-shot reasoners. In NeurIPS, 2022.
  22. [Levy et al., 2017] Omer Levy, Minjoon Seo, et al. Zero- shot relation extraction via reading comprehension. In CoNLL, 2017.
  23. [Li et al., 2022] Guozheng Li, Xu Chen, et al. Fastre: To- wards fast relation extraction with convolutional encoder and improved cascade binary tagging framework. In IJ- CAI, 2022. IJCAI
  24. [Li et al., 2023] Guozheng Li, Peng Wang, and Wenjun Ke. Revisiting large language models as zero-shot relation ex- tractors. In Findings of EMNLP, 2023.
  25. [Li et al., 2024] Guozheng Li, Wenjun Ke, et al. Unlock- ing instructive in-context learning with tabular prompt- ing for relational triple extraction. arXiv preprint arXiv:2402.13741, 2024.
  26. [Liu et al., 2019] Yinhan Liu, Myle Ott, et al. Roberta: A ro- bustly optimized bert pretraining approach. arXiv preprint arXiv:1907.11692, 2019.
  27. [Liu et al., 2024] Jiajun Liu, Wenjun Ke, et al. Towards con- tinual knowledge graph embedding via incremental distil- lation. In AAAI, 2024.
  28. [Luan et al., 2018] Yi Luan, Luheng He, et al. Multi-task identification of entities, relations, and coreference for scientific knowledge graph construction. arXiv preprint arXiv:1808.09602, 2018.
  29. [Ma et al., 2023a] Xilai Ma, Jing Li, and Min Zhang. Chain of thought with explicit evidence reasoning for few-shot relation extraction. In Findings of EMNLP, 2023.
  30. [Ma et al., 2023b] Yubo Ma, Yixin Cao, et al. Large lan- guage model is not a good few-shot information extrac- tor, but a good reranker for hard samples! In Findings of EMNLP, 2023.
  31. [Min et al., 2022] Sewon Min, Mike Lewis, et al. Metaicl: Learning to learn in context. In NAACL-HLT, 2022.
  32. [Mishra et al., 2022] Swaroop Mishra, Daniel Khashabi, et al. Cross-task generalization via natural language crowdsourcing instructions. In ACL, 2022.
  33. [OpenAI, 2022] OpenAI. Introducing chatgpt, 2022.
  34. [Peng et al., 2020] Hao Peng, Tianyu Gao, et al. Learning from context or names? an empirical study on neural rela- tion extraction. In ACL, 2020.
  35. [Radford et al., 2019] Alec Radford, Jeffrey Wu, et al. Lan- guage models are unsupervised multitask learners. OpenAI blog, 1(8):9, 2019.
  36. [Raffel et al., 2020] Colin Raffel, Noam Shazeer, et al. Ex- ploring the limits of transfer learning with a unified text- to-text transformer. JMLR, 21(1):5485-5551, 2020.
  37. [Rasley et al., 2020] Jeff Rasley, Samyam Rajbhandari, et al. Deepspeed: System optimizations enable training deep learning models with over 100 billion parameters. In SIGKDD, 2020.
  38. [Riedel et al., 2010] Sebastian Riedel, Limin Yao, and An- drew McCallum. Modeling relations and their mentions without labeled text. In ECML-PKDD, 2010.
  39. [Roth and Yih, 2004] Dan Roth and Wen-tau Yih. A linear programming formulation for global inference in natural language tasks. In CoNLL, 2004.
  40. [Sainz et al., 2021] Oscar Sainz, Oier Lopez de Lacalle, et al. Label verbalization and entailment for effective zero- and few-shot relation extraction. In EMNLP, 2021.
  41. [Sanh et al., 2022] Victor Sanh, Albert Webson, et al. Mul- titask prompted training enables zero-shot task generaliza- tion. In ICLR, 2022.
  42. [Shang et al., 2024] Ziyu Shang, Wenjun Ke, et al. Ontofact: Unveiling fantastic fact-skeleton of llms via ontology- driven reinforcement learning. In AAAI, 2024.
  43. [Snell et al., 2017] Jake Snell, Kevin Swersky, and Richard Zemel. Prototypical networks for few-shot learning. In NIPS, 2017.
  44. [Takanobu et al., 2019] Ryuichi Takanobu, Tianyang Zhang, et al. A hierarchical framework for relation extraction with reinforcement learning. In AAAI, 2019.
  45. [Touvron et al., 2023] Hugo Touvron, Thibaut Lavril, et al. Llama: Open and efficient foundation language models. arXiv preprint arXiv:2302.13971, 2023.
  46. [Vaswani et al., 2017] Ashish Vaswani, Noam Shazeer, et al. Attention is all you need. In NIPS, 2017.
  47. [Walker et al., 2005] Christopher Walker, Stephanie Strassel, et al. Ace 2005 multilingual training corpus- linguistic data consortium. URL: https://catalog. ldc. upenn. edu/LDC2006T06, 2005. Link
  48. [Wan et al., 2023] Zhen Wan, Fei Cheng, et al. GPT-RE: In-context learning for relation extraction using large lan- guage models. In EMNLP, 2023.
  49. [Wang and Lu, 2020] Jue Wang and Wei Lu. Two are better than one: Joint entity and relation extraction with table- sequence encoders. In EMNLP, 2020.
  50. [Wang et al., 2022] Chenguang Wang, Xiao Liu, et al. Deep- struct: Pretraining of language models for structure predic- tion. In Findings of ACL, 2022.
  51. [Wang et al., 2023a] Peng Wang, Tong Shao, et al. fmlre: a low-resource relation extraction model based on feature mapping similarity calculation. In AAAI, 2023.
  52. [Wang et al., 2023b] Peng Wang, Jiafeng Xie, et al. Pascore: a chinese overlapping relation extraction model based on global pointer annotation strategy. In IJCAI, 2023. IJCAI
  53. [Wang et al., 2023c] Xiao Wang, Weikang Zhou, et al. In- structuie: Multi-task instruction tuning for unified in- formation extraction. arXiv preprint arXiv:2304.08085, 2023.
  54. [Wei et al., 2020] Zhepei Wei, Jianlin Su, et al. A novel cas- cade binary tagging framework for relational triple extrac- tion. In ACL, 2020.
  55. [Wei et al., 2022a] Jason Wei, Maarten Bosma, et al. Fine- tuned language models are zero-shot learners. In ICLR, 2022.
  56. [Wei et al., 2022b] Jason Wei, Xuezhi Wang, et al. Chain of thought prompting elicits reasoning in large language models. In NeurIPS, 2022.
  57. [Yang and Katiyar, 2020] Yi Yang and Arzoo Katiyar. Sim- ple and effective few-shot named entity recognition with structured nearest neighbor learning. In EMNLP, 2020.
  58. [Yang et al., 2023] Chengmei Yang, Shuai Jiang, et al. Mu- tually guided few-shot learning for relational triple extrac- tion. In ICASSP, 2023.
  59. [Yu et al., 2020] Haiyang Yu, Ningyu Zhang, et al. Bridging text and knowledge with multi-prototype embedding for few-shot relational triple extraction. In COLING, 2020.
  60. [Zhang and Lu, 2022] Peiyuan Zhang and Wei Lu. Better few-shot relation extraction with label prompt dropout. In EMNLP, 2022.
  61. [Zhang and Wang, 2015] Dongxu Zhang and Dong Wang. Relation classification via recurrent neural network. arXiv preprint arXiv:1508.01006, 2015.
  62. [Zhang et al., 2017] Yuhao Zhang, Victor Zhong, et al. Position-aware attention and supervised data improve slot filling. In EMNLP, 2017.
  63. [Zhang et al., 2023] Liang Zhang, Chulun Zhou, et al. HyperNetwork-based decoupling to improve model gen- eralization for few-shot relation extraction. In EMNLP, 2023.
  64. [Zhao et al., 2021] Zihao Zhao, Eric Wallace, et al. Calibrate before use: Improving few-shot performance of language models. In ICML, 2021.
  65. [Zhao et al., 2023] Jun Zhao, WenYu Zhan, et al. RE- matching: A fine-grained semantic matching method for zero-shot relation extraction. In ACL, 2023.

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

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