End-to-End Learning of Probabilistic Constellation Shaping through Importance Sampling

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Chimmalgi, Shrinivas, Schmalen, Laurent, Aref, Vahid
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912440113954816
author Chimmalgi, Shrinivas
Schmalen, Laurent
Aref, Vahid
author_facet Chimmalgi, Shrinivas
Schmalen, Laurent
Aref, Vahid
contents Probabilistic constellation shaping enables easy rate adaption and has been proven to reduce the gap to Shannon capacity. Constellation point probabilities are optimized to maximize either the mutual information or the bit-wise mutual information. The optimization problem is however challenging even for simple channel models. While autoencoder-based machine learning has been applied successfully to solve this problem [1], it requires manual computation of additional terms for the gradient which is an error-prone task. In this work, we present novel loss functions for autoencoder-based learning of probabilistic constellation shaping for coded modulation systems using automatic differentiation and importance sampling. We show analytically that our proposed approach also uses exact gradients of the constellation point probabilities for the optimization. In simulations, our results closely match the results from [1] for the additive white Gaussian noise channel and a simplified model of the intensity-modulation direct-detection channel.
format Preprint
id arxiv_https___arxiv_org_abs_2506_16098
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle End-to-End Learning of Probabilistic Constellation Shaping through Importance Sampling
Chimmalgi, Shrinivas
Schmalen, Laurent
Aref, Vahid
Information Theory
Signal Processing
Probabilistic constellation shaping enables easy rate adaption and has been proven to reduce the gap to Shannon capacity. Constellation point probabilities are optimized to maximize either the mutual information or the bit-wise mutual information. The optimization problem is however challenging even for simple channel models. While autoencoder-based machine learning has been applied successfully to solve this problem [1], it requires manual computation of additional terms for the gradient which is an error-prone task. In this work, we present novel loss functions for autoencoder-based learning of probabilistic constellation shaping for coded modulation systems using automatic differentiation and importance sampling. We show analytically that our proposed approach also uses exact gradients of the constellation point probabilities for the optimization. In simulations, our results closely match the results from [1] for the additive white Gaussian noise channel and a simplified model of the intensity-modulation direct-detection channel.
title End-to-End Learning of Probabilistic Constellation Shaping through Importance Sampling
topic Information Theory
Signal Processing
url https://arxiv.org/abs/2506.16098