A learned constellation is a geometric or probabilistic shaping method where a neural network optimizes the positions and probabilities of constellation points in the I/Q plane to maximize data throughput for a specific channel model, moving beyond fixed QAM schemes. Unlike traditional quadrature amplitude modulation with rigid, evenly spaced lattice points, a learned constellation discovers an irregular, channel-adapted geometry through gradient descent, directly minimizing the bit or symbol error rate for a given signal-to-noise ratio.
Glossary
Learned Constellation

What is Learned Constellation?
A method for optimizing the positions and probabilities of symbols in the I/Q plane using a neural network to maximize data throughput for a specific channel model.
In a typical end-to-end training loop, the constellation points are treated as trainable parameters within a channel autoencoder. The transmitter maps bits to these learned points, the signal passes through a differentiable channel model, and the receiver attempts to decode them. The loss gradient backpropagates through the channel to adjust the I/Q coordinates, often yielding non-intuitive, asymmetric shapes that exploit the specific non-linearities and noise characteristics of the target hardware or propagation environment.
Key Characteristics
A learned constellation replaces fixed QAM grids with a neural network-optimized arrangement of points in the I/Q plane, maximizing mutual information for a specific channel model.
Geometric Shaping
The neural network directly optimizes the complex-valued positions of each constellation point in the I/Q plane. Unlike fixed QAM, points are not constrained to a rectangular grid. The optimizer places points to maximize mutual information under a specific signal-to-noise ratio (SNR) and channel model, often resulting in non-uniform, circular, or spiral arrangements that better match the channel's capacity-achieving distribution.
Probabilistic Shaping
Rather than moving points, the network learns a non-uniform probability mass function over a fixed set of constellation points. A distribution matcher encodes bits into symbols such that low-energy inner points are transmitted more frequently than high-energy outer points. This creates a Gaussian-like symbol distribution, providing a shaping gain of up to 1.53 dB over uniform QAM at high spectral efficiencies.
End-to-End Differentiability
The constellation is treated as a trainable parameter tensor within a larger channel autoencoder. During training, a differentiable channel model—either a mathematical approximation or a neural surrogate—allows gradients from the receiver's loss to flow back through the channel to the constellation points. This enables joint optimization of the constellation and the receiver's decision boundaries using standard stochastic gradient descent.
Channel-Conditioned Adaptation
A single neural network can learn to output different constellations based on the current channel state information (CSI) or estimated SNR. The constellation becomes a function of the channel, morphing its geometry in real-time to maintain optimal throughput. For example, at low SNR, points cluster for robustness; at high SNR, they spread out to maximize rate.
Hardware Impairment Compensation
When the differentiable channel model includes power amplifier non-linearity, phase noise, or I/Q imbalance, the learned constellation automatically pre-distorts to compensate. The resulting point cloud may appear asymmetric or warped, but it maximizes the receiver's ability to decode correctly after the real hardware impairments are applied, outperforming a standard QAM constellation followed by a separate pre-distorter.
Mutual Information Maximization
The training objective is often a mutual information neural estimator (MINE) or a variational lower bound on the channel capacity. Rather than minimizing symbol error rate, the network directly maximizes the amount of information that can be reliably transmitted per channel use. This aligns the optimization with the fundamental Shannon limit, producing constellations that are provably capacity-approaching for non-AWGN channels.
Frequently Asked Questions
Clear answers to common questions about how neural networks optimize constellation geometry and symbol probabilities to surpass the spectral efficiency limits of traditional QAM schemes.
A learned constellation is a set of complex-valued symbol points in the I/Q plane whose positions and/or probabilities are optimized by a neural network to maximize data throughput for a specific channel model, rather than being fixed to a rigid geometric grid like Quadrature Amplitude Modulation (QAM). Unlike QAM, which arranges points on a uniform rectangular lattice, a learned constellation can assume arbitrary, non-uniform geometries. This allows the transceiver to exploit the specific noise and distortion characteristics of the channel—for example, placing points closer together in low-noise regions and farther apart in high-noise regions. The result is a shaping gain that approaches the theoretical Shannon capacity limit more closely than traditional schemes, especially on non-linear or non-Gaussian channels where the assumptions of classical constellation design break down.
Enabling Efficiency, Speed & Accuracy
Intelligent Analysis, Decision & Execution
We build AI systems for teams that need search across company data, workflow automation across tools, or AI features inside products and internal software.
Talk to Us
Search across company data
Give teams answers from docs, tickets, runbooks, and product data with sources and permissions.
Useful when people spend too long searching or get different answers from different systems.

