arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.00564cs.CVcs.LG

单个图卷积足矣:高效的灰度图像分类

A Single Graph Convolution Is All You Need: Efficient Grayscale Image Classification

University of Southern California(南加州大学);DEVCOM Army Research Office(DEVCOM陆军研究办公室)

Jacob Fein-Ashley, Sachini Wickramasinghe, Bingyi Zhang, Rajgopal Kannan, Viktor Prasanna

AI总结:

针对灰度图像分类的高延迟问题,提出将图像向量化并利用MLP结合单个图卷积层的方法,同时在FPGA上开发定制加速器,实现显著降低延迟且保持竞争性能。

摘要:

用于合成孔径雷达自动目标识别(SAR ATR)和胸部X光分类等特定领域任务的图像分类器通常依赖于卷积神经网络(CNNs)。这些网络虽然强大,但由于执行的操作数量多而存在高延迟,这在实时应用中可能成为问题。许多图像分类模型被设计为同时适用于RGB和灰度数据集,但仅针对灰度图像操作的分类器较少见。灰度图像分类在医学成像和SAR ATR等领域具有关键应用。为此,我们提出了一种新颖的灰度图像分类方法,采用图像的向量化视角。通过利用多层感知机(MLPs)的轻量级特性,我们将图像视为向量,将问题简化为灰度图像分类。我们的方法以批处理方式融入单个图卷积层,提高了准确性并降低了性能方差。此外,我们在FPGA上为模型开发了定制加速器,整合多项优化以提升性能。在基准灰度图像数据集上的实验结果证明了方法的有效性,相比SAR ATR和医学图像分类的最先进模型,实现了显著更低的延迟(在MSTAR上低至16倍)和具有竞争力或更优的性能。

英文摘要:

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\times$ less on MSTAR) and competitive or superior performance compared to state-of-the-art models for SAR ATR and medical image classification.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