THz UM-MIMO Channel Estimation via a Noise-Conditioned Fixed-Point Network
-
摘要: 针对太赫兹超大规模MIMO系统在混合场传播与AoSA架构下,现有压缩感知与深度展开方法在低信噪(SNR)及噪声统计失配条件下易出现性能饱和与泛化性较差问题。为此,该文提出一种噪声条件化不动点展开网络FPN-NCAS。该方法在保持不动点理论框架的同时,将由重复导频差分获得的粗噪声功率估计值作为条件控制量耦合至非线性恢复过程,从而实现随噪声水平自适应调节的估计策略。在此基础上,设计噪声条件化的块稀疏近端恢复模块Block Shrink,通过噪声感知阈值与平滑切换机制增强混合场稀疏结构恢复能力;构造Token-Gate模块对子阵级特征进行轻量可靠性校准,并构建G-HMTD模块对主恢复后的残余误差进行全局—局部协同细化。仿真结果表明,所提方法在不同SNR条件下均取得更优的归一化均方误差(NMSE)性能;此外,在多种非理想信道与噪声条件下,所提方法仍表现出良好的鲁棒性。Abstract:
Objective Terahertz(THz)ultra-massive multiple-input multiple-output(UM-MIMO)systems are expected to support future high-capacity wireless communications. However, accurate channel estimation remains difficult under hybrid near-/far-field propagation and array-of-subarrays(AoSA)architectures, where limited radio-frequency chains, low signal-to-noise ratio(SNR), noise uncertainty, and structural perturbations degrade compressed observations. Existing compressed sensing, Bayesian inference, and deep unfolding methods usually rely on fixed statistical assumptions, limiting their adaptability across different SNR and mismatch conditions. To address these issues, this paper proposes a noise-conditioned fixed-point network, termed FPN-NCAS(FPN Noise-Conditioned AoSA-aligned Shrinkage), for robust THz UM-MIMO channel estimation. The objective is to improve estimation accuracy, cross-SNR adaptability, robustness, and iterative stability by introducing noise awareness into nonlinear recovery. Methods FPN-NCAS is developed based on an OAMP-style fixed-point unfolding framework. A coarse noise-power estimate is obtained from repeated pilot differences and injected into the nonlinear recovery module as an explicit conditioning variable. After each linear update, the vector-domain estimate is reshaped into an AoSA-aligned tensor to exploit subarray-level structural priors. The nonlinear restoration chain contains three components. Token-Gate performs lightweight subarray reliability pre-calibration to suppress unreliable responses under low-SNR or structurally inconsistent conditions. Block Shrink serves as the core noise-conditioned block-sparse proximal operator, where a smooth dual-threshold mechanism balances strong denoising at low SNR and structure preservation at medium-to-high SNR. G-HMTD(Gated Hybrid Multi-scale Transformer Denoiser)further refines residual errors by combining local multi-scale enhancement and global contextual modeling. Bridge relaxation and nonlinear residual scaling are also introduced to improve inter-stage stability. Results and Discussions Simulation results show that FPN-NCAS consistently outperforms LS, OAMP, ISTA-Net+, FPN-OAMP, and FPN-OTFN over the 0–20 dB SNR range ( Figure 6 ). At SNR = 0 dB, FPN-NCAS achieves approximately 3.0 dB and 1.9 dB NMSE gains over FPN-OAMP and FPN-OTFN, respectively; at SNR = 20 dB, the gains increase to about 4.5 dB and 2.6 dB (Figure 6(a) ). The convergence curves show that FPN-NCAS reaches a stable plateau after about three layers at SNR = 5 dB and about five layers at SNR = 15 dB, indicating stable fixed-point iteration behavior (Figure 6(b) ,Figure 6(c) ). Repeated-pilot analysis shows that four repeated pilot pairs introduce only 3.13% extra pilot overhead while reducing the relative standard deviation of the coarse noise estimate to 25.00%. Under moderate noise-power mismatch, NMSE degradation is generally within 0.3 dB. The method also remains robust under colored Gaussian noise, impulsive noise, field-region shifts, path-number variation, AoSA subarray shuffle, and RF gain/phase mismatch (Figure 7 ,Figure 8 ). Ablation results indicate that Block Shrink is the dominant contributor, while Token-Gate and G-HMTD provide complementary gains through structural calibration and residual refinement (Figure 9 ).Conclusions This paper proposes FPN-NCAS for noise-conditioned fixed-point channel estimation in THz UM-MIMO systems. By integrating repeated-pilot-based noise conditioning, AoSA-aware feature reshaping, Token-Gate calibration, Block Shrink recovery, and G-HMTD refinement, the proposed method improves NMSE performance, robustness, and iterative stability under different SNR and non-ideal conditions. However, the improved performance is accompanied by higher inference complexity. Future work will focus on lightweight implementation and extensions to wideband, multi-user, and hardware-impaired THz communication scenarios. -
Key words:
- THz communications /
- Ultra-Massive MIMO /
- Channel estimation /
- Fixed-Point Network /
- Noise conditioning
-
表 1 部分信道仿真参数
参数 载波频率 $ f $/GHz 子阵数量 $ {M}^{2} $ 天线单元数量 $ {N}^{2} $ 天线间距 $ {d}_{\mathrm{ae}} $/m 子阵间距 $ {d}_{\mathrm{sub}} $/m 导频长度 $ Q $ 总路径数 $ L $ 取值 300 4 256 2×10–4 0.056 128 5 表 2 不同重复导频对下的噪声估计敏感性
$ {\mathrm{Q}}_{1} $ 导频开销 相对标准差 0 dB 5 dB 10 dB 15 dB 20 dB 1 0.78% 50.00% –12.94 –17.57 –21.32 –24.42 –26.58 2 1.56% 35.36% –13.02 –17.64 –21.36 –24.45 –26.63 4 3.13% 25.00% –13.07 –17.68 –21.38 –24.47 –26.66 8 6.25% 17.68% –13.09 –17.69 –21.39 –24.48 –26.67 16 12.50% 12.50% –13.10 –17.70 –21.40 –24.48 –26.67 表 3 不同分布与主路径数条件下的 NMSE 对比
测试项 分布/条件变化 NMSE 相对变化 区域分布外推 混合场→近场 –20.57 dB +0.15 dB 区域分布外推 混合场→远场 –20.64 dB +0.08 dB 主路径数偏移 L=3–4 平均 -21.84 dB –1.22 dB 主路径数基线 L=5 平均 –20.62 dB 0 主路径数偏移 L=6–7 平均 –19.22 dB +1.40 dB 表 4 RF链幅度相位随机扰动下的Token-Gate统计验证
项目 统计对象/方法 均值 标准差 0 dB 5 dB 10 dB 幅度扰动 $ {\alpha }_{i} $ 1.005 0.101 - - - 相位扰动 $ {\phi }_{i} $ 0 rad 0.20 rad - - - NMSE RF mismatch + no-Gate - - -7.665 ± 0.523 -8.931 ± 0.765 -8.856 ± 0.883 NMSE RF mismatch + Gate - - -10.794 ±1.083 -13.001 ±1.892 -13.510 ± 2.501 改善量 no-Gate − Gate - - 3.130 ± 0.563 4.071 ± 1.145 4.654 ± 1.657 表 5 不同模型的复杂度对比
模型 固定迭代层数 可训练参数量/M 计算量/G 平均推理时间/ms OAMP - - 0.065 7.71 ISTA-Net+ 15 0.454 0.181 31.76 FPN-OAMP 15 0.362 0.888 64.89 FPN-OTFN 15 0.179 0.701 150.47 FPN-NCAS 15 0.523 1.764 251.33 -
[1] HE Yuanzhi, MA Chensheng, ZHU Zhongbo, et al. Terahertz communication payload architecture for space information networks: Challenges, key technologies, and trends[J]. IEEE Network, 2026, 40(1): 107–114. [2] BJÖRNSON E, CHAE C B, HEATH JR R W, et al. Towards 6G MIMO: Massive spatial multiplexing, dense arrays, and interplay between electromagnetics and processing[J]. arXiv preprint arXiv: 2401.02844, 2024. doi: 10.48550/arXiv.2401.02844.(查阅网上资料,不确定本条文献类型及格式是否正确,请确认). [3] HAN Chong, CHEN Yuhang, YAN Longfei, et al. Cross far- and near-field wireless communications in terahertz ultra-large antenna array systems[J]. IEEE Wireless Communications, 2024, 31(3): 148–154. [4] LIN Cen and LI G Y L. Terahertz communications: An array-of-subarrays solution[J]. IEEE Communications Magazine, 2016, 54(12): 124–131. [5] SHA Ziyuan and WANG Zhaocheng. Channel estimation and equalization for terahertz receiver with RF impairments[J]. IEEE Journal on Selected Areas in Communications, 2021, 39(6): 1621–1635. [6] TARBOUSH S, ALI A, and AL-NAFFOURI T Y, et al. Cross-field channel estimation for ultra massive-MIMO THz systems[J]. IEEE Transactions on Wireless Communications, 2024, 23(8): 8619–8635. [7] YANG Songjie, PENG Yizhou, LYU Wanting, et al. Near-field channel estimation for extremely large-scale terahertz communications[J]. Science China Information Sciences, 2024, 67(9): 192302. [8] GARG A, SRIVASTAVA S, YADAV N, et al. Angularly sparse channel estimation in dual- wideband Tera-Hertz (THz) hybrid MIMO systems relying on Bayesian learning[J]. IEEE Transactions on Communications, 2024, 72(7): 4384–4400. [9] LI Yuanjian and MADHUKUMAR A S. Hybrid near- and far-field THz UM-MIMO channel estimation: A sparsifying matrix learning-aided bayesian approach[J]. IEEE Transactions on Wireless Communications, 2025, 24(3): 1881–1897. [10] YE Neng, MIAO Sirui, PAN Jianxiong, et al. Artificial intelligence for wireless physical-layer technologies (AI4PHY): A comprehensive survey[J]. IEEE Transactions on Cognitive Communications and Networking, 2024, 10(3): 729–755. [11] GAO Jiabao, CHEN Xiaoming, and LI G Y. Deep unfolding based channel estimation for wideband terahertz near-field massive MIMO systems[J]. Frontiers of Information Technology & Electronic Engineering, 2024, 25(8): 1162–1172. [12] WANG Kuiyu, GAO Zhen, CHEN Sheng, et al. Knowledge and data dual-driven channel estimation and feedback for ultra-massive MIMO systems under hybrid field beam squint effect[J]. IEEE Transactions on Wireless Communications, 2024, 23(9): 11240–11259. [13] HE Hengtao, YU Xianghao, ZHANG Jun, et al. Message passing meets graph neural networks: A new paradigm for massive MIMO systems[J]. IEEE Transactions on Wireless Communications, 2024, 23(5): 4709–4723. [14] YU Wentao, SHEN Yifei, HE Hengtao, et al. An adaptive and robust deep learning framework for THz ultra-massive MIMO channel estimation[J]. IEEE Journal of Selected Topics in Signal Processing, 2023, 17(4): 761–776. [15] HU Zhengdong, HAN Chong, GERSTACKER W, et al. DiffPace: Diffusion-based plug-and-play augmented channel estimation in mmWave and terahertz ultra-massive MIMO systems[J]. IEEE Journal on Selected Areas in Communications, 2026, 44: 2559–2573. [16] DOVELOS K, MATTHAIOU M, NGO H Q, et al. Channel estimation and hybrid combining for wideband terahertz massive MIMO systems[J]. IEEE Journal on Selected Areas in Communications, 2021, 39(6): 1604–1620. [17] ZAMIR S W, ARORA A, KHAN S, et al. Restormer: Efficient transformer for high-resolution image restoration[C]. Proceedings of the 2022 IEEE/CVF Conference on Computer Vision and Pattern Recognition, New Orleans, USA, 2022: 5718–5729. [18] MA Junjie and LI Ping. Orthogonal AMP[J]. IEEE Access, 2017, 5: 2020–2033. [19] ZHANG Jian and GHANEM B. ISTA-Net: Interpretable optimization-inspired deep network for image compressive sensing[C]. Proceedings of the 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, Salt Lake City, USA, 2018: 1828–1837. [20] 于舒娟, 赵阳, 魏玉尧, 等. 超大规模太赫兹系统深度学习信道估计算法[J]. 通信学报, 2025, 46(1): 144–156. -
下载: