English

Interplay Between Belief Propagation and Transformer: Differential-Attention Message Passing Transformer

Information Theory 2025-09-22 v1 Signal Processing math.IT

Abstract

Transformer-based neural decoders have emerged as a promising approach to error correction coding, combining data-driven adaptability with efficient modeling of long-range dependencies. This paper presents a novel decoder architecture that integrates classical belief propagation principles with transformer designs. We introduce a differentiable syndrome loss function leveraging global codebook structure and a differential-attention mechanism optimizing bit and syndrome embedding interactions. Experimental results demonstrate consistent performance improvements over existing transformer-based decoders, with our approach surpassing traditional belief propagation decoders for short-to-medium length LDPC codes.

Keywords

Cite

@article{arxiv.2509.15637,
  title  = {Interplay Between Belief Propagation and Transformer: Differential-Attention Message Passing Transformer},
  author = {Chin Wa Lau and Xiang Shi and Ziyan Zheng and Haiwen Cao and Nian Guo},
  journal= {arXiv preprint arXiv:2509.15637},
  year   = {2025}
}

Comments

6 pages, 4 figures, to be published in ISIT2025