arrow
Return

Multiple-Masks Error Correction Code Transformer for Short Block Codes

delete2025-07-01
delete0
PRE
AI
S
Seong‐Joon Park
H
Hee-Youl Kwak
S
Sang‐Hyo Kim
S
Sunghwan Kim
Y
Yongjune Kim
J
Jong‐Seon No
DOI:10.1109/JSAC.2025.3559154delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
With the broadening applications of deep learning, neural decoders have emerged as a key research focus, specifically aimed at improving the decoding performance of conventional decoding algorithms. In particular, error correction code transformer (ECCT), which utilizes the transformer architecture, has achieved state-of-the-art performance among neural network-based decoders. We present three technical contributions to significantly enhance the performance of ECCT. First, we propose a novel transformer architecture of ECCT, termed the multiple-masks ECCT (MM ECCT). We employ multiple masked self-attention blocks with different mask matrices in a parallel manner to learn diverse relationships among the codeword bits. Second, we discover that constructing mask matrices based on systematic parity check matrices (PCMs) can make the attention maps sparse, which not only enhances the decoding performance but also reduces computational complexity. Finally, we propose using complementary mask matrices derived from cyclic permutations of the systematic PCM. These complementary mask matrices are specifically designed to enhance the decoding of cyclic codes. Our extensive simulation results show that the proposed MM ECCT architecture with carefully designed mask matrices outperforms the original ECCT by a large margin, achieving state-of-the-art decoding performance among neural decoders. The source code is available at https://github.com/iil-postech/mm-ecct.
Keywords:
Channel coding
error correcting code
error correction code transformer
mask matrix
transformer

Journal

IEEE Journal on Selected Areas in Communications cover
IEEE Journal on Selected Areas in Communications
IF:
17.2
Papers:
6.4K
Citations:
3.1W

Organization

S
Seoul National University
Scholars:
4.9K
Papers: 2.0K
Citations: 6.6W
S
Sungkyunkwan University
Scholars:
2.8K
Papers: 1.1K
Citations: 2
U
University of Ulsan
Scholars:
1.8W
Papers: 1.7W
Citations: 1.4W
K
Kyonggi University
Scholars:
1.5K
Papers: 2.1K
Citations: 2.6K
researcher View more organizations
Cited Papers

Cited Papers

errShare
errSave
Multiple-Bases Belief-Propagation Decoding of High-Density Cyclic Codes
err2010-01-01
err34
errOAAI
errHehn, Thorsten; Huber, Johannes B.; Milenkovic, Olgica; Laendner, Stefan
errShare
errSave
Neural Min-Sum Decoding for Generalized LDPC Codes
err2022-12-01
err2
PREAI
errKwak, Hee-Youl; Kim, Jae-Won; Kim, Yongjune; Kim, Sang-Hyo; No, Jong-Seon
errShare
errSave
errShare
errSave
Deep Learning Methods for Improved Decoding of Linear Codes
err2018-02-01
err373
errOAAI
errNachmani, Eliya; Marciano, Elad; Lugosch, Loren; Gross, Warren J.; Burshtein, David; Be'ery, Yair
errShare
errSave
Deep Learning Based Communication Over the Air
err2018-02-01
err534
errOAAI
errDoerner, Sebastian; Cammerer, Sebastian; Hoydis, Jakob; ten Brink, Stephan
errShare
errSave
Pruning Neural Belief Propagation Decoders
err2020-06-01
err0
errOAAI
errAndreas Buchberger; Christian Hager; Henry D. Pfister; Laurent Schmalen; Alexandre Graell Amat
errShare
errSave
Multiple-Bases Belief-Propagation for Decoding of Short Block Codes
err2007-06-01
err0
PREAI
errThorsten Hehn; Johannes B. Huber; Stefan Laendner; Olgica Milenkovic
errShare
errSave
researcher View more