Abstract:This paper addresses the problem of end-to-end (E2E) design of learning and communication in a task-oriented semantic communication system. In particular, we consider a multi-device cooperative edge inference system over a wireless multiple-input multiple-output (MIMO) multiple access channel, where multiple devices transmit extracted features to a server to perform a classification task. We formulate the E2E design of feature encoding, MIMO precoding, and classification as a conditional mutual information maximization problem. However, it is notoriously difficult to design and train an E2E network that can be adaptive to both the task dataset and different channel realizations. Regarding network training, we propose a decoupled pretraining framework that separately trains the feature encoder and the MIMO precoder, with a maximum a posteriori (MAP) classifier employed at the server to generate the inference result. The feature encoder is pretrained exclusively using the task dataset, while the MIMO precoder is pretrained solely based on the channel and noise distributions. Nevertheless, we manage to align the pretraining objectives of each individual component with the E2E learning objective, so as to approach the performance bound of E2E learning. By leveraging the decoupled pretraining results for initialization, the E2E learning can be conducted with minimal training overhead. Regarding network architecture design, we develop two deep unfolded precoding networks that effectively incorporate the domain knowledge of the solution to the decoupled precoding problem. Simulation results on both the CIFAR-10 and ModelNet10 datasets verify that the proposed method achieves significantly higher classification accuracy compared to various baselines.

Benchmarking End-to-end Learning of MIMO Physical-Layer Communication

M-Net: A Lightweight Network Based on Multilayer Perceptron for Massive MIMO CSI Feedback

Deep Learning Based MIMO Communications

Joint Deep Reinforcement Learning and Unfolding: Beam Selection and Precoding for Mmwave Multiuser MIMO with Lens Arrays

Learning at the Speed of Wireless: Online Real-Time Learning for AI-Enabled MIMO in NextG

End-to-End Learning for Task-Oriented Semantic Communications Over MIMO Channels: An Information-Theoretic Framework

Learning to Modulate for Non-coherent MIMO

End-to-End Deep Learning for TDD MIMO Systems in the 6G Upper Midbands

ML-Based Feedback-Free Adaptive MCS Selection for Massive Multi-User MIMO

Extremely Large-Scale MIMO: Fundamentals, Challenges, Solutions, and Future Directions

Hybrid Beamforming/Combining for Millimeter Wave MIMO: A Machine Learning Approach

Adaptive Spatial Modulation MIMO Based on Machine Learning

A Survey on Deep-Learning based Techniques for Modeling and Estimation of MassiveMIMO Channels

Passive Beamforming Design for Double RIS-muMIMO System: A Machine Learning Perspective

A Fast Deep Unfolding Learning Framework for Robust MU-MIMO Downlink Precoding.

Deep Learning-Based Secure MIMO Communications with Imperfect CSI for Heterogeneous Networks.

Zone-Specific CSI Feedback for Massive MIMO: A Situation-Aware Deep Learning Approach

Deep Learning-Based Pilotless Spatial Multiplexing

Deep Learning Based Uplink Multi-User SIMO Beamforming Design

Millimeter-Wave Networking in the Sky: A Machine Learning and Mean Field Game Approach for Joint Beamforming and Beam-Steering

TC-MIMONet: A Learning-based Transceiver for MIMO Systems with Temporal Correlations.