用极低精度权重训练神经网络,实现内存与效率的双重提升
Low-Resolution Neural Networks
- 将模型权重压缩至1到4.08比特,保持分类性能
- 2.32比特权重在内存与性能间达最佳平衡
- 适合资源受限设备部署,尤其对大模型更有效
大型神经网络模型规模不断扩张,带来显著的内存和计算效率挑战。本文研究了参数位精度对模型性能的影响,对比标准32比特模型,聚焦图像多类物体分类任务。分析涵盖全连接层、卷积层和Transformer模块,权重分辨率范围为1比特至4.08比特。结果表明,低精度模型性能可媲美32比特模型,适用于内存受限设备。小参数量模型需更多训练轮次才能达到相近准确率,而大参数量模型在相同轮次内即可达成相似性能。数据增强可能使低精度模型训练不稳定,但允许权重取零值可维持稳定性并防止性能下降。总体而言,2.32比特权重在内存节省、性能和效率间取得最优平衡。未来研究应拓展至其他数据集及更大更复杂的模型。这些发现预示着低内存需求神经网络的新时代,但需专用硬件支持以充分发挥潜力。
原文摘要 · Abstract (English)
The expanding scale of large neural network models introduces significant challenges, driving efforts to reduce memory usage and enhance computational efficiency. Such measures are crucial to ensure the practical implementation and effective application of these sophisticated models across a wide array of use cases. This study examines the impact of parameter bit precision on model performance compared to standard 32-bit models, with a focus on multiclass object classification in images. The models analyzed include those with fully connected layers, convolutional layers, and transformer blocks, with model weight resolution ranging from 1 bit to 4.08 bits. The findings indicate that models with lower parameter bit precision achieve results comparable to 32-bit models, showing promise for use in memory-constrained devices. While low-resolution models with a small number of parameters require more training epochs to achieve accuracy comparable to 32-bit models, those with a large number of parameters achieve similar performance within the same number of epochs. Additionally, data augmentation can destabilize training in low-resolution models, but including zero as a potential value in the weight parameters helps maintain stability and prevents performance degradation. Overall, 2.32-bit weights offer the optimal balance of memory reduction, performance, and efficiency. However, further research should explore other dataset types and more complex and larger models. These findings suggest a potential new era for optimized neural network models with reduced memory requirements and improved computational efficiency, though advancements in dedicated hardware are necessary to fully realize this potential.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。