An Adaptive and Robust Deep Learning Framework for THz Ultra-Massive MIMO Channel Estimation

Abstract

Terahertz ultra-massive MIMO (THz UM-MIMO) is envisioned as one of the key enablers of 6G wireless networks, for which channel estimation is highly challenging. Traditional analytical estimation methods are no longer effective, as the enlarged array aperture and the small wavelength result in a mixture of far-field and near-field paths, constituting a hybrid-field channel. Deep learning (DL)-based methods, despite the competitive performance, generally lack theoretical guarantees and scale poorly with the size of the array. In this paper, we propose a general DL framework for THz UM-MIMO channel estimation, which leverages existing iterative channel estimators and is with provable guarantees. Each iteration is implemented by a fixed point network (FPN), consisting of a closed-form linear estimator and a DL-based non-linear estimator. The proposed method perfectly matches the THz UM-MIMO channel estimation due to several unique advantages. First, the complexity is low and adaptive. It enjoys provable linear convergence with a low per-iteration cost and monotonically increasing accuracy, which enables an adaptive accuracy-complexity tradeoff. Second, it is robust to practical distribution shifts and can directly generalize to a variety of heavily out-of-distribution scenarios with almost no performance loss, which is suitable for the complicated THz channel conditions. For practical usage, the proposed framework is further extended to wideband THz UM-MIMO systems with beam squint effect. Theoretical analysis and extensive simulation results are provided to illustrate the advantages over the state-of-the-art methods in estimation accuracy, convergence rate, complexity, and robustness.

arXiv comment15 pages, 11 figures, 5 tables, accepted by IEEE Journal of Selected Topics in Signal Processing (JSTSP)

References in corpus (10)

• Message Passing Algorithms for Compressed Sensing
• Spatial- and Frequency-Wideband Effects in Millimeter-Wave Massive MIMO Systems
• Deep CNN-Based Channel Estimation for mmWave Massive MIMO Systems
• On Mean Absolute Error for Deep Neural Network Based Vector-to-Vector Regression
• Beamforming Technologies for Ultra-Massive MIMO in Terahertz Communications
• Deep Equilibrium Models
• Learning to Optimize: A Primer and A Benchmark
• Solving Sparse Linear Inverse Problems in Communication Systems: A Deep Learning Approach With Adaptive Depth
• DDPG-Driven Deep-Unfolding with Adaptive Depth for Channel Estimation with Sparse Bayesian Learning
• Test-Time Training Can Close the Natural Distribution Shift Performance Gap in Deep Learning Based Compressed Sensing

Cited by in corpus (2)

• Bayes-Optimal Unsupervised Learning for Channel Estimation in Near-Field Holographic MIMO
• Recurrent Transformer-Based Near- and Far-Field THz Wideband Channel Estimation for UM-MIMO