arXiv:2603.13931cs.LG2026-03

4-bit神经网络训练在普通CPU上实现与全精度相当的准确率

True 4-Bit Quantized Convolutional Neural Network Training on CPU: Achieving Full-Precision Parity

  • 采用基于tanh的软权值裁剪结合对称量化和动态层缩放
  • CIFAR-10上达92.34%准确率,仅比全精度低0.16%
  • 可在手机等普通设备上快速收敛,适合资源受限场景

低精度神经网络训练成为降低计算成本、推动深度学习普及的有力方向。然而现有4比特量化方法或依赖昂贵GPU,或导致显著精度下降。本文提出一种基于标准PyTorch操作的实用4比特卷积神经网络训练方法,仅需通用CPU即可实现。通过引入新型tanh-based软权值裁剪技术,结合对称量化、动态每层缩放及直通估计器,实现稳定收敛与优异性能。在Google Colab免费CPU环境下,从零训练一个含325万参数的VGG风格模型,在CIFAR-10上达到92.34%测试准确率,仅比全精度基线(92.5%)低0.16%。在更难的CIFAR-100上,同样架构与流程下获得70.94%准确率,验证了方法可泛化至复杂任务。两实验均实现8倍内存压缩,且每层保持精确15个唯一权值。此外,我们在消费级手机(OnePlus 9R)上验证硬件无关性,仅6轮即达83.16%准确率。据我们所知,此前无工作在标准CPU上无需专用内核或后训练量化即实现4比特训练与全精度媲美。

原文摘要 · Abstract (English)

Low-precision neural network training has emerged as a promising direction for reducing computational costs and democratizing access to deep learning research. However, existing 4-bit quantization methods either rely on expensive GPU infrastructure or suffer from significant accuracy degradation. In this work, we present a practical method for training convolutional neural networks at true 4-bit precision using standard PyTorch operations on commodity CPUs. We introduce a novel tanh-based soft weight clipping technique that, combined with symmetric quantization, dynamic per-layer scaling, and straight-through estimators, achieves stable convergence and competitive accuracy. Training a VGG-style architecture with 3.25 million parameters from scratch on CIFAR-10, our method achieves 92.34% test accuracy on Google Colab's free CPU tier -- matching full-precision baseline performance (92.5%) with only a 0.16% gap. We further validate on CIFAR-100, achieving 70.94% test accuracy across 100 classes with the same architecture and training procedure, demonstrating that 4-bit training from scratch generalizes to harder classification tasks. Both experiments achieve 8x memory compression over FP32 while maintaining exactly 15 unique weight values per layer throughout training. We additionally validate hardware independence by demonstrating rapid convergence on a consumer mobile device (OnePlus 9R), achieving 83.16% accuracy in only 6 epochs. To the best of our knowledge, no prior work has demonstrated 4-bit quantization-aware training achieving full-precision parity on standard CPU hardware without specialized kernels or post-training quantization.

4-bit量化低资源训练CPU推理轻量模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。