Policy-Guided MCTS for near Maximum-Likelihood Decoding of Short Codes

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Tian, Y., Yue, C., Cheng, P., Pang, G., Vucetic, B., Li, Y.
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917075893616640
author Tian, Y.
Yue, C.
Cheng, P.
Pang, G.
Vucetic, B.
Li, Y.
author_facet Tian, Y.
Yue, C.
Cheng, P.
Pang, G.
Vucetic, B.
Li, Y.
contents In this paper, we propose a policy-guided Monte Carlo Tree Search (MCTS) decoder that achieves near maximum-likelihood decoding (MLD) performance for short block codes. The MCTS decoder searches for test error patterns (TEPs) in the received information bits and obtains codeword candidates through re-encoding. The TEP search is executed on a tree structure, guided by a neural network policy trained via MCTS-based learning. The trained policy guides the decoder to find the correct TEPs with minimal steps from the root node (all-zero TEP). The decoder outputs the codeword with maximum likelihood when the early stopping criterion is satisfied. The proposed method requires no Gaussian elimination (GE) compared to ordered statistics decoding (OSD) and can reduce search complexity by 95\% compared to non-GE OSD. It achieves lower decoding latency than both OSD and non-GE OSD at high SNRs.
format Preprint
id arxiv_https___arxiv_org_abs_2511_09054
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Policy-Guided MCTS for near Maximum-Likelihood Decoding of Short Codes
Tian, Y.
Yue, C.
Cheng, P.
Pang, G.
Vucetic, B.
Li, Y.
Information Theory
In this paper, we propose a policy-guided Monte Carlo Tree Search (MCTS) decoder that achieves near maximum-likelihood decoding (MLD) performance for short block codes. The MCTS decoder searches for test error patterns (TEPs) in the received information bits and obtains codeword candidates through re-encoding. The TEP search is executed on a tree structure, guided by a neural network policy trained via MCTS-based learning. The trained policy guides the decoder to find the correct TEPs with minimal steps from the root node (all-zero TEP). The decoder outputs the codeword with maximum likelihood when the early stopping criterion is satisfied. The proposed method requires no Gaussian elimination (GE) compared to ordered statistics decoding (OSD) and can reduce search complexity by 95\% compared to non-GE OSD. It achieves lower decoding latency than both OSD and non-GE OSD at high SNRs.
title Policy-Guided MCTS for near Maximum-Likelihood Decoding of Short Codes
topic Information Theory
url https://arxiv.org/abs/2511.09054