mlsanityData to 14 Oct 2026

NEURON-Fabric: CXL-Side Low-Bit Gradient Aggregation for Distributed Training

arXiv 2606.15045 ↗First posted 13 Jun 2026cs.DCcited by 0

Not on the map. Why?

Abstract (arXiv)

In large-model distributed training, especially large language model workloads, gradient All-Reduce increasingly stresses the memory and communication path. This paper asks whether a Compute Express Link (CXL) memory controller can aggregate low-bit gradient signals as gradient cache lines pass through it, while preserving a 32-bit floating-point (FP32) path for workloads, layers, or phases that should not use low-bit approximation. We present NEURON-Fabric, a CXL-side controller architecture that performs packed gradient-binary (G-Binary) sign-count aggregation and gradient-ternary (G-Ternary) gated aggregation near CXL memory, with a control interface for selecting low-bit or FP32 paths. Cycle-level timing experiments show that the measured five-cycle low-bit aggregation datapath adds at most 1.67 percent exposed runtime overhead in the full last-level-cache miss regime; under bandwidth pressure, the same compute stage is hidden by CXL service time. Functional tests confirm byte-exact identity read-back, G-Binary sign-count aggregation, and G-Ternary gating. Training checks quantify the communication and accuracy tradeoff: low-bit aggregation remains close to FP32 on CIFAR-10/ResNet-18 and SST-2/DistilBERT, while full-path low-bit aggregation fails on CIFAR-100/ResNet-18. Layer-aware admission identifies the classifier head as sensitive; keeping the head on FP32 while applying low-bit aggregation to the backbone recovers most accuracy and reduces gradient traffic to 3.6-5.4 percent of the FP32 baseline. Hardware synthesis and FPGA place-and-route estimates suggest that the 512-bit aggregation datapath is small enough to be treated as a near-memory datapath extension rather than a separate accelerator-scale block.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 8 of its 29 references that are in the map. First posted on arXiv 13 Jun 2026.

Builds on

8 of 29 references.

References · 8

  1. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  3. 2020Language Models are Few-Shot Learnerscited by 32,419
  4. 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,899
  5. 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
  6. 2018signSGD: Compressed Optimisation for Non-Convex Problemscited by 571
  7. 2017Deep Gradient Compression: Reducing the Communication Bandwidth for Distributed Trainingcited by 478
  8. 2024The Era of 1-bit LLMs: All Large Language Models are in 1.58 Bitscited by 308

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026NEURON-Fabric: Architecture-Runtime Co-Design for Controlled Low-Bit Gradient Communication
  2. 2026ChunkFT: Byte-Streamed Optimization for Memory-Efficient Full Fine-Tuning
  3. 2026Gradient Under Microscope: Benchmarking Resource Utilization of Memory-Efficient Gradient Computation Methods
  4. 2024Beyond Throughput and Compression Ratios: Towards High End-to-end Utility of Gradient Compression
  5. 2026How Weight Encoding Affects Language Model Placement and Performance on the Apple Neural Engine
  6. 2026DynamiQ: Accelerating Gradient Synchronization using Compressed Multi-hop All-reduce
  7. 2026Spectra: Rethinking Optimizers for LLMs Under Spectral Anisotropy
  8. 2026Spike-Aware INT8 Execution for Spiking Language Models on Commodity CPUs