Wei Zhongcheng, Guo Wenjie, Zhang Yunping, Zhang Jieying, Zhao Jijun
School of Information and Electrical Engineering, Hebei University of Engineering, Handan, Hebei, China.
Hebei Key Laboratory of Security & Protection Information Sensing and Processing, Handan, Hebei, China.
PeerJ Comput Sci. 2023 Mar 6;9:e1272. doi: 10.7717/peerj-cs.1272. eCollection 2023.
Few-shot relation extraction is used to solve the problem of long tail distribution of data by matching between query instances and support instances. Existing methods focus only on the single direction process of matching, ignoring the symmetry of the data in the process. To address this issue, we propose the bidirectional matching and aggregation network (BMAN), which is particularly powerful when the training data is symmetrical. This model not only tries to extract relations for query instances, but also seeks relational prototypes about the query instances to validate the feature representation of the support set. Moreover, to avoid overfitting in bidirectional matching, the data enhancement method was designed to scale up the number of instances while maintaining the scope of the instance relation class. Extensive experiments on FewRel and FewRel2.0 public datasets are conducted and evaluate the effectiveness of BMAN.
少样本关系抽取用于通过查询实例与支持实例之间的匹配来解决数据长尾分布问题。现有方法仅关注匹配的单向过程,忽略了该过程中数据的对称性。为了解决这个问题,我们提出了双向匹配与聚合网络(BMAN),当训练数据对称时,该网络特别强大。该模型不仅尝试为查询实例提取关系,还寻找关于查询实例的关系原型以验证支持集的特征表示。此外,为了避免双向匹配中的过拟合,设计了数据增强方法,在保持实例关系类范围的同时扩大实例数量。在FewRel和FewRel2.0公共数据集上进行了广泛实验,并评估了BMAN的有效性。