Dynamic Prototype Selection by Fusing Attention Mechanism for Few-Shot Relation Classification

Linfang Wu, Hua Ping Zhang*, Yaofei Yang, Xin Liu, Kai Gao

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

7 Citations (Scopus)

Abstract

In a relation classification task, few-shot learning is an effective method when the number of training instances decreases. The prototypical network is a few-shot classification model that generates a point to represent each class, and this point is called a prototype. The mean is used to select prototypes for each class from a support set in a prototypical network. This method is fixed and static, and will lose some information at the sentence level. Therefore, we treat the mean selection as a special attention mechanism, then we expand the mean selection to dynamic prototype selection by fusing a self-attention mechanism. We also propose a query-attention mechanism to more accurately select prototypes. Experimental results on the FewRel dataset show that our model achieves significant and consistent improvements to baselines on few-shot relation classification.

Original languageEnglish
Title of host publicationIntelligent Information and Database Systems - 12th Asian Conference, ACIIDS 2020, Proceedings
EditorsNgoc Thanh Nguyen, Bogdan Trawinski, Kietikul Jearanaitanakij, Suphamit Chittayasothorn, Ali Selamat
PublisherSpringer
Pages431-441
Number of pages11
ISBN (Print)9783030419639
DOIs
Publication statusPublished - 2020
Event12th Asian Conference on Intelligent Information and Database Systems, ACIIDS 2020 - Phuket, Thailand
Duration: 23 Mar 202026 Mar 2020

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume12033 LNAI
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference12th Asian Conference on Intelligent Information and Database Systems, ACIIDS 2020
Country/TerritoryThailand
CityPhuket
Period23/03/2026/03/20

Keywords

  • Attention mechanism
  • Few-shot learning
  • Relation classification

Fingerprint

Dive into the research topics of 'Dynamic Prototype Selection by Fusing Attention Mechanism for Few-Shot Relation Classification'. Together they form a unique fingerprint.

Cite this