Task-driven meta-learning based on graph neural network
DOI:
CSTR:
Author:
Affiliation:

College of Automation and Electronic Engineering, Qingdao University of Science & Technology, Qingdao 266061, China

Clc Number:

TP183

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    In order to solve the issue of poor generalization ability of traditional classification models, an innovative meta-learning method was proposed in this paper. Firstly, a dynamic task-driven module was employed, which enabled the classification model to mine external task information; Secondly, a collaborative training method was adopted to overcome the increased complexity of calculation caused by the introduction of task-driven module; Finally, in order to exploit the relevant information between the samples, the graph neural network was used to be the classification module to achieve the purpose of further improving the classification accuracy. Compared with baselines, the accuracy of this method has increased by 6.81% and 6.20% on the MiniImageNet and CIFAR-FS datasets respectively. The experimental results show that the method in this paper can effectively improve the generalization ability of the classification model.

    Reference
    Related
    Cited by
Get Citation
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:
  • Revised:
  • Adopted:
  • Online: August 22,2024
  • Published:
Article QR Code