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

10/07/2022
by   Yuzhi Yang, et al.
0

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.

READ FULL TEXT

page 1

page 4

page 5

page 9

page 11

page 13

research
02/10/2023

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

This paper studies the exploitation of triple polarization (TP) for mult...
research
11/07/2022

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

Signal processing in wireless communications, such as precoding, detecti...
research
09/10/2018

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

Hybrid precoding design can be challenging for broadband millimeter-wave...
research
06/15/2020

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

Optimization theory assisted algorithms have received great attention fo...
research
11/11/2019

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...
research
01/05/2021

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...
research
07/28/2021

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