用轻量修正流让1.58比特大模型更稳定且性能提升近一半
Hybrid Gated Flow (HGF): Stabilizing 1.58-bit LLMs via Selective Low-Rank Correction
- 主干用1.58比特三值网络,辅以可学习的低秩浮点修正路径
- 在小数据集上验证损失降至0.9306,恢复了55%的性能差距
- 适合边缘部署、对量化稳定性有要求的研究与工程人员
将大语言模型部署到边缘设备受制于内存墙——内存带宽成为瓶颈。近期1.58比特量化技术(如BitNet b1.58)虽大幅降低内存占用,但相比FP16基准通常导致困惑度下降20-25%。本文提出混合门控流(HGF),采用双流架构:主干为1.58比特三值网络,辅以可学习的低秩FP16修正路径,由自适应门控控制。在TinyStories数据集上,两种训练策略(2500和3500步)下,HGF实现验证损失0.9306,优于BitNet的1.0294,恢复约55%的性能差距(全精度基线为0.8490)。仅增加约12-15%内存开销。此外,实证发现量化具有结构正则化效应:全精度微分注意力基线(Diff_Only)训练不稳定,验证损失超1.68;而三值锚定的HGF全程收敛稳定。初步结果扩展至1.2B和3B参数模型,基于SlimPajama和FineWeb-Edu训练,表明该架构的稳定性与性能恢复能力可线性扩展至生产级语言建模场景。
原文摘要 · Abstract (English)
The deployment of Large Language Models (LLMs) on edge devices is fundamentally constrained by the "Memory Wall" -- a hardware limitation where memory bandwidth, not compute, becomes the bottleneck. Recent 1.58-bit quantization techniques (e.g., BitNet b1.58) dramatically reduce memory footprint but typically incur a perplexity degradation of 20-25% compared to FP16 baselines. In this work, we introduce Hybrid Gated Flow (HGF), a dual-stream architecture that couples a 1.58-bit ternary backbone with a learnable, low-rank FP16 correction path controlled by adaptive gates. Through extensive experiments on the TinyStories dataset across two training regimes (2500 and 3500 steps), we demonstrate that HGF 5.4 achieves a validation loss of 0.9306 compared to BitNet's 1.0294, recovering approximately 55% of the quality gap between pure ternary quantization and the FP16 baseline (0.8490). This recovery is achieved with only ~12-15% memory overhead beyond the ternary backbone. Furthermore, we provide empirical evidence for an emergent phenomenon: quantization as structural regularization. While a full-precision differential attention baseline (Diff_Only) exhibited training instability with validation loss exceeding 1.68, the ternary-anchored HGF maintained robust convergence throughout training. Finally, we report preliminary results extending this architecture to 1.2B and 3B parameter models trained on SlimPajama and FineWeb-Edu. These larger-scale experiments confirm that the architectural stability and quality recovery observed in small-scale proxies scale linearly to production-grade language modeling regimes.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。