# A Study on the Robustness of a DNN Under Scenario Shifts for Power Control in Cell-Free Massive MIMO

## TL;DR

This paper studies how well a simple deep learning model for power control in wireless networks works under different scenarios, showing it performs robustly even when conditions change.

## Contribution

The paper introduces a low-complexity DNN for power control in cell-free massive MIMO that demonstrates strong robustness across diverse scenarios.

## Key findings

- The DNN shows strong robustness, especially for sum SE maximization with D statistics below 0.05.
- The model maintains performance across varying access points, users, and propagation environments.
- A reproducible framework and dataset are provided for future research in ML-based power control.

## Abstract

The emergence of 6G wireless networks presents new challenges, for which cell-free massive MIMO combined with machine learning (ML) offers a promising solution. A key requirement for practical deployment is the generalizability of ML models—their ability to maintain robust performance across varying propagation conditions, user distributions, and network topologies. However, achieving generalizability typically demands large, diverse training datasets and high model complexity, which can hinder practical feasibility. This study analyzes the robustness of a low-complexity deep neural network (DNN) trained for power control under a single network configuration. The model’s robustness is assessed by testing it across a wide range of unseen scenarios, including changes in the number of access points, user equipment, and propagation environments. The DNN is trained to emulate three power control schemes: max-min spectral efficiency (SE) fairness, sum SE maximization, and fractional power control. To rigorously evaluate robustness, we compare the cumulative distribution functions of performance metrics quantitatively using the Kolmogorov–Smirnov test. Results show strong robustness, particularly for the sum SE scheme, with D statistics below 0.05 and p-values above 0.001. This work provides a reproducible framework and dataset to support further research into practical ML-based power control in cell-free massive MIMO systems.