A DNN Framework on Waveform Design for Over-the-Air Computation

One of the main applications expected to be enabled by next-generation networks is computing. The goal-oriented nature of computing allows the use of different implementation techniques, with over-the-air (OTA) computation being one of the main proposed schemes due to its effective resource management and computational efficiency. In this work, we aim at optimizing the waveform of the system in the presence of intersymbol interference (ISI) and sampling error. To this end, we propose a deep neural network (DNN) framework that generates an optimal waveform that minimizes the mean square error (MSE) of the OTA computation system. To ensure that the generated waveform exhibits the same behavior as other common waveforms, weighted energy and spectrum constraints are included in the loss function of the training phase. To better mitigate ISI, the spectrum constraint integrates the roll-off factor of the waveform, allowing for the generation of different waveforms. Simulation results verify that the desired constraints are met and show a significant performance gain over state-of-the-art waveforms.