Learning Beamforming in Cell-Free Massive MIMO ISAC Systems
Umut Demirhan, Ahmed Alkhateeb
TL;DR
This work tackles joint beamforming for ISAC in cell-free massive MIMO, where many distributed APs create complex coupling between communication and sensing. It introduces a heterogeneous GNN that models APs, UEs, and sensing targets as node types with two edge types to capture communication and sensing channels, enabling scalable beamforming predictions without full retraining when network size changes. The method uses unsupervised training with edge embeddings initialized from channel data and a power-normalized final output to maximize the objective $\sum_{u} R_u + \beta_s \log_2(1 + {\rm SNR}^{(s)})$, achieving near-optimal communication performance and favorable sensing trade-offs. Results demonstrate strong generalization to different numbers of APs and robustness to network changes, highlighting the approach as a practical, scalable ML-based solution for real-time JSC beamforming in cell-free ISAC systems.
Abstract
Beamforming design is critical for the efficient operation of integrated sensing and communication (ISAC) MIMO systems. ISAC beamforming design in cell-free massive MIMO systems, compared to colocated MIMO systems, is more challenging due to the additional complexity of the distributed large number of access points (APs). To address this problem, this paper first shows that graph neural networks (GNNs) are a suitable machine learning framework. Then, it develops a novel heterogeneous GNN model inspired by the specific characteristics of the cell-free ISAC MIMO systems. This model enables the low-complexity scaling of the cell-free ISAC system and does not require full retraining when additional APs are added or removed. Our results show that the proposed architecture can achieve near-optimal performance, and applies well to various network structures.
