Link prediction aims at inferring missing links or predicting future ones based on the currently observed network. This topic is important for many applications such as social media, bioinformatics and recommendation systems. Most existing methods focus on homogeneous settings and consider only low-order pairwise relations while ignoring either the heterogeneity or high-order complex relations among different types of nodes, which tends to lead to a sub-optimal embedding result. This paper presents a method named Heterogeneous Hypergraph Variational Autoencoder (HeteHG-VAE) for link prediction in heterogeneous information networks (HINs). It first maps a conventional HIN to a heterogeneous hypergraph with a certain kind of semantics to capture both the high-order semantics and complex relations among nodes, while preserving the low-order pairwise topology information of the original HIN. Then, deep latent representations of nodes and hyperedges are learned by a Bayesian deep generative framework from the heterogeneous hypergraph in an unsupervised manner. Moreover, a hyperedge attention module is designed to learn the importance of different types of nodes in each hyperedge. The major merit of HeteHG-VAE lies in its ability of modeling multi-level relations in heterogeneous settings. Extensive experiments on real-world datasets demonstrate the effectiveness and efficiency of the proposed method.
As network and autonomous driving technologies rapidly advance, traffic flow prediction has become a crucial area of research. It plays a significant role in optimizing urban traffic management and enhancing road safety, drawing increasing attention from researchers. Traffic flow data, as a specific form of time-series data, are often used in prediction tasks utilizing large language models. Recent developments in graph data and improvements in graph neural networks have led researchers to employ methods like adjacency and Laplacian matrices to address relational issues among distant nodes. However, most existing methods focus on enhancing prediction performance through network architectures or using adaptive matrices to capture spatiotemporal relationships, with limited exploration into the impact of input embedding. This paper introduces the ASTRformer, an innovative approach to traffic flow prediction that emphasizes the fusion of spatial and temporal information in historical data through an adaptive spatio-temporal relation learning mechanism. This mechanism integrates feature embedding with adaptive spatial and temporal embeddings. A learnable spatio-temporal fusion network then parameterizes these embeddings, producing the input representations. Subsequently, a transformer model captures these representations to predict future traffic flows. Experimental results on six datasets demonstrate that our method effectively captures spatiotemporal dependencies, achieving state-of-the-art performance across various prediction metrics.
1 December 2021 in Journal of Environmental Chemical Engineering
In unsupervised graph anomaly detection, existing methods usually focus on detecting outliers by learning local context information of nodes, while often ignoring the importance of global context. However, global context information can provide more comprehensive relationship information between nodes in the network. By considering the structure of the entire network, detection methods are able to identify potential dependencies and interaction patterns between nodes, which is crucial for anomaly detection. Therefore, we propose an innovative graph anomaly detection framework, termed CoCo (Context Correlation Discrepancy Analysis), which detects anomalies by meticulously evaluating variances in correlations. Specifically, CoCo leverages the strengths of Transformers in sequence processing to effectively capture both global and local contextual features of nodes by aggregating neighbor features at various hops. Subsequently, a correlation analysis module is employed to maximize the correlation between local and global contexts of each normal node. Unseen anomalies are ultimately detected by measuring the discrepancy in the correlation of nodes’ contextual features. Extensive experiments conducted on six datasets with synthetic outliers and five datasets with organic outliers have demonstrated the significant effectiveness of CoCo compared to existing methods.
1 November 2023 in IEEE Transactions on Instrumentation and Measurement
The Kalman filter performs well in system state estimation by inferring a joint probability distribution over time variables, which has numerous technological applications in time series analysis. However, the complex Kalman filter parameter settings prevent it from optimally estimating the system state, and this suboptimal estimation makes it difficult to effectively distinguish between normal and anomalous system states in anomaly detection. In this paper, we propose a deep embedding-optimized Kalman filter for unsupervised time series anomaly detection, where the system state of a normal time series can be fitted by the embedding-optimized Kalman filter in an unsupervised manner and anomalies can be detected from data points that deviate from the normal system state. Specifically, we use an autoencoder-enhanced Kalman filter to capture the normal pattern of the time series, where the original time series signal is first fed into the Kalman filter, then the autoencoder encodes the filtered signal and reconstructs the original signal, and the learned embedding from encoder is used to sequentially optimize the Kalman filter. Finally, the optimized filter captures the normal pattern of the time series and the reconstruction error from the filtered signal can be measured to detect anomalies. The validity of the method is verified on real-world time series datasets.