A Single Graph Convolution Is All You Need: Efficient Grayscale Image Classification
Not on the map. Why?
Abstract (arXiv)
Image classifiers for domain-specific tasks like Synthetic Aperture Radar Automatic Target Recognition (SAR ATR) and chest X-ray classification often rely on convolutional neural networks (CNNs). These networks, while powerful, experience high latency due to the number of operations they perform, which can be problematic in real-time applications. Many image classification models are designed to work with both RGB and grayscale datasets, but classifiers that operate solely on grayscale images are less common. Grayscale image classification has critical applications in fields such as medical imaging and SAR ATR. In response, we present a novel grayscale image classification approach using a vectorized view of images. By leveraging the lightweight nature of Multi-Layer Perceptrons (MLPs), we treat images as vectors, simplifying the problem to grayscale image classification. Our approach incorporates a single graph convolutional layer in a batch-wise manner, enhancing accuracy and reducing performance variance. Additionally, we develop a customized accelerator on FPGA for our model, incorporating several optimizations to improve performance. Experimental results on benchmark grayscale image datasets demonstrate the effectiveness of our approach, achieving significantly lower latency (up to 16× less on MSTAR) and competitive or superior performance compared to state-of-the-art models for SAR ATR and medical image classification.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 21 of its 30 references that are in the map. First posted on arXiv 1 Feb 2024.
Builds on
21 of 30 references.
References · 21
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,364
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
- 2017Inductive Representation Learning on Large Graphscited by 6,320
- 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,349
- 2016Convolutional Neural Networks on Graphs with Fast Localized Spectral Filteringcited by 2,089
- 2021MLP-Mixer: An all-MLP Architecture for Visioncited by 1,446
- 2017Residual Gated Graph ConvNetscited by 346
- 2021ResMLP: Feedforward networks for image classification with data-efficient trainingcited by 339
- 2022Vision GNN: An Image is Worth Graph of Nodescited by 186
9 more
- 2023FastViT: A Fast Hybrid Vision Transformer using Structural Reparameterizationcited by 162
- 2021Vision Transformer for Small-Size Datasetscited by 122
- 2017Topology Adaptive Graph Convolutional Networkscited by 113
- 2023Accurate, Low-latency, Efficient SAR Automatic Target Recognition on FPGAcited by 10
- 2023GraphAGILE: An FPGA-based Overlay Accelerator for Low-latency GNN Inferencecited by 7
- 2021BA^2M: A Batch Aware Attention Module for Image Classificationcited by 5
- 2021Mini-batch graphs for robust image classificationcited by 3
- 2019Identity Connections in Residual Nets Improve Noise Stabilitycited by 2
- 2022Graph Neural Networks for Image Classification and Reinforcement Learning using Graph representationscited by 2
Built on by
Citing papers per year: 2024: 2.
2 papers.