Bit-Mass理论——容器原理
摘要
Bit-Mass理论提出,决定模型准确率的是权重位的总数,而非计算格式。在MNIST上的实验表明,相同的Bit-Mass下,二进制网络与浮点网络性能相当。
**Bit-Mass决定信息容量,进而决定模型准确率,而非所选的计算格式。**
此处介绍的Bit-Mass理论以权重位的总数为核心量,重新审视神经网络。
在MNIST上,使用相同的203264位Bit-Mass,Float32矩阵乘法与基于XNOR加Popcount的BV32取得了完全可比的结果。
**三种训练器对比(架构784→8→10,三个周期):**
- 带动量和自适应学习率的AdamW:81.3%
- 普通SGD(Float32):76.0%
- BV32-Hebbian(二进制):76.4%
**其他核心发现:**
- 在相同Bit-Mass下,Float32容器与二进制容器提供几乎相同的准确率。
- 与AdamW之间的剩余差距完全取决于动量与自适应学习率。
- 仅改变算术运算不会改善结果。
每个神经元充当32个二进制决策的容器。因此,传统的神经元视角会导致系统性误判:8个Float神经元在信息上对应256个二进制神经元。
这一见解得到了同一权重矩阵的三种等价描述(神经元视角、位视角和数据视角)的支持。
需要指出的是,这是一项此前未经同行评审的单一研究,标注为未来日期。多个实验室的独立复现仍然至关重要。
尽管如此,该理论为无反向传播的Hebbian更新为何能取得与经典SGD相同的性能提供了一致的解释。
历史上,Hebbian规则长期被认为不稳定。本研究显示,更新公式中的一个简单错误导致了超过65个百分点的性能损失。修正后,二进制方法恰好收敛到普通SGD的水平。
从架构理论角度来看,一个明确的结论是:性能提升需要要么通过更宽的层增加更多位数,要么通过动量与自适应方法更高效地利用现有位数。计算格式本身是次要的。
实验控制程度高:所有训练器使用相同的数据(50,000个MNIST样本)、相同的周期数和相同的架构。仅更新规则不同。这使得效应可以清晰隔离。
**研究的长期意义:**
Bit-Mass理论实现了模型的硬件无关可比性。一个具有64个隐藏神经元的宽Float网络与一个具有2048个神经元的二进制网络具有相同的Bit-Mass。这为模型压缩和专用加速器的开发开辟了新途径。
总之,该工作为高效神经网络的讨论提供了基于事实的贡献。结果以可复现的方式记录,但在能够谈论广泛有效的范式转变之前,仍需进一步的外部验证。
📎 Source 1: https://forward-prop.nhi1.de/
相似文章
可变位宽量化:为“更大但更小”的语言模型学习每组的精度
介绍了可变位宽量化(VBQ),一种训练时的方法,其中每组64个权重通过Gumbel-Softmax松弛学习自己的位宽(1、2、4、8)。VBQ发现了一种异构分配,实现了“更大但更小”的机制,例如,平均位宽1.82的1.31亿参数模型在TinyStories上的困惑度为4.2,击败了5500万FP16模型(困惑度4.4),同时存储减少3.8倍;而1.46B模型在FineWeb-Edu上与593M FP16控制模型表现相当,存储减少约3.7倍。
1位 / 2位 / 三值 / Bitnet 模型 - 更新与跟踪
本文追踪各种低比特AI模型的更新,包括Bonsai、BitCPM等,涵盖性能改进及对llama.cpp等后端的兼容性更新。
那么关于1-bit模型的共识是什么?它仍然是一个幻想吗?
讨论关于Bonsai 8b和27b等1-bit模型的可行性,这些模型在保持功能的同时实现了较小的文件大小,质疑其主要用例和未来。
2-bit QAT 模型发布
关于2位量化感知训练(QAT)在更大规模MoE模型上的潜力的讨论,比较其与4位QAT及三元LLM的性能,并探讨在消费级硬件上的可行性。
思路:在CPU上,解码速度取决于每个token的活跃参数,而非总参数。我的目标是尝试在中等配置PC(无GPU)上以100tok/s运行一个10B模型。
提出了CPU解码速度取决于每个token的活跃参数而非总参数的见解,并提出一个使用三元权重和细粒度MoE的10B参数模型,以在中等配置PC上实现高token速率。报告了沙箱测量结果,显示在8.3M模型上速度从176 tok/s提升到848 tok/s,且质量损失极小。