Optimized Power Control for Over-the-Air Computation in Fading Channels

Over-the-air computation (AirComp) of a function (e.g., averaging) has recently emerged as an efficient multiple-access scheme for fast aggregation of distributed data at mobile devices (e.g., sensors) at a fusion center (FC) over wireless channels. To realize reliable AirComp in practice, it is crucial to adaptively control the devices' transmit power for coping with channel distortion to achieve the desired magnitude alignment of simultaneous signals. In this paper, we solve the power control problem. Our objective is to minimize the computation error by jointly optimizing the transmit power at devices and a signal scaling factor (called denoising factor) at the FC, subject to individual average power constraints at devices. The problem is generally non-convex due to the coupling of the transmit powers at devices and denoising factor at the FC. To tackle the challenge, we first consider the special case with static channels, for which we derive the optimal solution in closed form. The derived power control exhibits a threshold-based structure: if the product of the channel quality and power budget for each device, called quality indicator, exceeds an optimized threshold, this device applies channel-inversion power control; otherwise, it performs full power transmission. We proceed to consider the general case with time-varying channels. To solve the more challenging non-convex power control problem, we use the Lagrange-duality method via exploiting its time-sharing property. The derived power control exhibits a regularized channel inversion structure, where the regularization balances the tradeoff between the signal-magnitude alignment and noise suppression. Moreover, for the special case with only one device being power limited, we show that the power control for the power-limited device has an interesting channel-inversion water-filling structure, while those for other devices (with sufficiently large power budgets) reduce to channel-inversion power control. Numerical results show that the derived power control significantly reduces the computation error as compared with the conventional designs.