Zhengwei Zhai , Rongli Fan , Jie Huang , Neal Xiong , Lijuan Zhang , Jian Wan , Lei Zhang
{"title":"基于交叉注意机制和全局指针的新型联合提取模型(使用上下文屏蔽窗口","authors":"Zhengwei Zhai , Rongli Fan , Jie Huang , Neal Xiong , Lijuan Zhang , Jian Wan , Lei Zhang","doi":"10.1016/j.csl.2024.101643","DOIUrl":null,"url":null,"abstract":"<div><p>Relational triple extraction is a critical step in knowledge graph construction. Compared to pipeline-based extraction, joint extraction is gaining more attention because it can better utilize entity and relation information without causing error propagation issues. Yet, the challenge with joint extraction lies in handling overlapping triples. Existing approaches adopt sequential steps or multiple modules, which often accumulate errors and interfere with redundant data. In this study, we propose an innovative joint extraction model with cross-attention mechanism and global pointers with context shield window. Specifically, our methodology begins by inputting text data into a pre-trained RoBERTa model to generate word vector representations. Subsequently, these embeddings are passed through a modified cross-attention layer along with entity type embeddings to address missing entity type information. Next, we employ the global pointer to transform the extraction problem into a quintuple extraction problem, which skillfully solves the issue of overlapping triples. It is worth mentioning that we design a context shield window on the global pointer, which facilitates the identification of correct entities within a limited range during the entity extraction process. Finally, the capability of our model against malicious samples is improved by adding adversarial training during the training process. Demonstrating superiority over mainstream models, our approach achieves impressive results on three publicly available datasets.</p></div>","PeriodicalId":50638,"journal":{"name":"Computer Speech and Language","volume":"87 ","pages":"Article 101643"},"PeriodicalIF":3.1000,"publicationDate":"2024-03-18","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"https://www.sciencedirect.com/science/article/pii/S0885230824000263/pdfft?md5=6db6d29053e0503fc07e8e1ded002d0e&pid=1-s2.0-S0885230824000263-main.pdf","citationCount":"0","resultStr":"{\"title\":\"A novel joint extraction model based on cross-attention mechanism and global pointer using context shield window\",\"authors\":\"Zhengwei Zhai , Rongli Fan , Jie Huang , Neal Xiong , Lijuan Zhang , Jian Wan , Lei Zhang\",\"doi\":\"10.1016/j.csl.2024.101643\",\"DOIUrl\":null,\"url\":null,\"abstract\":\"<div><p>Relational triple extraction is a critical step in knowledge graph construction. Compared to pipeline-based extraction, joint extraction is gaining more attention because it can better utilize entity and relation information without causing error propagation issues. Yet, the challenge with joint extraction lies in handling overlapping triples. Existing approaches adopt sequential steps or multiple modules, which often accumulate errors and interfere with redundant data. In this study, we propose an innovative joint extraction model with cross-attention mechanism and global pointers with context shield window. Specifically, our methodology begins by inputting text data into a pre-trained RoBERTa model to generate word vector representations. Subsequently, these embeddings are passed through a modified cross-attention layer along with entity type embeddings to address missing entity type information. Next, we employ the global pointer to transform the extraction problem into a quintuple extraction problem, which skillfully solves the issue of overlapping triples. It is worth mentioning that we design a context shield window on the global pointer, which facilitates the identification of correct entities within a limited range during the entity extraction process. Finally, the capability of our model against malicious samples is improved by adding adversarial training during the training process. Demonstrating superiority over mainstream models, our approach achieves impressive results on three publicly available datasets.</p></div>\",\"PeriodicalId\":50638,\"journal\":{\"name\":\"Computer Speech and Language\",\"volume\":\"87 \",\"pages\":\"Article 101643\"},\"PeriodicalIF\":3.1000,\"publicationDate\":\"2024-03-18\",\"publicationTypes\":\"Journal Article\",\"fieldsOfStudy\":null,\"isOpenAccess\":false,\"openAccessPdf\":\"https://www.sciencedirect.com/science/article/pii/S0885230824000263/pdfft?md5=6db6d29053e0503fc07e8e1ded002d0e&pid=1-s2.0-S0885230824000263-main.pdf\",\"citationCount\":\"0\",\"resultStr\":null,\"platform\":\"Semanticscholar\",\"paperid\":null,\"PeriodicalName\":\"Computer Speech and Language\",\"FirstCategoryId\":\"94\",\"ListUrlMain\":\"https://www.sciencedirect.com/science/article/pii/S0885230824000263\",\"RegionNum\":3,\"RegionCategory\":\"计算机科学\",\"ArticlePicture\":[],\"TitleCN\":null,\"AbstractTextCN\":null,\"PMCID\":null,\"EPubDate\":\"\",\"PubModel\":\"\",\"JCR\":\"Q2\",\"JCRName\":\"COMPUTER SCIENCE, ARTIFICIAL INTELLIGENCE\",\"Score\":null,\"Total\":0}","platform":"Semanticscholar","paperid":null,"PeriodicalName":"Computer Speech and Language","FirstCategoryId":"94","ListUrlMain":"https://www.sciencedirect.com/science/article/pii/S0885230824000263","RegionNum":3,"RegionCategory":"计算机科学","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"Q2","JCRName":"COMPUTER SCIENCE, ARTIFICIAL INTELLIGENCE","Score":null,"Total":0}
引用次数: 0
摘要
关系三重抽取是知识图谱构建的关键步骤。与基于流水线的提取相比,联合提取更受关注,因为它能更好地利用实体和关系信息,而不会造成错误传播问题。然而,联合提取的挑战在于如何处理重叠的三元组。现有方法采用顺序步骤或多个模块,往往会积累错误并干扰冗余数据。在本研究中,我们提出了一种创新的联合提取模型,该模型具有交叉关注机制和带上下文屏蔽窗口的全局指针。具体来说,我们的方法首先将文本数据输入预先训练好的 RoBERTa 模型,生成词向量表示。随后,这些嵌入将与实体类型嵌入一起通过修改后的交叉注意层,以解决实体类型信息缺失的问题。接下来,我们利用全局指针将提取问题转化为五元提取问题,巧妙地解决了三元重叠的问题。值得一提的是,我们在全局指针上设计了一个上下文屏蔽窗口,这有助于在实体提取过程中识别有限范围内的正确实体。最后,我们在训练过程中加入了对抗训练,从而提高了模型对抗恶意样本的能力。与主流模型相比,我们的方法在三个公开数据集上取得了令人印象深刻的结果。
A novel joint extraction model based on cross-attention mechanism and global pointer using context shield window
Relational triple extraction is a critical step in knowledge graph construction. Compared to pipeline-based extraction, joint extraction is gaining more attention because it can better utilize entity and relation information without causing error propagation issues. Yet, the challenge with joint extraction lies in handling overlapping triples. Existing approaches adopt sequential steps or multiple modules, which often accumulate errors and interfere with redundant data. In this study, we propose an innovative joint extraction model with cross-attention mechanism and global pointers with context shield window. Specifically, our methodology begins by inputting text data into a pre-trained RoBERTa model to generate word vector representations. Subsequently, these embeddings are passed through a modified cross-attention layer along with entity type embeddings to address missing entity type information. Next, we employ the global pointer to transform the extraction problem into a quintuple extraction problem, which skillfully solves the issue of overlapping triples. It is worth mentioning that we design a context shield window on the global pointer, which facilitates the identification of correct entities within a limited range during the entity extraction process. Finally, the capability of our model against malicious samples is improved by adding adversarial training during the training process. Demonstrating superiority over mainstream models, our approach achieves impressive results on three publicly available datasets.
期刊介绍:
Computer Speech & Language publishes reports of original research related to the recognition, understanding, production, coding and mining of speech and language.
The speech and language sciences have a long history, but it is only relatively recently that large-scale implementation of and experimentation with complex models of speech and language processing has become feasible. Such research is often carried out somewhat separately by practitioners of artificial intelligence, computer science, electronic engineering, information retrieval, linguistics, phonetics, or psychology.