arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2408.04460cs.LG
收藏

用于图像分类的二值神经网络训练中反向传播及其替代方案的实验比较研究

An experimental comparative study of backpropagation and alternatives for training binary neural networks for image classification

University of Tours(图尔大学);LIFAT(LIFAT实验室);CETU ILIAD3

Ben Crulis, Barthelemy Serres, Cyril de Runz, Gilles Venturini

AI总结:

本文针对二值神经网络难以用反向传播训练的问题,在ImageNette数据集上比较三种模型架构,并扩展了两种反向传播替代方案,验证其在图像分类中的有效性。

摘要:

当前的人工神经网络使用以浮点数编码的参数进行训练,这些参数在推理时占用大量内存空间。由于深度学习模型规模的增大,在边缘设备上考虑训练和使用人工神经网络正变得非常困难。二值神经网络有望减小深度神经网络模型的规模,同时提高推理速度并降低能耗。因此,它们可能允许在边缘设备上部署更强大的模型。然而,二值神经网络仍被证明难以使用基于反向传播的梯度下降方案进行训练。本文扩展了\cite{crulis2023alternatives}的工作,该工作提出将两种最初为连续神经网络设计的、有前景的反向传播替代方案适配到二值神经网络,并在简单的图像分类数据集上进行了实验。本文在ImageNette数据集上提出了新的实验,比较了三种不同的图像分类模型架构,并增加了两种额外的反向传播替代方案。

英文摘要:

Current artificial neural networks are trained with parameters encoded as floating point numbers that occupy lots of memory space at inference time. Due to the increase in the size of deep learning models, it is becoming very difficult to consider training and using artificial neural networks on edge devices. Binary neural networks promise to reduce the size of deep neural network models, as well as to increase inference speed while decreasing energy consumption. Thus, they may allow the deployment of more powerful models on edge devices. However, binary neural networks are still proven to be difficult to train using the backpropagation-based gradient descent scheme. This paper extends the work of \cite{crulis2023alternatives}, which proposed adapting to binary neural networks two promising alternatives to backpropagation originally designed for continuous neural networks, and experimented with them on simple image classification datasets. This paper proposes new experiments on the ImageNette dataset, compares three different model architectures for image classification, and adds two additional alternatives to backpropagation.

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

↑