Mixture Distribution Graph Network for Few Shot Learning

Autor: Qian Wang, Jie Lei, Yuxuan Shi, Lei Wu, Ping Li, Hefei Ling, Jialie Shen, Baiyan Zhang
Rok vydání: 2022
Předmět:
Zdroj: IEEE Transactions on Cognitive and Developmental Systems. 14:892-901
ISSN: 2379-8939
2379-8920
DOI: 10.1109/tcds.2021.3075280
Popis: Few-shot learning aims at heuristically resolving new tasks with limited labeled data; most of the existing approaches are affected by knowledge learned from similar experiences. However, inter-class barriers and new samples insufficiency limit the transfer of knowledge. In this paper, we propose a novel mixture distribution graph network, in which the inter-class relation is explicitly modeled and propagated via graph generation. Owing to the weighted distribution features based on Gaussian Mixture Model, we take class diversity into consideration, thereby utilizing information precisely and efficiently. Equipped with Minimal Gated Units, the “memory" of similar tasks can be preserved and reused through episode training, which fills a gap in temporal characteristics and softens the impact of data insufficiency. Extensive trials are carried out based on the MiniImageNet and CIFAR-FS datasets. Results turn out that our method exceeds most state-of-the-art approaches, which shows the validity and universality of our method in few-shot learning.
Databáze: OpenAIRE