用张量分解实现高维通信系统的高效贝叶斯推断
A Tensor-Train Framework for Bayesian Inference in High-Dimensional Systems: Applications to MIMO Detection and Channel Decoding
- 将联合后验概率表示为低秩张量列车格式,压缩存储并加速计算
- 在不同信噪比下接近最优误码率,仅需较小张量秩
- 适用于多输入多输出检测与纠错码软判决,适合通信系统设计者
高维离散输入加性噪声模型中的贝叶斯推断是通信系统的核心挑战,因所需联合后验概率质量函数的支持集随未知变量数呈指数增长。本文提出一种张量列车(TT)框架,实现离散输入加性噪声模型中可计算、近似最优的贝叶斯推断。核心思想是联合对数后验概率质量函数在TT格式下具有精确低秩表示,支持紧凑存储与高效运算。为获取符号级后验边际,我们设计了一种实用推断方法,利用截断泰勒展开初始化的TT-交叉算法逼近后验指数。通过两个典型通信问题验证方法通用性:加性白高斯噪声下的线性观测模型(用于多输入多输出检测),以及二进制线性分组码在二进制输入高斯信道上的软判决译码。数值结果表明,在广泛信噪比范围内达到近最优误码率性能,且仅需适中的张量秩。这凸显了张量网络方法在通信系统高效贝叶斯推断中的潜力。
原文摘要 · Abstract (English)
Bayesian inference in high-dimensional discrete-input additive noise models is a fundamental challenge in communication systems, as the support of the required joint a posteriori probability (APP) mass function grows exponentially with the number of unknown variables. In this work, we propose a tensor-train (TT) framework for tractable, near-optimal Bayesian inference in discrete-input additive noise models. The central insight is that the joint log-APP mass function admits an exact low-rank representation in the TT format, enabling compact storage and efficient computations. To recover symbol-wise APP marginals, we develop a practical inference procedure that approximates the exponential of the log-posterior using a TT-cross algorithm initialized with a truncated Taylor-series. To demonstrate the generality of the approach, we derive explicit low-rank TT constructions for two canonical communication problems: the linear observation model under additive white Gaussian noise (AWGN), applied to multiple-input multiple-output (MIMO) detection, and soft-decision decoding of binary linear block error correcting codes over the binary-input AWGN channel. Numerical results show near-optimal error-rate performance across a wide range of signal-to-noise ratios while requiring only modest TT ranks. These results highlight the potential of tensor-network methods for efficient Bayesian inference in communication systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。