Hybrid Gated Flow (HGF): Stabilizing 1.58-bit LLMs via Selective Low-Rank Correction
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 6 of its 18 references that are in the map. First posted on arXiv 5 Feb 2026.
Builds on
6 of 18 references.
References · 6
- 2017Attention Is All You Needcited by 60,361
- 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,418
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
- 2022GPTQ: Accurate Post-Training Quantization for Generative Pre-trained Transformerscited by 1,919
- 2019Generating Long Sequences with Sparse Transformerscited by 1,546
- 2023BitNet: Scaling 1-bit Transformers for Large Language Modelscited by 189
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026APreQEL: Adaptive Mixed Precision Quantization For Edge LLMs
- 2024Optimizing Large Language Models through Quantization: A Comparative Analysis of PTQ and QAT Techniques
- 2025A Systematic Evaluation of On-Device LLMs: Quantization, Performance, and Resources
- 2026ScaleBITS: Scalable Bitwidth Search for Hardware-Aligned Mixed-Precision LLMs
- 2026TOM: A Ternary Read-only Memory Accelerator for LLM-powered Edge Intelligence
- 2025TeLLMe: An Energy-Efficient Ternary LLM Accelerator for Prefilling and Decoding on Edge FPGAs
- 2026Recover-LoRA for Aggressive Quantization: Reclaiming Accuracy in 2-Bit Language Models via Low-Rank Adaptation with Knowledge Distillation on Synthetic Data
- 2026RAMP: Reinforcement Adaptive Mixed Precision Quantization for Efficient On Device LLM Inference