arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2405.08020cs.LGcs.CV

ReActXGB: 一种用于提升性能与计算效率的混合二值卷积神经网络架构

ReActXGB: A Hybrid Binary Convolutional Neural Network Architecture for Improved Performance and Computational Efficiency

National Central University(中央大学)

Po-Hsun Chu, Ching-Han Chen

AI总结:

针对二值卷积神经网络全连接层计算瓶颈及性能与资源平衡难题,提出用XGBoost替换ReActNet-A全卷积层的ReActXGB混合模型,在FashionMNIST基准上实现精度提升与计算、存储开销降低。

摘要:

二值卷积神经网络(BCNNs)为降低深度神经网络(DNNs)的内存需求与计算成本提供了潜在解决方案。然而,在性能与计算资源之间实现平衡仍是一项重大挑战。此外,BCNNs的全连接层已成为显著的计算瓶颈,这主要是因为常规做法会将输入层和全连接层排除在二值化之外,以避免精度大幅下降。本文提出了一种名为ReActXGB的混合模型,我们用XGBoost替换了ReActNet-A的全卷积层。该修改旨在缩小BCNNs与实值网络之间的性能差距,同时保持较低的计算成本。在FashionMNIST基准上的实验结果表明,ReActXGB的top-1精度比ReActNet-A高1.47%,同时浮点运算量(FLOPs)降低7.14%,模型大小减小1.02%。

英文摘要:

Binary convolutional neural networks (BCNNs) provide a potential solution to reduce the memory requirements and computational costs associated with deep neural networks (DNNs). However, achieving a trade-off between performance and computational resources remains a significant challenge. Furthermore, the fully connected layer of BCNNs has evolved into a significant computational bottleneck. This is mainly due to the conventional practice of excluding the input layer and fully connected layer from binarization to prevent a substantial loss in accuracy. In this paper, we propose a hybrid model named ReActXGB, where we replace the fully convolutional layer of ReActNet-A with XGBoost. This modification targets to narrow the performance gap between BCNNs and real-valued networks while maintaining lower computational costs. Experimental results on the FashionMNIST benchmark demonstrate that ReActXGB outperforms ReActNet-A by 1.47% in top-1 accuracy, along with a reduction of 7.14% in floating-point operations (FLOPs) and 1.02% in model size.

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

↑