Physics-Informed Uncertainty-Aware Beamforming for HAPS Massive MIMO under Imperfect CSI

Akram Y. Sarhan, Osamah A. Abdullah, Khalid T. Musri, Hayder Al-Hraishawi

High-altitude platform station (HAPS) massive multiple-input multiple-output (MIMO) systems are expected to support wide-area, low-latency, and energy-efficient connectivity in future non-terrestrial networks. However, Doppler-induced channel aging, finite-rate feedback quantization, packet loss, and estimation noise impair transmitter-side channel state information (CSI), making robust downlink beamforming challenging. In HAPS channels, these impairments are strongly structured by elevation-dependent Rician propagation and line-of-sight (LoS)-dominant geometry, whereas conventional robust beamforming methods often rely on generic uncertainty models and computationally intensive optimization. This paper develops a physics-informed uncertainty-aware beamforming framework for HAPS massive MIMO systems under imperfect CSI. First, a geometry-aware channel and feedback-impairment model is developed, where CSI errors due to aging, quantization, packet loss, and noise are represented through tangent-space ellipsoidal uncertainty sets. Second, a physics-informed variational autoencoder (VAE) exploits the LoS-dominant steering manifold to enhance channel direction information and propagate learned uncertainty through unit-sphere projection. Third, the learned uncertainty representation is embedded into a robust energy-efficiency maximization formulation with probabilistic QoS awareness. To enable scalable online operation, the resulting beamforming policy is approximated using a multi-agent deterministic policy gradient framework with centralized training, decentralized execution, and differentiable power projection. Simulation results show that the proposed framework improves energy efficiency, SINR robustness, outage reliability, convergence behavior, and online runtime compared with imperfect-CSI, SDR-based, and no-VAE baselines.