Paper The following article is Open access

Image classification model based on GAT

, and

Published under licence by IOP Publishing Ltd
, , Citation Binghui Xu et al 2020 J. Phys.: Conf. Ser. 1570 012082 DOI 10.1088/1742-6596/1570/1/012082

1742-6596/1570/1/012082

Abstract

In the field of image classification, graph neural network (GNN) is a kind of structured data modeling architecture with larger functions. However, there are still some problems, such as low efficiency of updating nodes, fixed network parameters and the inability to effectively model the information features of some edges in the graph. In order to solve these problems, this paper introduces attention mechanism on the basis of GNN to improve it, proposes a graph attention network (GAT), establishes a double-layer GAT model, and uses regularization method in model iterative training to achieve image classification. The model is applied to three datasets for experiments. The experimental results show that the average classification accuracy of the proposed model is high and it has good application performance.

Export citation and abstract BibTeX RIS

Content from this work may be used under the terms of the Creative Commons Attribution 3.0 licence. Any further distribution of this work must maintain attribution to the author(s) and the title of the work, journal citation and DOI.

Please wait… references are loading.
10.1088/1742-6596/1570/1/012082