Towards Optimal Constellation Design for Digital Over-the-Air Computation
Saeed Razavikia, Deniz Gündüz, Carlo Fischione
TL;DR
This work addresses the problem of computing sums over a wireless MAC using digital modulation by designing optimal two-dimensional constellations under a power constraint to minimize the mean-squared error. It develops a rigorous optimization framework for encoder/decoder design, derives nonlinear optimality conditions for ML and MAP decoding, and proves uniqueness of the solutions. In the high-SNR regime, it provides closed-form solutions based on a generalized Lambert function, offering deep insight into the constellation structure and asymptotic behavior, including convergence of ML and MAP. Extensions to heavy-tailed noise, real-domain computation via hybrid modulation, and higher-dimensional grids demonstrate the framework’s broad applicability, and numerical results show substantial performance gains over standard QAM-based schemes. The approach thus enables robust, hardware-friendly digital OAC with theoretically grounded design rules and practical implications for edge computing and distributed inference.
Abstract
Over-the-air computation (OAC) has emerged as a key technique for efficient function computation over multiple-access channels (MACs) by exploiting the waveform superposition property of the wireless domain. While conventional OAC methods rely on analog amplitude modulation, their performance is often limited by noise sensitivity and hardware constraints, motivating the use of digital modulation schemes. This paper proposes a novel digital modulation framework optimized for computation over additive white Gaussian noise (AWGN) channels. The design is formulated as an additive mapping problem to determine the optimal constellation that minimizes the mean-squared error (MSE) under a transmit power constraint. We express the optimal constellation design as a system of nonlinear equations and establish the conditions guaranteeing the uniqueness of its solution. In the high signal-to-noise-ratio (SNR) regime, we derive closed-form expressions for the optimal modulation parameters using the generalized Lambert function, providing analytical insight into the system's behavior. Furthermore, we discuss extensions of the framework to higher-dimensional grids corresponding to multiple channel uses, to non-Gaussian noise models, and to computation over real-valued domains via hybrid digital-analog modulation.
