LGNov 27, 2023
Out-of-Distribution Generalized Dynamic Graph Neural Network for Human Albumin PredictionZeyang Zhang, Xingwang Li, Fei Teng et al. · tsinghua
Human albumin is essential for indicating the body's overall health. Accurately predicting plasma albumin levels and determining appropriate doses are urgent clinical challenges, particularly in critically ill patients, to maintain optimal blood levels. However, human albumin prediction is non-trivial that has to leverage the dynamics of biochemical markers as well as the experience of treating patients. Moreover, the problem of distribution shift is often encountered in real clinical data, which may lead to a decline in the model prediction performance and reduce the reliability of the model's application. In this paper, we propose a framework named Out-of-Distribution Generalized Dynamic Graph Neural Network for Human Albumin Prediction (DyG-HAP), which is able to provide accurate albumin predictions for Intensity Care Unit (ICU) patients during hospitalization. We first model human albumin prediction as a dynamic graph regression problem to model the dynamics and patient relationship. Then, we propose a disentangled dynamic graph attention mechanism to capture and disentangle the patterns whose relationship to labels under distribution shifts is invariant and variant respectively. Last, we propose an invariant dynamic graph regression method to encourage the model to rely on invariant patterns to make predictions. Moreover, we propose a dataset named Albumin level testing and nutritional dosing data for Intensive Care (ANIC) for evaluation. Extensive experiments demonstrate the superiority of our method compared to several baseline methods in human albumin prediction.
ITMar 16
Rotatable Antenna Assisted Mobile Edge ComputingJi Wang, Hao Chen, Yixuan Li et al.
This paper investigates a rotatable antenna (RA) assisted mobile edge computing (MEC) network, where multiple users offload their computation tasks to an edge server equipped with an RA array under a time-division multiple access protocol. To maximize the weighted sum computation rate, we formulate a joint optimization problem over the RA rotation angles, time-slot allocation, transmit power, and local CPU frequencies. Due to the non-convex nature of the formulated problem, a scenario-adaptive hybrid optimization algorithm is proposed. Specifically, for the dynamic rotating scenario, where RAs can flexibly reorient within each time slot, we derive closed-form optimal antenna pointing vectors to enable a low-complexity sequential solution. In contrast, for the static rotating scenario where RAs maintain a unified orientation, we develop an alternating optimization framework, where the non-convex RA rotation constraints are handled using successive convex approximation iteratively with the resource allocation. Simulation results demonstrate that the proposed RA assisted MEC network significantly outperforms conventional fixed-antenna MEC networks. Owing to the additional spatial degrees of freedom introduced by mechanical rotation, the flexibility of RAs effectively mitigates the severe beam misalignment inherent in fixed-antenna systems, particularly under high antenna directivity.
ITApr 20
Channel Estimation for Rydberg Atomic Quantum Receivers: Unrolled Phase Retrieval from Holographic SnapshotsJian Xiao, Ji Wang, Ming Zeng et al.
A model-driven deep learning framework is proposed for channel estimation in Rydberg atomic quantum receivers (RAQRs) based on the measurement of holographic snapshots. Specifically, we develop a Transformer-based unrolling architecture, termed URformer, to solve the non-linear biased phase retrieval problem, which is derived by unrolling a stabilized variant of the expectation-maximization Gerchberg-Saxton (EM-GS) algorithm. Each layer of the proposed URformer incorporates three trainable modules: 1) a learnable filter network that replaces the fixed Bessel kernel in the classic EM-GS algorithm; 2) a trainable gating mechanism that adaptively combines classic updates to ensure training stability; and 3) an efficient channel Transformer module that learns to correct residual errors by capturing non-local channel dependencies. Numerical results demonstrate that the proposed URformer significantly outperforms classic iterative algorithms and conventional black-box neural networks with less pilot overhead.
ITMar 25
Wireless AI Evolution: From Statistical Learners to Electromagnetic-Guided Foundation ModelsJian Xiao, Ji Wang, Kunrui Cao et al.
While initial applications of artificial intelligence (AI) in wireless communications over the past decade have demonstrated considerable potential using specialized models for targeted communication tasks, the revolutionary demands of sixth-generation (6G) networks for holographic communications, ubiquitous sensing, and native intelligence are propelling a necessary evolution towards AI-native wireless networks. The arrival of large AI models paves the way for the next phase of Wireless AI, driven by wireless foundation models (WFMs). In particular, pre-training on universal electromagnetic (EM) principles equips WFMs with the essential adaptability for a multitude of demanding 6G applications. However, existing large AI models face critical limitations, including pre-training strategies disconnected from EM-compliant constraints leading to physically inconsistent predictions, a lack of embedded understanding of wave propagation physics, and the inaccessibility of massive labeled datasets for comprehensive EM-aware training. To address these challenges, this article presents an electromagnetic information theory-guided self-supervised pre-training (EIT-SPT) framework designed to systematically inject EM physics into WFMs. The EIT-SPT framework aims to infuse WFMs with intrinsic EM knowledge, thereby enhancing their physical consistency, generalization capabilities across varied EM landscapes, and overall data efficiency. Building upon the proposed EIT-SPT framework, this article first elaborates on diverse potential applications in 6G scenarios of WFMs, then validates the efficacy of the proposed framework through illustrative case studies, and finally summarizes critical open research challenges and future directions for WFMs.
ITMar 19
Channel Estimation for Flexible Intelligent Metasurfaces: From Model-Based Approaches to Neural OperatorsJian Xiao, Ji Wang, Qimei Cui et al.
Flexible intelligent metasurfaces (FIMs) offer a new solution for wireless communications by introducing morphological degrees of freedom, dynamically morphing their three-dimensional shape to ensure multipath signals interfere constructively. However, realizing the desired performance gains in FIM systems is critically dependent on acquiring accurate channel state information across a continuous and high-dimensional deformation space. Therefore, this paper investigates this fundamental channel estimation problem for FIM assisted millimeter-wave communication systems. First, we develop model-based frameworks that structure the problem as either function approximation using interpolation and kernel methods or as a sparse signal recovery problem that leverages the inherent angular sparsity of millimeter-wave channels. To further advance the estimation capability beyond explicit assumptions in model-based channel estimation frameworks, we propose a deep learning-based framework using a Fourier neural operator (FNO). By parameterizing a global convolution operator in the Fourier domain, we design an efficient FNO architecture to learn the continuous operator that maps FIM shapes to channel responses with mesh-independent properties. Furthermore, we exploit a hierarchical FNO (H-FNO) architecture to efficiently capture the multi-scale features across a hierarchy of spatial resolutions. Numerical results demonstrate that the proposed H-FNO significantly outperforms the model-based benchmarks in estimation accuracy and pilot efficiency. In particular, the interpretability analysis show that the proposed H-FNO learns an anisotropic spatial filter adapted to the physical geometry of FIM and is capable of accurately reconstructing the non-linear channel response across the continuous deformation space.
ITApr 10
Robust Single- and Multi-Pinching Antenna Systems Under User Location UncertaintyHao Feng, Ebrahim Bedeer, Ming Zeng et al.
Pinching antenna (PA) systems have recently emerged as a promising architecture for reconfigurable wireless communications by enabling flexible antenna placement along a dielectric waveguide. However, existing works typically assume perfect knowledge of user locations, which is impractical in real systems where location estimation errors are inevitable. In this paper, we investigate robust power allocation and antenna placement for PA systems under user location uncertainty. We consider both single-antenna and multi-antenna configurations, where the true user locations are unknown but lie within bounded uncertainty regions. For the single-antenna case, we adopt a worst-case robust design and leverage the S-procedure to transform the joint power allocation and antenna placement problem into a convex semidefinite program (SDP), ensuring that quality-of-service (QoS) constraints are satisfied for all possible user locations. For the multi-antenna case, we address the additional challenges arising from the superposition of channel components from multiple antennas by developing an efficient numerical procedure to evaluate the worst-case channel gain. Then, we derive a closed-form solution for optimal power allocation and develop a block coordinate descent algorithm to optimize antenna placement. Simulation results show that the proposed framework provides robustness to location uncertainty while achieving power consumption close to that of outage-based benchmark schemes.
ITMay 18
Movable Antenna-Aided Secure LEO Satellite Networks: Joint Antenna Position and Beamforming OptimizationSuhong Luo, Pan Tang, Jianhua Zhang et al.
The broadcast characteristics of sixth-generation (6G) low-earth orbit (LEO) satellite communications raise serious security issues. Movable antenna (MA) technology offers a promising physical layer security (PLS) solution by flexibly reconfiguring antenna positions to exploit additional spatial degrees of freedom. However, in highly dense LEO satellite constellations, the legitimate satellite and potential eavesdropping satellites may exhibit small angular separations, which poses significant challenges for the design of secure transmission schemes. To address this challenge, this paper proposes an MA-assisted secure transmission scheme for time-varying LEO satellite communications, where a ground station equipped with an MA array communicates with a serving satellite, while the other visible satellites are regarded as potential eavesdroppers. We maximize the average secrecy rate by jointly optimizing the transmit beamforming and MA positions. An alternating optimization (AO) framework is developed, where semidefinite relaxation is adopted for the beamforming optimization subproblem, while high-accuracy successive convex approximation (SCA) and low-complexity differential evolution (DE) algorithms are proposed for the MA position optimization subproblem. Numerical results demonstrate that the proposed MA-assisted LEO secure transmission scheme consistently achieves superior performance compared to the conventional fixed-position antenna scheme.
LGDec 2, 2025
TabGRU: An Enhanced Design for Urban Rainfall Intensity Estimation Using Commercial Microwave LinksXingwang Li, Mengyun Chen, Jiamou Liu et al.
In the face of accelerating global urbanization and the increasing frequency of extreme weather events, highresolution urban rainfall monitoring is crucial for building resilient smart cities. Commercial Microwave Links (CMLs) are an emerging data source with great potential for this task.While traditional rainfall retrieval from CMLs relies on physicsbased models, these often struggle with real-world complexities like signal noise and nonlinear attenuation. To address these limitations, this paper proposes a novel hybrid deep learning architecture based on the Transformer and a Bidirectional Gated Recurrent Unit (BiGRU), which we name TabGRU. This design synergistically captures both long-term dependencies and local sequential features in the CML signal data. The model is further enhanced by a learnable positional embedding and an attention pooling mechanism to improve its dynamic feature extraction and generalization capabilities. The model was validated on a public benchmark dataset from Gothenburg, Sweden (June-September 2015). The evaluation used 12 sub-links from two rain gauges (Torp and Barl) over a test period (August 22-31) covering approximately 10 distinct rainfall events. The proposed TabGRU model demonstrated consistent advantages, outperforming deep learning baselines and achieving high coefficients of determination (R2) at both the Torp site (0.91) and the Barl site (0.96). Furthermore, compared to the physics-based approach, TabGRU maintained higher accuracy and was particularly effective in mitigating the significant overestimation problem observed in the PL model during peak rainfall events. This evaluation confirms that the TabGRU model can effectively overcome the limitations of traditional methods, providing a robust and accurate solution for CML-based urban rainfall monitoring under the tested conditions.
DCMar 26
PRISM: Dynamic Primitive-Based Forecasting for Large-Scale GPU Cluster WorkloadsXin Wu, Fei Teng, Xingwang Li et al.
Accurately forecasting GPU workloads is essential for AI infrastructure, enabling efficient scheduling, resource allocation, and power management. Modern workloads are highly volatile, multiple periodicity, and heterogeneous, making them challenging for traditional predictors. We propose PRISM, a primitive-based compositional forecasting framework combining dictionary-driven temporal decomposition with adaptive spectral refinement. This dual representation extracts stable, interpretable workload signatures across diverse GPU jobs. Evaluated on large-scale production traces, PRISM achieves state-of-the-art results. It significantly reduces burst-phase errors, providing a robust, architecture-aware foundation for dynamic resource management in GPU-powered AI platforms.
LGMar 18, 2025
Out-of-Distribution Generalization in Time Series: A SurveyXin Wu, Fei Teng, Xingwang Li et al.
Time series frequently manifest distribution shifts, diverse latent features, and non-stationary learning dynamics, particularly in open and evolving environments. These characteristics pose significant challenges for out-of-distribution (OOD) generalization. While substantial progress has been made, a systematic synthesis of advancements remains lacking. To address this gap, we present the first comprehensive review of OOD generalization methodologies for time series, organized to delineate the field's evolutionary trajectory and contemporary research landscape. We organize our analysis across three foundational dimensions: data distribution, representation learning, and OOD evaluation. For each dimension, we present several popular algorithms in detail. Furthermore, we highlight key application scenarios, emphasizing their real-world impact. Finally, we identify persistent challenges and propose future research directions. A detailed summary of the methods reviewed for the generalization of OOD in time series can be accessed at https://tsood-generalization.com.
SPFeb 25, 2025
A Novel Spatiotemporal Correlation Anomaly Detection Method Based on Time-Frequency-Domain Feature Fusion and a Dynamic Graph Neural Network in Wireless Sensor NetworkMiao Ye, Zhibang Jiang, Xingsi Xue et al.
Attention-based transformers have played an important role in wireless sensor network (WSN) timing anomaly detection due to their ability to capture long-term dependencies. However, there are several issues that must be addressed, such as the fact that their ability to capture long-term dependencies is not completely reliable, their computational complexity levels are high, and the spatiotemporal features of WSN timing data are not sufficiently extracted for detecting the correlation anomalies of multinode WSN timing data. To address these limitations, this paper proposes a WSN anomaly detection method that integrates frequency-domain features with dynamic graph neural networks (GNN) under a designed self-encoder reconstruction framework. First, the discrete wavelet transform effectively decomposes trend and seasonal components of time series to solve the poor long-term reliability of transformers. Second, a frequency-domain attention mechanism is designed to make full use of the difference between the amplitude distributions of normal data and anomalous data in this domain. Finally, a multimodal fusion-based dynamic graph convolutional network (MFDGCN) is designed by combining an attention mechanism and a graph convolutional network (GCN) to adaptively extract spatial correlation features. A series of experiments conducted on public datasets and their results demonstrate that the anomaly detection method designed in this paper exhibits superior precision and recall than the existing methods do, with an F1 score of 93.5%, representing an improvement of 2.9% over that of the existing models.
LGAug 19, 2025
ERIS: An Energy-Guided Feature Disentanglement Framework for Out-of-Distribution Time Series ClassificationXin Wu, Fei Teng, Ji Zhang et al.
An ideal time series classification (TSC) should be able to capture invariant representations, but achieving reliable performance on out-of-distribution (OOD) data remains a core obstacle. This obstacle arises from the way models inherently entangle domain-specific and label-relevant features, resulting in spurious correlations. While feature disentanglement aims to solve this, current methods are largely unguided, lacking the semantic direction required to isolate truly universal features. To address this, we propose an end-to-end Energy-Regularized Information for Shift-Robustness (ERIS) framework to enable guided and reliable feature disentanglement. The core idea is that effective disentanglement requires not only mathematical constraints but also semantic guidance to anchor the separation process. ERIS incorporates three key mechanisms to achieve this goal. Specifically, we first introduce an energy-guided calibration mechanism, which provides crucial semantic guidance for the separation, enabling the model to self-calibrate. Additionally, a weight-level orthogonality strategy enforces structural independence between domain-specific and label-relevant features, thereby mitigating their interference. Moreover, an auxiliary adversarial generalization mechanism enhances robustness by injecting structured perturbations. Experiments across four benchmarks demonstrate that ERIS achieves a statistically significant improvement over state-of-the-art baselines, consistently securing the top performance rank.