Joint Constellation Shaping Using Gradient Descent Approach for MU-MIMO Broadcast Channel

Maxime Vaillant,Alix Jeannerot,Jean-Marie Gorce

2024-08-21

Abstract:We introduce a learning-based approach to optimize a joint constellation for a multi-user MIMO broadcast channel ($T$ Tx antennas, $K$ users, each with $R$ Rx antennas), with perfect channel knowledge. The aim of the optimizer (MAX-MIN) is to maximize the minimum mutual information between the transmitter and each receiver, under a sum-power constraint. The proposed optimization method do neither impose the transmitter to use superposition coding (SC) or any other linear precoding, nor to use successive interference cancellation (SIC) at the receiver. Instead, the approach designs a joint constellation, optimized such that its projection into the subspace of each receiver $k$, maximizes the minimum mutual information $I(W_k;Y_k)$ between each transmitted binary input $W_k$ and the output signal at the intended receiver $Y_k$. The rates obtained by our method are compared to those achieved with linear precoders.

Information Theory,Machine Learning,Networking and Internet Architecture

What problem does this paper attempt to address?

The problem that this paper attempts to solve is: in the multi - user multi - input multi - output (MU - MIMO) broadcast channel, how to optimize the joint constellation to maximize the minimum mutual information, so as to ensure fairness and communication efficiency among all users. Specifically, the author proposes a learning method based on gradient descent to design the joint constellation, instead of relying on traditional techniques such as linear precoding or successive interference cancellation.

### Specific description of the problem

1. **Background and challenges**:
- In the MU - MIMO broadcast channel, the base station (BS) needs to send data to multiple users simultaneously.
- Traditional methods such as linear precoding (MU - LP), zero - forcing filtering (ZF), and minimum mean - square error (MMSE), although effective, may not achieve optimal performance in some cases and may lead to unfairness among users.
- Techniques such as superposition coding (SC) and successive interference cancellation (SIC) can increase capacity, but are complex to implement and assume specific decoders.

2. **Research objectives**:
- Design a new joint constellation optimization method to maximize the minimum mutual information received by each user under a given total power constraint.
- Ensure that this method does not rely on specific coding or decoding techniques and has broader applicability and flexibility.

3. **Solutions**:
- Propose a learning method based on gradient descent to directly optimize the positions of joint constellation points.
- Use the Adam variant in the stochastic gradient descent (SGD) algorithm for optimization to ensure that a local optimal solution can be found in each iteration.
- Verify the effectiveness of this method through numerical simulation and compare it with the traditional MU - LP method.

### Main contributions

1. **Joint constellation design**:
- Design a joint constellation that can map the input symbols of all users to constellation points simultaneously.
- Maximize the minimum mutual information to ensure fairness among all users.

2. **Optimization algorithm**:
- Propose an optimization algorithm based on stochastic gradient descent that can find the optimal constellation point positions while satisfying the power constraint.
- Does not rely on any specific coding or decoding techniques and has higher flexibility.

3. **Performance evaluation**:
- Verify the performance of this method under different signal - to - noise ratio (SNR) conditions through numerical simulation.
- The results show that this method outperforms the traditional MU - LP method in all SNR ranges, especially with a significant improvement in the minimum mutual information.

### Summary

This paper aims to optimize the joint constellation in the MU - MIMO broadcast channel by introducing a learning method based on gradient descent, thereby maximizing the minimum mutual information and ensuring fairness and communication efficiency among all users. This method not only improves the overall performance of the system but also shows the advantages of learning methods in nonlinear optimization.