Relation classification via BERT with piecewise convolution and focal loss

Recent relation extraction models’ architecture are evolved from the shallow neural networks to natural language model, such as convolutional neural networks or recurrent neural networks to Bert. However, these methods did not consider the semantic information in the sequence or the distance depende...

Full description

Saved in:
Bibliographic Details
Published in:PloS one 2021-09, Vol.16 (9), p.e0257092-e0257092
Main Authors: Liu, Jianyi, Duan, Xi, Zhang, Ru, Sun, Youqiang, Guan, Lei, Lin, Bingjie
Format: Article
Language:English
Subjects:
Algorithms
Analysis
Artificial neural networks
Biology and Life Sciences
Classification
Communications networks
Computational linguistics
Computer and Information Sciences
Convolution
Deep learning
Evaluation
Information management
Language processing
Machine learning
Methods
Modelling
Natural language
Natural language interfaces
Neural networks
Recurrent neural networks
Research and Analysis Methods
Semantics
Social Sciences
Citations: Items that this one cites
Items that cite this one
Online Access:Get full text
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:Recent relation extraction models’ architecture are evolved from the shallow neural networks to natural language model, such as convolutional neural networks or recurrent neural networks to Bert. However, these methods did not consider the semantic information in the sequence or the distance dependence problem, the internal semantic information may contain the useful knowledge which can help relation classification. Focus on these problems, this paper proposed a BERT-based relation classification method. Compare with the existing Bert-based architecture, the proposed model can obtain the internal semantic information between entity pair and solve the distance semantic dependence better. The pre-trained BERT model after fine tuning is used in this paper to abstract the semantic representation of sequence, then adopt the piecewise convolution to obtain semantic information which influence the extraction results. Compare with the existing methods, the proposed method can achieve a better accuracy on relational extraction task because of the internal semantic information extracted in the sequence. While, the generalization ability is still a problem that cannot be ignored, and the numbers of the relationships are difference between different categories. In this paper, the focal loss function is adopted to solve this problem by assigning a heavy weight to less number or hard classify categories. Finally, comparing with the existing methods, the F1 metric of the proposed method can reach a superior result 89.95% on the SemEval-2010 Task 8 dataset.
ISSN:1932-6203
1932-6203
DOI:10.1371/journal.pone.0257092