VLDB 2026 Research / reviewers in the wild / expert
Hengtao He
dblp:194/2383
· DBLP profile ↗
27ranked-venue papers
6as first author
25since 2021 · last 2026
0000-0002-4659-6941ORCID · verified
Domains — the database's venue-derived domains; a paper can count in several
Computer networks · 23 · 5 first-author · 23 since 2021Graphics, computer vision, multimedia, augmented reality and games · 1Applied, interdisciplinary, general and emerging computing · 1 · 1 first-author
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2026 | DL-Aided Super-Resolution Beam Alignment for Low-Overhead mmWave Massive MIMO
Weijie Jin, Jing Zhang 0031, Hengtao He, Chao-Kai Wen, Shi Jin 0002, Jing Jina, Ziye Shi |
ICC | 3 |
| 2026 | Unlocking Bistatic Target Detection for ISAC: Synergizing Deterministic Pilots and Unknown Random Data PayloadsabstractIntegrated sensing and communications (ISAC) is a key enabler for 6G applications such as drone surveillance, urban air mobility, and low-altitude logistics. However, the hybrid ISAC signal, which comprises deterministic pilots and random data payloads, poses challenges for target detection, since 1) these components jointly affect both the mean and variance of the received signal, and 2) the random data payloads are typically unknown to the sensing receiver in bistatic systems. To address these, we develop a generalized likelihood ratio test (GLRT)-based detector that exploits the known pilots and the statistical properties of the unknown payloads. Given the exact performance is analytically intractable, an asymptotic analysis of the false alarm probability is conducted. Simulation results validate the theoretical derivations and demonstrate the superiority of the proposed detector, which highlights the importance of tailored ISAC detection that fully leverages data payload resources. Lei Xie 0009, Hengtao He, Shenghui Song 0001, Shi Jin 0002, Khaled Ben Letaief |
ICC | 2 |
| 2026 | Joint Beamforming and Antenna Position Optimization for Fluid Antenna-Assisted MU-MIMO Networks
Tianyi Liao, Wei Guo 0030, Hengtao He, Shenghui Song 0001, Jun Zhang 0004, Khaled Ben Letaief |
IEEE J. Sel. Areas Commun. | 3 |
| 2025 | Fluid Antenna-Assisted MU-MIMO Systems with Decentralized Baseband Processing
Tianyi Liao, Wei Guo 0030, Hengtao He, Shenghui Song 0001, Jun Zhang 0004, Khaled Ben Letaief |
GLOBECOM | 3 |
| 2025 | Accurate and Fast Channel Estimation for Fluid Antenna Systems with Diffusion ModelsabstractFluid antenna systems (FAS) offer enhanced spatial diversity for next-generation wireless systems. However, acquiring accurate channel state information (CSI) remains challenging due to the large number of reconfigurable ports and the limited availability of radio-frequency (RF) chains— particularly in high-dimensional FAS scenarios. To address this challenge, we propose an efficient posterior sampling-based channel estimator that leverages a diffusion model (DM) with a simplified U-Net architecture to capture the spatial correlation structure of two-dimensional FAS channels. The DM is initially trained offline in an unsupervised way and then applied online as a learned implicit prior to reconstruct CSI from partial observations via posterior sampling through a denoising diffusion restoration model (DDRM). To accelerate the online inference, we introduce a skipped sampling strategy that updates only a subset of latent variables during the sampling process, thereby reducing the computational cost with minimal accuracy degradation. Simulation results demonstrate that the proposed approach achieves significantly higher estimation accuracy and over 20× speedup compared to state-of-the-art compressed sensing-based methods, highlighting its potential for practical deployment in high-dimensional FAS. Erqiang Tang, Wei Guo 0030, Hengtao He, Shenghui Song 0001, Jun Zhang 0004, Khaled Ben Letaief |
GLOBECOM | 3 |
| 2025 | Remote Training in Task-Oriented Communication: Supervised or Self-Supervised with Fine-Tuning?abstractTask-oriented communication focuses on extracting and transmitting only the information relevant to specific tasks, effectively minimizing communication overhead. Most existing methods prioritize reducing this overhead during inference, often assuming feasible local training or minimal training communication resources. However, in real-world wireless systems with dynamic connection topologies, training models locally for each new connection is impractical, and task-specific information is often unavailable before establishing connections. Therefore, minimizing training overhead and enabling label-free, task-agnostic pre-training before the connection establishment are essential for effective task-oriented communication. In this paper, we tackle these challenges by employing a mutual information maximization approach grounded in self-supervised learning and information-theoretic analysis. We propose an efficient strategy that pre-trains the transmitter in a task-agnostic and label-free manner, followed by joint fine-tuning of both the transmitter and receiver in a task-specific, label-aware manner. Simulation results show that our proposed method reduces training communication overhead to about half that of full-supervised methods using the SGD optimizer, demonstrating significant improvements in training efficiency. Hengtao He, Shenghui Song 0001, Jun Zhang 0004, Khaled Ben Letaief |
ICC | 3 |
| 2025 | Distributed on-Device LLM Inference with Over-the-Air ComputationabstractLarge language models (LLMs) have achieved remarkable success across various artificial intelligence tasks. However, their enormous sizes and computational demands pose significant challenges for the deployment on edge devices. To address this issue, we present a distributed on-device LLM inference framework based on tensor parallelism, which partitions neural network tensors (e.g., weight matrices) of LLMs among multiple edge devices for collaborative inference. Nevertheless, tensor parallelism involves frequent all-reduce operations to aggregate intermediate layer outputs across participating devices during inference, resulting in substantial communication overhead. To mitigate this bottleneck, we propose an over-the-air computation method that leverages the analog superposition property of wireless multipleaccess channels to facilitate fast all-reduce operations. To minimize the average transmission mean-squared error, we investigate joint model assignment and transceiver optimization, which can be formulated as a mixed-timescale stochastic non-convex optimization problem. Then, we develop a mixed-timescale algorithm leveraging semidefinite relaxation and stochastic successive convex approximation methods. Comprehensive simulation results will show that the proposed approach significantly reduces inference latency while improving accuracy. This makes distributed ondevice LLM inference practical for resource-constrained edge devices. Hengtao He, Shenghui Song 0001, Jun Zhang 0004, Khaled Ben Letaief |
ICC | 2 |
| 2025 | Multimodal Deep Learning-Empowered Beam Prediction in Future THz ISAC SystemsabstractIntegrated sensing and communication (ISAC) systems operating at terahertz (THz) bands are envisioned to enable both ultra-high data-rate communication and precise environmental awareness for next-generation wireless networks. However, the narrow width of THz beams makes them prone to misalignment and necessitates frequent beam prediction in dynamic environments. Multimodal sensing, which integrates complementary modalities such as camera images, positional data, and radar measurements, has recently emerged as a promising solution for proactive beam prediction. Nevertheless, existing multimodal approaches typically employ static fusion architectures that cannot adjust to varying modality reliability and contributions, thereby degrading predictive performance and robustness. To address this challenge, we propose a novel and efficient multimodal mixtureof-experts (MoE) deep learning framework for proactive beam prediction in THz ISAC systems. The proposed multimodal MoE framework employs multiple modality-specific expert networks to extract representative features from individual sensing modalities, and dynamically fuses them using adaptive weights generated by a gating network according to the instantaneous reliability of each modality. Simulation results in realistic vehicle-to-infrastructure (V2I) scenarios demonstrate that the proposed MoE framework outperforms traditional static fusion methods and unimodal baselines in terms of prediction accuracy and adaptability, highlighting its potential in practical THz ISAC systems with ultra-massive multiple-input multiple-output (MIMO). Hengtao He, Shenghui Song 0001, Jun Zhang 0004, Khaled Ben Letaief |
PIMRC | 3 |
| 2025 | Fractional Delay and Doppler Estimation for OTFS Systems with Doppler Squint EffectabstractOrthogonal time frequency space (OTFS) modulation is a promising technology for mitigating severe Doppler effects in high-mobility scenarios. However, existing OTFS channel estimation methods neglect the Doppler Squint Effect (DSE), which incurs serious performance loss. In this paper, we propose a channel estimation algorithm based on Newton's method to accurately estimate fractional delay and Doppler in OTFS systems with DSE. In particular, we obtain the maximum delay and Doppler grid spacing for codebook design to guarantee the convergence of the algorithm. Additionally, we derive the Cramér-Rao lower bound (CRLB) for testing channel parameter estimation performance of our proposed algorithm. Simulation results demonstrate that our proposed algorithm outperforms the orthogonal matching pursuit (OMP) algorithm in terms of normalized mean square error (NMSE), surpassing the Newtonized OMP algorithm with traditional dictionary matrix and approaching the CRLB performance. Meiying Zhang, Ruoxiao Cao, Hengtao He, Shenghui Song 0001, Jun Zhang 0004, Khaled Ben Letaief |
WCNC | 4 |
| 2025 | Tackling Distribution Shifts in Task-Oriented Communication With Information BottleneckabstractTask-oriented communication aims to extract and transmit task-relevant information to significantly reduce the communication overhead and transmission latency. However, theunpredictabledistribution shifts between training and test data, includingdomain shiftandsemantic shift, can dramatically undermine the system performance. In order to tackle these challenges, it is crucial to ensure that the encoded features can generalize todomain-shifteddata and detectsemantic-shifteddata, while remaining compact for transmission. In this paper, we propose a novel approach based on the information bottleneck (IB) principle and invariant risk minimization (IRM) framework. The proposed method aims to extract compact and informative features that possess high capability for effectivedomain-shift generalizationand accuratesemantic-shift detectionwithout any knowledge of the test data during training. Specifically, we propose an invariant feature encoding approach based on the IB principle and IRM framework fordomain-shiftgeneralization, which aims to find the causal relationship between the input data and task result by minimizing the complexity and domain dependence of the encoded feature. Furthermore, we enhance the task-oriented communication with the label-dependent feature encoding approach forsemantic-shift detectionwhich achieves joint gains in IB optimization and detection performance. To avoid the intractable computation of the IB-based objective, we leverage variational approximation to derive a tractable upper bound for optimization. Extensive simulation results on image classification tasks demonstrate that the proposed scheme outperforms state-of-the-art approaches and achieves a better rate-distortion tradeoff. Jiawei Shao, Hengtao He, Shenghui Song 0001, Jun Zhang 0004, Khaled Ben Letaief |
IEEE J. Sel. Areas Commun. | 3 |
| 2025 | Toward Real-Time Edge AI: Model-Agnostic Task-Oriented Communication With Visual Feature AlignmentabstractTask-oriented communication presents a promising approach to improve the communication efficiency of edge inference systems by optimizing learning-based modules to extract and transmit relevant task information. However, real-time applications face practical challenges, such as incomplete coverage and potential malfunctions of edge servers. This situation necessitates cross-model communication between different inference systems, enabling edge devices from one service provider to collaborate effectively with edge servers from another. Independent optimization of diverse edge systems often leads to incoherent feature spaces, which hinders the cross-model inference for existing task-oriented communication. To facilitate and achieve effective cross-model task-oriented communication, this study introduces a novel framework that utilizes shared anchor data across diverse systems. This approach addresses the challenge of feature alignment in both server-based and on-device scenarios. In particular, by leveraging the linear invariance of visual features, we propose efficient server-based feature alignment techniques to estimate linear transformations using encoded anchor data features. For on-device alignment, we exploit the angle-preserving nature of visual features and propose to encode relative representations with anchor data to streamline cross-model communication without additional alignment procedures during the inference. The experimental results on computer vision benchmarks demonstrate the superior performance of the proposed feature alignment approaches in cross-model task-oriented communications. The runtime and computation overhead analysis further confirm the effectiveness of the proposed feature alignment approaches in real-time applications. Songjie Xie, Hengtao He, Shenghui Song 0001, Jun Zhang 0004, Khaled Ben Letaief |
IEEE J. Sel. Areas Commun. | 2 |
| 2025 | Cell-Free Massive MIMO Detection: A Distributed Expectation Propagation ApproachabstractCell-free massive MIMO is one of the core technologies for next-generation wireless networks. It is expected to bring enormous benefits, including ultra-high reliability, data throughput, energy efficiency, and uniform coverage. However, the radically distributed architecture of cell-free massive MIMO necessitates new paradigms for transceiver design, especially by exploiting efficient distributed processing algorithms. In this paper, we propose a distributed expectation propagation (EP) detector for cell-free massive MIMO, which consists of two modules: a nonlinear module at the central processing unit (CPU) and a linear module at each access point (AP). The turbo principle in iterative channel decoding is utilized to compute and pass the extrinsic information between the two modules. An analytical framework is provided to characterize the asymptotic performance of the proposed EP detector with a large number of antennas. Furthermore, a distributed iterative channel estimation and data detection (ICD) algorithm is developed to handle the practical scenario with imperfect channel state information (CSI). Simulation results will show that the proposed method outperforms existing detectors for cell-free massive MIMO systems in terms of the bit-error rate and the developed theoretical analysis can be utilized as an asymptotic lower bound. Finally, it is shown that with imperfect CSI, the proposed ICD algorithm can significantly improve the system performance and reduce the pilot overhead. Hengtao He, Xianghao Yu, Jun Zhang 0004, Shenghui Song 0001, Ross Murch, Khaled Ben Letaief |
IEEE Trans. Mob. Comput. | 1 |
| 2025 | Low-Complexity CSI Feedback for FDD Massive MIMO Systems via Learning to OptimizeabstractIn frequency-division duplex (FDD) massive multiple-input multiple-output (MIMO) systems, the growing number of base station antennas leads to prohibitive feedback overhead for downlink channel state information (CSI). To address this challenge, state-of-the-art (SOTA) fully data-driven deep learning (DL)-based CSI feedback schemes have been proposed. However, the high computational complexity and memory requirements of these methods hinder their practical deployment on resource-constrained devices like mobile phones. To solve the problem, we propose a model-driven DL-based CSI feedback approach by integrating the wisdom of compressive sensing and learning to optimize (L2O). Specifically, only a linear learnable projection is adopted at the encoder side to compress the CSI matrix, thereby significantly cutting down the user-side complexity and memory expenditure. On the other hand, the decoder incorporates two specially designed components, i.e., a learnable sparse transformation and an element-wise L2O reconstruction module. The former is developed to learn a sparse basis for CSI within the angular domain, which explores channel sparsity effectively. The latter shares the same long short term memory (LSTM) network across all elements of the optimization variable, eliminating the retraining cost when problem scale changes. Simulation results show that the proposed method achieves a comparable performance with the SOTA CSI feedback scheme but with much-reduced complexity, and enables multiple-rate feedback. Hengtao He, Shenghui Song 0001, Jun Zhang 0004, Khaled Ben Letaief |
IEEE Trans. Wirel. Commun. | 2 |
| 2025 | Model-Driven Sensing-Node Selection and Power Allocation for Tracking Maneuvering Targets in Perceptive Mobile NetworksabstractManeuvering target tracking is an important service of future wireless networks to assist innovative applications such as intelligent transportation. However, tracking maneuvering targets by cellular networks faces many challenges. In particular, the dense network and high-speed targets make the selection of the sensing nodes (SNs) and the associated power allocation very challenging. Existing methods demonstrated engaging performance, but with high computational complexity. In this paper, we propose a model-driven deep learning (DL)-based approach for SN selection. To this end, we first propose an iterative SN selection method by jointly exploiting the majorization-minimization (MM) framework and the alternating direction method of multipliers (ADMM). Then, we unfold the iterative algorithm as a deep neural network and prove its convergence. The proposed method achieves lower computational complexity, as the number of layers is less than the number of iterations required by the original algorithm, and each layer only involves simple matrix-vector additions/multiplications. Finally, we propose an efficient power allocation method based on fixed point (FP) water filling and solve the joint SN selection and power allocation problem under the alternative optimization framework. Simulation results show that the proposed method achieves better performance than conventional optimization-based algorithms with much lower computational complexity. Lei Xie 0009, Hengtao He, Shenghui Song 0001, Yonina C. Eldar |
IEEE Trans. Wirel. Commun. | 2 |
| 2024 | Learning Bayes-Optimal Channel Estimation for Holographic MIMO in Unknown EM EnvironmentsabstractHolographic MIMO (HMIMO) has recently been recognized as a promising enabler for future 6G systems through the use of an ultra-massive number of antennas in a compact space to exploit the propagation characteristics of the electromagnetic (EM) channel. Nevertheless, the promised gain of HMIMO could not be fully unleashed without an efficient means to estimate the high-dimensional channel. Bayes-optimal estimators typically necessitate either a large volume of supervised training samples or a priori knowledge of the true channel distribution, which could hardly be available in practice due to the enormous system scale and the complicated EM environments. It is thus important to design a Bayes-optimal estimator for the HMIMO channels in arbitrary and unknown EM environments, free of any supervision or priors. This work proposes a self-supervised minimum mean-square-error (MMSE) channel estimation algorithm based on powerful machine learning tools, i.e., score matching and principal component analysis. The training stage requires only the pilot signals, without knowing the spatial correlation, the ground-truth channels, or the received signal-to-noise-ratio. Simulation results will show that, even being totally self-supervised, the proposed algorithm can still approach the performance of the oracle MMSE method with an extremely low complexity, making it a competitive candidate in practice. Hengtao He, Xianghao Yu, Shenghui Song 0001, Jun Zhang 0004, Ross Murch, Khaled Ben Letaief |
ICC | 2 |
| 2024 | Newtonized Near-Field Channel Estimation for Ultra-Massive MIMO SystemsabstractTo meet the stringent requirements of future communication systems, ultra-massive multiple-input and multiple-output (UM-MIMO) technology has garnered significant attention as a key enabling technology for 6G. However, the deployment of UM-MIMO introduces new challenges, particularly the near-field effect. In this paper, by leveraging the unique characteristics of near-field channels, we propose a novel near-field channel estimation algorithm based on the Newton's method. We also design a near-field codebook that meets the requirements for convergence guarantee. Our algorithm overcomes the limitations of existing approaches by offering a low-complexity, tuning-free, and convergence-guaranteed solution. Simulation results show that our proposed algorithm outperforms state-of-the-art baselines in terms of estimation accuracy, establishing its effectiveness in near-field channel estimation for UM-MIMO systems. Ruoxiao Cao, Hengtao He, Xianghao Yu, Shenghui Song 0001, Jun Zhang 0004, Yi Gong 0001, Khaled Ben Letaief |
WCNC | 3 |
| 2024 | Message Passing Meets Graph Neural Networks: A New Paradigm for Massive MIMO SystemsabstractAs one of the core technologies for 5G systems, massive multiple-input multiple-output (MIMO) introduces dramatic capacity improvements along with very high beamforming and spatial multiplexing gains. When developing efficient physical layer algorithms for massive MIMO systems, message passing is one promising candidate owing to its superior performance. However, as their computational complexity increases dramatically with the problem size, the state-of-the-art message passing algorithms cannot be directly applied to future 6G systems, where an exceedingly large number of antennas are expected to be deployed. To address this issue, we propose a model-driven deep learning (DL) framework, namely the AMP-GNN for massive MIMO transceiver design, by considering thelow complexityof the AMP algorithm andadaptabilityof GNNs. Specifically, the structure of the AMP-GNN network is customized by unfolding the approximate message passing (AMP) algorithm and introducing a graph neural network (GNN) module into it. The permutation equivariance property of AMP-GNN is proved, which enables the AMP-GNN to learn more efficiently and to adapt to different numbers of users. We also reveal the underlying reason why GNNs improve the AMP algorithm from the perspective of expectation propagation, which motivates us to amalgamate various GNNs with different message passing algorithms. In the simulation, we take the massive MIMO detection to exemplify that the proposed AMP-GNN significantly improves the performance of the AMP detector, achieves comparable performance as the state-of-the-art DL-based MIMO detectors, and presents strong robustness to various mismatches. Hengtao He, Xianghao Yu, Jun Zhang 0004, Shenghui Song 0001, Khaled Ben Letaief |
IEEE Trans. Wirel. Commun. | 1 |
| 2023 | Task-Oriented Communication with Out-of-Distribution Detection: An Information Bottleneck FrameworkabstractTask-oriented communication is an emerging paradigm for next-generation communication networks, which extracts and transmits task-relevant information, instead of raw data, for downstream applications. Most existing deep learning (DL)-based task-oriented communication systems adopt a closed-world assumption, assuming either the same data distribution for training and testing, or the system could have access to a large out-of-distribution (OoD) dataset for retraining. However, in practical open-world scenarios, task-oriented communication systems will be exposed to unknown OoD data. The powerful approximation ability of learning methods may force the task-oriented communication systems to overfit the training data (i.e., in-distribution data). Therefore, these systems tend to provide overconfident judgments when encountering OoD data. Based on the information bottleneck (IB) framework, we propose a class conditional IB (CCIB) approach to address this problem, supported by information-theoretical insights. The idea is to extract distinguishable features from in-distribution data while keeping their compactness and informativeness. It is achieved by imposing the class conditional latent prior distribution and enforcing the latent of different classes to be far away from each other. Simulation results shall demonstrate that the proposed approach detects OoD data more efficiently than the baselines and state-of-the-art approaches, without compromising the rate-distortion tradeoff. Hengtao He, Jiawei Shao, Shenghui Song 0001, Jun Zhang 0004, Khaled Ben Letaief |
GLOBECOM | 3 |
| 2023 | Blind Performance Prediction for Deep Learning Based Ultra-Massive MIMO Channel EstimationabstractReliability is of paramount importance for the physical layer of wireless systems due to its decisive impact on end-to-end performance. However, the uncertainty of prevailing deep learning (DL)-based physical layer algorithms is hard to quantify due to the black-box nature of neural networks. This limitation is a major obstacle that hinders their practical deployment. In this paper, we attempt to quantify the uncertainty of an important category of DL-based channel estimators. An efficient statistical method is proposed to make blind predictions for the mean squared error of the DL-estimated channel solely based on received pilots, without knowledge of the ground-truth channel, the prior distribution of the channel, or the noise statistics. The complexity of the blind performance prediction is low and scales only linearly with the number of antennas. Simulation results for ultra-massive multiple-input multiple-output (UM-MIMO) channel estimation with a mixture of far-field and near-field paths are provided to verify the accuracy and efficiency of the proposed method. Hengtao He, Xianghao Yu, Shenghui Song 0001, Jun Zhang 0004, Khaled Ben Letaief |
ICC | 2 |
| 2023 | Opponent Modeling Based Dynamic Resource Trading for UAV-Assisted Edge ComputingabstractThis paper proposes a dynamic resource trading scheme in unmanned aerial vehicle (UAV)-assisted mobile edge computing (MEC) network. A UAV-assisted MEC server adaptively adjusts its trajectory to sell the computation offloading services to the mobile users (MUs), where the MUs have stochastic task arrivals. In this context, we formulate the sequential resource trading problem as a stochastic Stackelberg game, which is composed of two stages for each trading round. In the first stage, the self-interested UAV jointly optimizes its trajectory and service price to maximize its long-term profits. In the second stage, the non-cooperative MUs optimize their binary offloading decisions to minimize the average task processing delay and service payment. However, it is challenging to obtain the equilibrium across the fully decentralized agents with constantly evolving and tightly coupled policies, where each agent is confronted with a non-stationary environment. To solve this problem, we propose an opponent modeling based double deep Q learning (OM-DDQN) algorithm, where each agent adopts opponent modeling to effectively predict the trading strategies of other agents in the network. Simulation results demonstrate that, compared with the baseline algorithms, the proposed algorithm can achieve a win-win resource trading outcome that not only enhances the UAV's profit but also reduces the MUs' costs. Jinxiang Bai, Zhe Wang 0005, Jun Li 0004, Long Shi 0001, Jie Zhang 0076, Kang Wei 0004, Hengtao He |
VTC Fall | 7 |
| 2023 | GNN-Enhanced Approximate Message Passing for Massive/Ultra-Massive MIMO DetectionabstractEfficient massive/ultra-massive multiple-input multiple-output (MIMO) detection algorithms with satisfactory performance and low complexity are critical to meet the high throughput and ultra-low latency requirements in 5G and beyond communications, given the extremely large number of antennas. In this paper, we propose a low complexity graph neural network (GNN) enhanced approximate message passing (AMP) algorithm, AMP-GNN, for massive/ultra-massive MIMO detection. The structure of the neural network is customized by unfolding the AMP algorithm and introducing the GNN module for multiuser interference cancellation. Numerical results will show that the proposed AMP-GNN significantly improves the performance of the AMP detector and achieves comparable performance as the state-of-the-art deep learning-based MIMO detectors but with reduced computational complexity. Furthermore, it presents strong robustness to the change of the number of users. Hengtao He, Alva Kosasih, Xianghao Yu, Jun Zhang 0004, Shenghui Song 0001, Wibowo Hardjawana, Khaled Ben Letaief |
WCNC | 1 |
| 2023 | Beamspace Channel Estimation for Wideband Millimeter-Wave MIMO: A Model-Driven Unsupervised Learning ApproachabstractMillimeter-wave (mmWave) communications have been one of the promising technologies for future wireless networks that integrate a wide range of data-demanding applications. To compensate for the large channel attenuation in mmWave band and avoid high hardware cost, a lens-based beamspace massive multiple-input multiple-output (MIMO) system is considered. However, the spatial-wideband effect in wideband mmWave systems makes channel estimation very challenging, especially when the receiver is equipped with a limited number of radio-frequency (RF) chains. Furthermore, the real channel data cannot be obtained before the mmWave system is used in a new environment, which makes it impossible to train a deep learning (DL)-based channel estimator using real data set beforehand. To solve the problem, we propose a model-driven unsupervised learning network, named learned denoising-based generalized expectation consistent (LDGEC) signal recovery network. By utilizing the Stein’s unbiased risk estimator loss, the LDGEC network can be trained only with limited measurements corresponding to the pilot symbols, instead of the real channel data. Even if designed for unsupervised learning, the LDGEC network can be supervisingly trained with the real channel via the denoiser-by-denoiser way. The numerical results demonstrate that the LDGEC-based channel estimator significantly outperforms state-of-the-art compressive sensing-based algorithms when the receiver is equipped with a small number of RF chains and low-resolution ADCs. Hengtao He, Rui Wang 0001, Weijie Jin, Shi Jin 0002, Chao-Kai Wen, Geoffrey Ye Li |
IEEE Trans. Wirel. Commun. | 1 |
| 2022 | Hybrid Far- and Near-Field Channel Estimation for THz Ultra-Massive MIMO via Fixed Point NetworksabstractTerahertz ultra-massive multiple-input multiple-output (THz UM-MIMO) is envisioned as one of the key enablers of 6G wireless systems. Due to the joint effect of its large array aperture and small wavelength, the near-field region of THz UM-MIMO is greatly enlarged. The high-dimensional channel of such systems thus consists of a stochastic mixture of far and near fields, which renders channel estimation extremely challenging. Previous works based on uni-field assumptions cannot capture the hybrid far- and near-field features, thus suffering significant performance loss. This motivates us to consider hybrid-field channel estimation. We draw inspirations from fixed point theory to develop an efficient deep learning based channel estimator with adaptive complexity and linear convergence guarantee. Built upon classic orthogonal approximate message passing, we transform each iteration into a contractive mapping, comprising a closed-form linear estimator and a neural network based non-linear estimator. A major algorithmic innovation involves applying fixed point iteration to compute the channel estimate while modeling neural networks with arbitrary depth and adapting to the hybrid-field channel conditions. Simulation results verify our theoretical analysis and show significant performance gains over state-of-the-art approaches in the estimation accuracy and convergence rate. Yifei Shen 0004, Hengtao He, Xianghao Yu, Jun Zhang 0004, Khaled Ben Letaief |
GLOBECOM | 3 |
| 2021 | Distributed Expectation Propagation Detection for Cell-Free Massive MIMOabstractIn cell-free massive MIMO networks, an efficient distributed detection algorithm is of significant importance. In this paper, we propose a distributed expectation propagation (EP) detector for cell-free massive MIMO. The detector is composed of two modules, a nonlinear module at the central processing unit (CPU) and a linear module at the access point (AP). The turbo principle in iterative decoding is utilized to compute and pass the extrinsic information between modules. An analytical framework is then provided to characterize the asymptotic performance of the proposed EP detector with a large number of antennas. Simulation results will show that the proposed method outperforms the distributed detectors in terms of bit-error rate. Hengtao He, Hanqing Wang 0002, Xianghao Yu, Jun Zhang 0004, Shenghui Song 0001, Khaled Ben Letaief |
GLOBECOM | 1 |
| 2021 | Adaptive Channel Estimation Based on Model-Driven Deep Learning for Wideband mmWave SystemsabstractChannel estimation in wideband millimeter-wave (mmWave) systems is very challenging due to the beam squint effect. To solve the problem, we propose a learnable iterative shrinkage thresholding algorithm-based channel estimator (LISTA-CE) based on deep learning. The proposed channel estimator can learn to transform the beam-frequency mmWave channel into the domain with sparse features through training data. The transform domain enables us to adopt a simple denoiser with few trainable parameters. We further enhance the adaptivity of the estimator by introducing hypernetwork to automatically generate learnable parameters for LISTA-CE online. Simulation results show that the proposed approach can significantly outperform the state-of-the-art deep learning-based algorithms with lower complexity and fewer parameters and adapt to new scenarios rapidly. Weijie Jin, Hengtao He, Chao-Kai Wen, Shi Jin 0002, Geoffrey Ye Li |
GLOBECOM | 2 |
| 2019 | Deep Learning Based on Orthogonal Approximate Message Passing for CP-Free OFDMabstractChannel estimation and signal detection are very challenging for an orthogonal frequency division multiplexing (OFDM) system without cyclic prefix (CP). In this article, deep learning based on orthogonal approximate message passing (DL-OAMP) is used to address these problems. The DL-OAMP receiver includes a channel estimation neural network (CE-Net) and a signal detection neural network based on OAM-P, called OAMP-Net. The CE-Net is initialized by the least square channel estimation algorithm and refined by minimum mean-squared error (MMSE) neural network. The OAMP-Net is established by unfolding the iterative OAMP algorithm and adding some trainable parameters to improve the detection performance. The DL-OAMP receiver is with low complexity and can estimate time-varying channels with only a single training. Simulation results demonstrate that the bit-error rate (BER) of the proposed scheme is lower than those of competitive algorithms for high-order modulation. Jing Zhang 0031, Hengtao He, Chao-Kai Wen, Shi Jin 0002, Geoffrey Ye Li |
ICASSP | 2 |
| 2017 | Generalized expectation consistent signal recovery for nonlinear measurementsabstractIn this paper, we propose a generalized expectation consistent signal recovery algorithm to estimate the signal x from the nonlinear measurements of a linear transform output z = Ax. This estimation problem has been encountered in many applications, such as communications with front-end impairments, compressed sensing, and phase retrieval. The proposed algorithm extends the prior art called generalized turbo signal recovery from a partial discrete Fourier transform matrix A to a class of general matrices. Numerical results show the excellent agreement of the proposed algorithm with the theoretical Bayesian-optimal estimator derived using the replica method. Hengtao He, Chao-Kai Wen, Shi Jin 0002 |
ISIT | 1 |