arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2005.04968cs.CVcs.LGeess.IV
收藏

存储受限器械图像分类技术的定量分析

Quantitative Analysis of Image Classification Techniques for Memory-Constrained Devices

School of Informatics, University of Edinburgh(爱丁堡大学信息学院)

Sebastian Müksch, Theo Olausson, John Wilhelm, Pavlos Andreadis

摘要:

卷积神经网络(CNN)是图像分类的最新技术,但通常以占用大量内存为代价。这限制了它们在依赖于嵌入式设备的应用程序中的实用性,其中内存通常是稀缺资源。最近,在这种内存受限的设备上的图像分类领域取得了重大进展,其中包括ProtoNN,Bonsai和FastGRNN算法等新的贡献。使用MNIST-10进行光学字符识别时,这些已被证明可达到高达98.2%的准确度,内存占用量仅为6 KB。然而,它们在更复杂的多类别和多通道图像分类上的潜力还有待确定。在本文中,我们将CNN与ProtoNN,Bonsai和FastGRNN应用于使用CIFAR-10的3通道图像分类时进行了比较。在我们的分析中,我们使用现有的直接卷积算法来实现CNN的内存优化,并提出了调整FastGRNN模型以处理多通道图像的新方法。我们将每个算法的评估扩展到8 KB,16 KB,32 KB,64 KB和128 KB的内存大小预算,以定量显示直接卷积CNN在所有选定的预算中表现最佳,在58.23KB的内存占用下,最高性能为65.7%的准确度。

英文摘要:

Convolutional Neural Networks, or CNNs, are the state of the art for image classification, but typically come at the cost of a large memory footprint. This limits their usefulness in applications relying on embedded devices, where memory is often a scarce resource. Recently, there has been significant progress in the field of image classification on such memory-constrained devices, with novel contributions like the ProtoNN, Bonsai and FastGRNN algorithms. These have been shown to reach up to 98.2% accuracy on optical character recognition using MNIST-10, with a memory footprint as little as 6KB. However, their potential on more complex multi-class and multi-channel image classification has yet to be determined. In this paper, we compare CNNs with ProtoNN, Bonsai and FastGRNN when applied to 3-channel image classification using CIFAR-10. For our analysis, we use the existing Direct Convolution algorithm to implement the CNNs memory-optimally and propose new methods of adjusting the FastGRNN model to work with multi-channel images. We extend the evaluation of each algorithm to a memory size budget of 8KB, 16KB, 32KB, 64KB and 128KB to show quantitatively that Direct Convolution CNNs perform best for all chosen budgets, with a top performance of 65.7% accuracy at a memory footprint of 58.23KB.

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

↑