Title
Word Embedding Distribution Propagation Graph Network for Few-Shot Learning
Abstract
Few-shot learning (FSL) is of great significance to the field of machine learning. The ability to learn and generalize using a small number of samples is an obvious distinction between artificial intelligence and humans. In the FSL domain, most graph neural networks (GNNs) focus on transferring labeled sample information to an unlabeled query sample, ignoring the important role of semantic information during the classification process. Our proposed method embeds semantic information of classes into a GNN, creating a word embedding distribution propagation graph network (WPGN) for FSL. We merge the attention mechanism with our backbone network, use the Mahalanobis distance to calculate the similarity of classes, select the Funnel ReLU (FReLU) function as the activation function of the Transform layer, and update the point graph and word embedding distribution graph. In extensive experiments on FSL benchmarks, compared with the baseline model, the accuracy of the WPGN on the 5-way-1/2/5 shot tasks increased by 9.03, 4.56, and 4.15%, respectively.
Year
DOI
Venue
2022
10.3390/s22072648
SENSORS
Keywords
DocType
Volume
few-shot learning, graph neural network, semantic information, attention mechanism, Mahalanobis distance, FReLU
Journal
22
Issue
ISSN
Citations 
7
1424-8220
0
PageRank 
References 
Authors
0.34
0
3
Name
Order
Citations
PageRank
Chaoran Zhu100.34
Ling Wang200.34
Cheng Han301.01