KEoG: A knowledge-aware edge-oriented graph neural network for document-level relation extraction
Tao Li, Weihua Peng, Qingcai Chen, Xiaolong Wang, Buzhou Tang · 2020
Document-level relation extraction (RE) has attracted more and more attentions recently. Edge-oriented graph neural network (EoG) is a new neural network exhibiting greater potential than previous node-oriented graph neural networks for document-level RE. In this paper, we propose a novel EoG, called knowledge-aware edge-oriented GNN (KEoG) for document-level RE. In KEoG, we further introduce not only two types of nodes to represent documents and external knowledge respectively, but also soft F-Measure loss function to solve the inherent class imbalance problem in document-level RE. Experiments conducted on two document-level datasets show that KEoG outperforms other state-of-the-art methods for comparison on both intra-sentence and inter-sentence relation extractions, indicating that KEoG is an effective extension of EoG.