Contrastive Triple Extraction With Generative Transformer | Awesome LLM Papers Contribute to Awesome LLM Papers

Contrastive Triple Extraction With Generative Transformer

Hongbin Ye, Ningyu Zhang, Shumin Deng, Mosha Chen, Chuanqi Tan, Fei Huang, Huajun Chen . Proceedings of the AAAI Conference on Artificial Intelligence 2021 – 110 citations

[Paper]   Search on Google Scholar   Search on Semantic Scholar
AAAI Datasets Model Architecture Training Techniques

Triple extraction is an essential task in information extraction for natural language processing and knowledge graph construction. In this paper, we revisit the end-to-end triple extraction task for sequence generation. Since generative triple extraction may struggle to capture long-term dependencies and generate unfaithful triples, we introduce a novel model, contrastive triple extraction with a generative transformer. Specifically, we introduce a single shared transformer module for encoder-decoder-based generation. To generate faithful results, we propose a novel triplet contrastive training object. Moreover, we introduce two mechanisms to further improve model performance (i.e., batch-wise dynamic attention-masking and triple-wise calibration). Experimental results on three datasets (i.e., NYT, WebNLG, and MIE) show that our approach achieves better performance than that of baselines.

Similar Work