Heterogeneous graph neural networks for noisy few-shot relation classification

被引:43
作者
Xie, Yuxiang [1 ,2 ]
Xu, Hua [1 ]
Li, Jiaoe [2 ]
Yang, Congcong [1 ,2 ]
Gao, Kai [2 ]
机构
[1] Tsinghua Univ, Dept Comp Sci & Technol, State Key Lab Intelligent Technol & Syst, Beijing 100084, Peoples R China
[2] Hebei Univ Sci & Technol, Sch Informat Sci & Engn, Shijiazhuang 050018, Hebei, Peoples R China
基金
美国国家科学基金会;
关键词
Relation extraction; Heterogeneous graph neural networks; Few-shot learning; Adversarial learning;
D O I
10.1016/j.knosys.2020.105548
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Relation classification is an essential and fundamental task in natural language processing. Distant supervised methods have achieved great success on relation classification, which improve the performance of the task through automatically extending the dataset. However, the distant supervised methods also bring the problem of wrong labeling. Inspired by people learning new knowledge from only a few samples, we focus on predicting formerly unseen classes with a few labeled data. In this paper, we propose a heterogeneous graph neural network for few-shot relation classification, which contains sentence nodes and entity nodes. We build the heterogeneous graph based on the message passing between entity nodes and sentence nodes in the graph, which can capture rich neighborhood information of the graph. Besides, we introduce adversarial learning for training a robust model and evaluate our heterogeneous graph neural networks under the scene of introducing different rates of noise data. Experimental results have demonstrated that our model outperforms the state-of-the-art baseline models on the FewRel dataset. (C) 2020 Elsevier B.V. All rights reserved.
引用
收藏
页数:9
相关论文
共 42 条
[1]  
[Anonymous], 2017, 5 INT C LEARN REPR I
[2]  
[Anonymous], 2018, 6 INT C LEARN REPR I
[3]  
[Anonymous], 2018, ICLR
[4]  
Bekoulis Giannis, 2018, P 2018 C EMP METH NA, P2830
[5]  
Feng J, 2018, AAAI CONF ARTIF INTE, P5779
[6]  
Finn C, 2017, PR MACH LEARN RES, V70
[7]  
Gao S., 2019, Proc. 2019 Conf. Empirical Methods Natural Lang. Process. 9th Int. Joint Conf. Natural Lang. Process. (EMNLP-IJCNLP), P3741, DOI [DOI 10.18653/V1/D19-1388, 10.18 653/v1/D19-1388]
[8]  
Gao S, 2019, AAAI CONF ARTIF INTE, P6399
[9]   Product-Aware Answer Generation in E-Commerce Question-Answering [J].
Gao, Shen ;
Ren, Zhaochun ;
Zhao, Yihong ;
Zhao, Dongyan ;
Yin, Dawei ;
Yan, Rui .
PROCEEDINGS OF THE TWELFTH ACM INTERNATIONAL CONFERENCE ON WEB SEARCH AND DATA MINING (WSDM'19), 2019, :429-437
[10]  
Gao TY, 2019, AAAI CONF ARTIF INTE, P6407