English

Scalable Mamba-Based Message-Passing Neural Decoder for Error-Correcting Codes

Information Theory 2026-05-12 v1 Machine Learning math.IT

Abstract

Forward error correction is essential for reliable communication over noisy channels. Attention-based model-free neural decoders have shown strong performance for short codes, but their scalability to longer codes is limited by the quadratic memory and computational cost of attention. In this paper, we introduce the Mamba message-passing decoder (MMPD), an attention-free syndrome-based neural decoder for binary linear codes. MMPD retains the Tanner-graph structure of a message-passing decoder by performing local pairwise aggregation along variable-check edges. To enable efficient long-range information propagation, these local updates are combined with bidirectional Mamba state-space blocks. By avoiding dense attention matrices, MMPD scales more favorably for long codes in both memory and computation. Experiments on the (1056, 880) LDPC code show that MMPD achieves a 0.45 dB gain over the state-of-the-art CrossMPT decoder at a specified target bit error rate, while reducing memory consumption by a factor of 1.5. This reduction factor increases substantially for longer codes, demonstrating the applicability of MMPD to scalable neural decoding of practical long codes.

Keywords

Cite

@article{arxiv.2605.10681,
  title  = {Scalable Mamba-Based Message-Passing Neural Decoder for Error-Correcting Codes},
  author = {Rostislav Gusev and Nikita Aleksandrov and Artem Solomkin and Dmitry Artemasov},
  journal= {arXiv preprint arXiv:2605.10681},
  year   = {2026}
}

Comments

This work has been submitted to the IEEE for possible publication