Automate internal workflows
Use AI to route work, draft outputs, trigger actions, and keep approvals and logs in place.
Useful when repetitive work moves across multiple tools and teams.

Add AI to products and internal tools
Build assistants, guided actions, or decision support into the software your team or customers already use.
Useful when AI needs to be part of the product, not a separate tool.
Related Terms
Learned constellations are a core component of end-to-end learned communication systems. The following concepts form the theoretical and architectural foundation for understanding how neural networks optimize geometric symbol placement in the I/Q plane.
Probabilistic Shaping
An optimization technique that assigns a non-uniform probability distribution to constellation points to approach the Shannon capacity limit. Unlike geometric shaping, which moves point locations, probabilistic shaping uses a distribution matcher to transmit low-energy inner symbols more frequently than high-energy outer symbols.
- Achieves shaping gain of up to 1.53 dB on AWGN channels
- Implemented via constant composition distribution matching (CCDM)
- Often combined with learned constellations for joint geometric-probabilistic shaping
- Key enabler for flexible-rate optical and wireless systems
Differentiable Channel Model
A mathematical or neural surrogate model that allows gradients to backpropagate from the receiver loss function through the channel to the transmitter parameters. This is the critical enabler for learning constellation geometries via gradient descent.
- Can be a statistical model (e.g., differentiable AWGN, Rayleigh fading)
- Can be a GAN-based channel surrogate for real-world hardware-in-the-loop training
- Must accurately capture non-linear impairments like PA distortion and phase noise
- The quality of the learned constellation depends directly on channel model fidelity
Geometric Shaping
The direct optimization of constellation point positions in the I/Q plane to maximize mutual information for a specific signal-to-noise ratio and channel model. Unlike probabilistic shaping, geometric shaping modifies the location of points rather than their frequency of use.
- Produces non-regular lattice structures optimized for non-AWGN channels
- Can be combined with neural demappers for joint geometric-demapper optimization
- Particularly effective for non-linear fiber channels and satellite links
- Learned via gradient descent on a mutual information lower bound
Neural Network Demapper
A receiver component that uses a neural network to compute soft bit estimates (log-likelihood ratios) directly from received I/Q symbols. When paired with a learned constellation, the demapper learns a non-linear decision boundary that outperforms classical maximum-likelihood demapping in the presence of hardware impairments.
- Learns to compensate for residual phase noise and I/Q imbalance
- Can be trained jointly with the constellation geometry
- Replaces Euclidean distance-based demapping with learned manifolds
- Critical for extracting full performance from irregular constellations
Mutual Information Neural Estimator
A neural network trained to estimate the mutual information between channel input and output, serving as a differentiable optimization objective for constellation design. MINE provides a tractable lower bound on the achievable information rate without requiring closed-form capacity expressions.
- Uses a dual representation of KL-divergence (Donsker-Varadhan bound)
- Enables optimization of constellations for channels with unknown or intractable capacity
- Can incorporate practical constraints like peak-to-average power ratio (PAPR)
- Replaces heuristic shaping metrics with information-theoretic objectives

About the author
Prasad Kumkar
CEO & MD, Inference Systems
Prasad Kumkar is the CEO & MD of Inference Systems and writes about AI systems architecture, LLM infrastructure, model serving, evaluation, and production deployment. Over 5+ years, he has worked across computer vision models, L5 autonomous vehicle systems, and LLM research, with a focus on taking complex AI ideas into real-world engineering systems.
His work and writing cover AI systems, large language models, AI agents, multimodal systems, autonomous systems, inference optimization, RAG, evaluation, and production AI engineering.
Partnered with leading AI, data, and software stack.
How We Work
Custom AI workflows for your Business
One-fit-all AI don't work for modern businesses. At Inferensys, we aim to understand your business & custom requirements; which we use to define most efficient agentic workflows, the data, and the tools for your business.
01
Review the use case
We understand the task, the users, and where AI can actually help.
Read more02
Pick the right approach
We define what needs search, automation, or product integration.
Read more03
Build the first useful version
We implement the part that proves the value first.
Read more04
Improve from there
We add the checks and visibility needed to keep it useful.
Read moreThe first call is a practical review of your use case and the right next step.
Talk to Us