Over-the-Air Split Machine Learning in Wireless MIMO Networks

by   Yuzhi Yang, et al.

In split machine learning (ML), different partitions of a neural network (NN) are executed by different computing nodes, requiring a large amount of communication cost. To ease communication burden, over-the-air computation (OAC) can efficiently implement all or part of the computation at the same time of communication. Based on the proposed system, the system implementation over wireless network is introduced and we provide the problem formulation. In particular, we show that the inter-layer connection in a NN of any size can be mathematically decomposed into a set of linear precoding and combining transformations over MIMO channels. Therefore, the precoding matrix at the transmitter and the combining matrix at the receiver of each MIMO link, as well as the channel matrix itself, can jointly serve as a fully connected layer of the NN. The generalization of the proposed scheme to the conventional NNs is also introduced. Finally, we extend the proposed scheme to the widely used convolutional neural networks and demonstrate its effectiveness under both the static and quasi-static memory channel conditions with comprehensive simulations. In such a split ML system, the precoding and combining matrices are regarded as trainable parameters, while MIMO channel matrix is regarded as unknown (implicit) parameters.


page 1

page 4

page 5

page 9

page 11

page 13


Channel Modeling and Multi-User Precoding for Tri-Polarized Holographic MIMO Communications

This paper studies the exploitation of triple polarization (TP) for mult...

Extremely-Fast, Energy-Efficient Massive MIMO Precoding with Analog RRAM Matrix Computing

Signal processing in wireless communications, such as precoding, detecti...

Machine Learning Based Hybrid Precoding for MmWave MIMO-OFDM with Dynamic Subarray

Hybrid precoding design can be challenging for broadband millimeter-wave...

Iterative Algorithm Induced Deep-Unfolding Neural Networks: Precoding Design for Multiuser MIMO Systems

Optimization theory assisted algorithms have received great attention fo...

Hybrid Precoding for Multi-User Millimeter Wave Massive MIMO Systems: A Deep Learning Approach

In multi-user millimeter wave (mmWave) multiple-input-multiple-output (M...

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

The millimeter wave (mmWave) multiuser multiple-input multiple-output (M...

Finding Better Precoding in Massive MIMO using Optimization Approach

The paper studies the multi-user precoding problem as a non-convex optim...

Please sign up or login with your details

Forgot password? Click here to reset