Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Superpixel Image Classification with Graph Attention Networks

About

This paper presents a methodology for image classification using Graph Neural Network (GNN) models. We transform the input images into region adjacency graphs (RAGs), in which regions are superpixels and edges connect neighboring superpixels. Our experiments suggest that Graph Attention Networks (GATs), which combine graph convolutions with self-attention mechanisms, outperforms other GNN models. Although raw image classifiers perform better than GATs due to information loss during the RAG generation, our methodology opens an interesting avenue of research on deep learning beyond rectangular-gridded images, such as 360-degree field of view panoramas. Traditional convolutional kernels of current state-of-the-art methods cannot handle panoramas, whereas the adapted superpixel algorithms and the resulting region adjacency graphs can naturally feed a GNN, without topology issues.

Pedro H. C. Avelar, Anderson R. Tavares, Thiago L. T. da Silveira, Cl\'audio R. Jung, Lu\'is C. Lamb• 2020

Related benchmarks

TaskDatasetResultRank
Image ClassificationFashion MNIST (test)--
568
Handwritten digit classificationSuperpixel MNIST 75 vertices (test)
Accuracy0.9619
7
Image ClassificationMNIST 75 (test)
Accuracy96.19
5
Image ClassificationFashionMNIST 75 (test)
Accuracy83.07
2
Showing 4 of 4 rows

Other info

Code

Follow for update