Lei Bai 0001

dblp:119/1223-1 · DBLP profile ↗
← Back
15ranked-venue papers in the field
3as first author
9since 2021 · last 2026
0000-0003-3378-7201ORCID · conflict

Domains — venue-derived; a paper can count in several

Data Mining & Knowledge Discovery · 7 (2 first)Database Systems & Data Management · 4Information Retrieval & Web Search · 4 (1 first)
YearPublicationVenuePosition
2026 Modeling Point-to-Point Dependency for High-Dimensional Long-Term Series Forecasting
Xinyu Li 0014, Kexi Chen, Ying Zheng 0004, Zhiyi Yao, Yi Xie 0003, Jihan Dai, Lei Bai 0001, Jin Zhao 0001, Jiajie Shen, Yunqi Cai, Hong Lu 0001, Xin Wang 0002
WWW7
2026 TRACK: Temporal Decoupled Kriging for Inductive Spatio-Temporal Graph
abstract
The deployment of sensors enables data-driven urban management, but necessitates inductive spatio-temporal kriging to infer unmonitored areas. Existing methods impute these unknown observations by smoothing temporal features based on spatial dependencies, overlooking the decoupling ofinherent propertiesanddynamic correlationsin message passing. In particular, the inherent properties reveal non-transitive signals, and current coupled aggregation leads to inaccurate results. To this end, we proposeTempoRAl deCoupledKriging, named TRACK, to decouple two factors with the help of node-specific inherency. Specifically, we first construct a node-specific profile to represent its inherency including geographical and periodic features, which is subsequently transformed into decoupling prompts. Secondly, the coupled temporal features are separated through querying each prompt embedding, facilitating precise temporal aggregation for inherent properties and spatial aggregation for dynamic correlations. Finally, a multi-task training strategy is further adopted to mimic the inductive scenarios during testing. We evaluate TRACK on four real-world datasets spanning urban traffic and air quality prediction tasks. TRACK achieves state-of-the-art performance, with average improvements of 3.10% in MAE and 4.45% in RMSE over strong baselines. Moreover, we further demonstrated its robust generalization in a challenging cross-city inductive setting. Code is available athttps://github.com/JeremyChou28/TRACK.
Jianping Zhou 0004, Weida Wang, Bin Lu 0005, Guanjie Zheng, Lei Bai 0001, Xinbing Wang, Chenghu Zhou
IEEE Trans. Knowl. Data Eng.5
2025 DIFFODE: Neural ODE with Differentiable Hidden State for Irregular Time Series Analysis
abstract
Irregular time series analysis is increasingly essential in data management due to the proliferation of complex data irregularly sampled by real-world systems. Traditional time series models, including RNN-based models and transformer variants, face significant challenges in generalizing to continuous-time paradigms, which are essential for capturing the ongoing dynamics of irregular time series. Neural Ordinary Differential Equations (NODEs) assume a continuous latent dynamic and provide an elegant framework for irregular time series analysis, yet they suffer from limitations like fragmented latent processes and the inability to fully exploit interdependencies among observations. To address these challenges, we propose a novel Differentiable hidden state enhanced neural ODE framework, termed DIFFODE, designed to effectively model irregular time series. Concretely, we introduce an attention-based differential hidden state that maps irregular observations into a continuous hidden state space, enabling the extraction of latent dynamics while preserving temporal continuity. Leveraging the theory of generalized inverses, DIFFODE innovatively derives ODEs to describe hidden state dynamics. Furthermore, we incorporate the Hoyer metric into our framework to enhance its capacity to capture subtle yet critical temporal shifts, significantly improving the accuracy of time series modeling. Extensive experiments on both synthetic and real-world datasets demonstrate the effectiveness of DIFFODE across three key tasks, including irregular time series classification, interpolation, and extrapolation.
Yudong Zhang 0005, Xu Wang 0029, Zhengyang Zhou, Lei Bai 0001, Yang Wang 0015
ICDE6
2023 Long-Tailed Time Series Classification via Feature Space Rebalancing
Pengkun Wang 0001, Xu Wang 0029, Binwu Wang, Yudong Zhang 0005, Lei Bai 0001, Yang Wang 0015
DASFAA (1)5
2023 A Knowledge-Driven Memory System for Traffic Flow Prediction
Binwu Wang, Yudong Zhang 0005, Pengkun Wang 0001, Xu Wang 0029, Lei Bai 0001, Yang Wang 0015
DASFAA (4)5
2023 MM-DAG: Multi-task DAG Learning for Multi-modal Data - with Application for Traffic Congestion Analysis
abstract
This paper proposes to learn Multi-task, Multi-modal Direct Acyclic Graphs (MM-DAGs), which are commonly observed in complex systems, e.g., traffic, manufacturing, and weather systems, whose variables are multi-modal with scalars, vectors, and functions. This paper takes the traffic congestion analysis as a concrete case, where a traffic intersection is usually regarded as a DAG. In a road network of multiple intersections, different intersections can only have someoverlapping and distinct variables observed. For example, a signalized intersection has traffic light-related variables, whereas unsignalized ones do not. This encourages the multi-task design: with each DAG as a task, the MM-DAG tries to learn the multiple DAGs jointly so that their consensus and consistency are maximized. To this end, we innovatively propose a multi-modal regression for linear causal relationship description of different variables. Then we develop a novel Causality Difference (CD) measure and its differentiable approximator. Compared with existing SOTA measures, CD can penalize the causal structural difference among DAGs with distinct nodes and can better consider the uncertainty of causal orders. We rigidly prove our design's topological interpretation and consistency properties. We conduct thorough simulations and one case study to show the effectiveness of our MM-DAG. The code is available under https://github.com/Lantian72/MM-DAG.
Ziyue Li 0002, Zhishuai Li, Lei Bai 0001, Man Li 0003, Fugee Tsung, Wolfgang Ketter, Rui Zhao 0001, Chen Zhang 0007
KDD4
2023 Pattern Expansion and Consolidation on Evolving Graphs for Continual Traffic Prediction
abstract
Recently, spatiotemporal graph convolutional networks are becoming popular in the field of traffic flow prediction and significantly improve prediction accuracy. However, the majority of existing traffic flow prediction models are tailored to static traffic networks and fail to model the continuous evolution and expansion of traffic networks. In this work, we move to investigate the challenge of traffic flow prediction on an expanding traffic network. And we propose an efficient and effective continual learning framework to achieve continuous traffic flow prediction without the access to historical graph data, namely Pattern Expansion and Consolidation based on Pattern Matching based (PECPM). Specifically, we first design a pattern bank based on pattern matching to store representative patterns of the road network. With the expansion of the road network, the model configured with such a bank module can achieve continuous traffic prediction by effectively managing patterns stored in the bank. The core idea is to continuously update new patterns while consolidating learned ones. Specifically, we design a pattern expansion mechanism that can detect evolved and new patterns from the updated network, then these unknown patterns are expanded into the pattern bank to adapt to the updated road network. Additionally, we propose a pattern consolidation mechanism that includes both a bank preservation mechanism and a pattern traceability mechanism. This can effectively consolidate the learned patterns in the bank without requiring access to detailed historical graph data. We construct experiments on real-world traffic datasets to demonstrate the competitive performance, superior efficiency, and strong generalization ability of PECPM.
Binwu Wang, Yudong Zhang 0005, Xu Wang 0029, Pengkun Wang 0001, Zhengyang Zhou, Lei Bai 0001, Yang Wang 0015
KDD6
2022 Jointly Contrastive Representation Learning on Road Network and Trajectory
abstract
Road network and trajectory representation learning are essential for traffic systems since the learned representation can be directly used in various downstream tasks (e.g., traffic speed inference, travel time estimation). However, most existing methods only contrast within the same scale, i.e., treating road network and trajectory separately, which ignores valuable inter-relations. In this paper, we aim to propose a unified framework that jointly learns the road network and trajectory representations end-to-end. We design domain-specific augmentations for road-road contrast and trajectory-trajectory contrast separately, i.e., road segment with its contextual neighbors and trajectory with its detour replaced and dropped alternatives, respectively. On top of that, we further introduce the road-trajectory cross-scale contrast to bridge the two scales by maximizing the total mutual information. Unlike the existing cross-scale contrastive learning methods on graphs that only contrast a graph and its belonging nodes, the contrast between road segment and trajectory is elaborately tailored via novel positive sampling and adaptive weighting strategies. We conduct prudent experiments based on two real-world datasets with four downstream tasks, demonstrating improved performance and effectiveness.
Zhenyu Mao, Ziyue Li 0002, Dedong Li, Lei Bai 0001, Rui Zhao 0001
CIKM4
2022 Countering Modal Redundancy and Heterogeneity: A Self-Correcting Multimodal Fusion
abstract
Fusing multimodal heterogeneous data plays a vital role in recognition and prediction tasks in various fields, e.g., action recognition and traffic accident forecast. Yet, there remain some key challenges, such as heterogeneous feature interaction and feature redundancies, that significantly affect the performance of multimodal fusion. To tackle these challenges, we first devise a Unified Feature Interaction Module (UFIM) in which a novel orthogonal attention component is designed to obtain fine-grained inter-modal interaction information among heterogeneous features. Then, we propose a novel Self-Correcting Transformer Module (SCTM) which employs a modified transformer to obtain the one-to-many correlation information between the current modal feature and the merged features of other modalities to alleviate the redundancy problem. Extensive experiments on four cross-domain tasks demonstrate the effectiveness and generalization ability of our proposed method.
Pengkun Wang 0001, Xu Wang 0029, Binwu Wang, Yudong Zhang 0005, Lei Bai 0001, Yang Wang 0015
ICDM5
2020 Knowledge Adaption for Demand Prediction based on Multi-task Memory Neural Network
abstract
Accurate demand forecasting of different public transport modes (e.g., buses and light rails) is essential for public service operation. However, the development level of various modes often varies significantly, which makes it hard to predict the demand of the modes with insufficient knowledge and sparse station distribution (i.e., station-sparse mode). Intuitively, different public transit modes may exhibit shared demand patterns temporally and spatially in a city. As such, we propose to enhance the demand prediction of station-sparse modes with the data from station-intensive mode and design a Memory-Augmented Multi-task Re current Network (MATURE) to derive the transferable demand patterns from each mode and boost the prediction of station-sparse modes through adapting the relevant patterns from the station-intensive mode. Specifically, MATURE comprises three components: 1) a memory-augmented recurrent network for strengthening the ability to capture the long-short term information and storing temporal knowledge of each transit mode; 2) a knowledge adaption module to adapt the relevant knowledge from a station-intensive source to station-sparse sources; 3) a multi-task learning framework to incorporate all the information and forecast the demand of multiple modes jointly. The experimental results on a real-world dataset covering four public transport modes demonstrate that our model can promote the demand forecasting performance for the station-sparse modes.
Can Li 0014, Lei Bai 0001, Wei Liu 0101, Lina Yao 0001, S. Travis Waller
CIKM2
2020 Spectrum-Guided Adversarial Disparity Learning
abstract
It has been a significant challenge to portray intraclass disparity precisely in the area of activity recognition, as it requires a robust representation of the correlation between subject-specific variation for each activity class. In this work, we propose a novel end-to-end knowledge directed adversarial learning framework, which portrays the class-conditioned intraclass disparity using two competitive encoding distributions and learns the purified latent codes by denoising learned disparity. Furthermore, the domain knowledge is incorporated in an unsupervised manner to guide the optimization and further boosts the performance. The experiments on four HAR benchmark datasets demonstrate the robustness and generalization of our proposed methods over a set of state-of-the-art. We further prove the effectiveness of automatic domain knowledge incorporation in performance enhancement.
Zhe Liu 0023, Lina Yao 0001, Lei Bai 0001, Xianzhi Wang 0001, Can Wang 0004
KDD3
2020 Prototype Similarity Learning for Activity Recognition
abstract
Human Activity Recognition (HAR) plays an irreplaceable role in various applications such as security, gaming, and assisted living. Recent studies introduce deep learning to mitigate the manual feature extraction (i.e., data representation) efforts and achieve high accuracy. However, there are still challenges in learning accurate representations for sensory data due to the weakness of representation modules and the subject variances. We propose a scheme called Distance-based HAR from Ensembled spatial-temporal Representations (DHARER) to address above challenges. The idea behind DHARER is straightforward—the same activities should have similar representations. We first learn representations of the input sensory segments and latent prototype representations of each class, using a Convolution Neural Network (CNN)-based dual-stream representation module; then the learned representations are projected to activity types by measuring their similarity to the learned prototypes. We have conducted extensive experiments under a strict subject-independent setting on three large-scale datasets to evaluate the proposed scheme, and our experimental results demonstrate superior performance of DHARER to several state-of-the-art methods.
Lei Bai 0001, Lina Yao 0001, Xianzhi Wang 0001, Salil S. Kanhere, Yang Xiao 0014
PAKDD (1)1
2020 Mobility Irregularity Detection with Smart Transit Card Data
Xuesong Wang 0002, Lina Yao 0001, Wei Liu 0101, Can Li 0014, Lei Bai 0001, S. Travis Waller
PAKDD (1)5
2019 Spatio-Temporal Graph Convolutional and Recurrent Networks for Citywide Passenger Demand Prediction
abstract
Online ride-sharing platforms have become a critical part of the urban transportation system. Accurately recommending hotspots to drivers in such platforms is essential to help drivers find passengers and improve users' experience, which calls for efficient passenger demand prediction strategy. However, predicting multi-step passenger demand is challenging due to its high dynamicity, complex dependencies along spatial and temporal dimensions, and sensitivity to external factors (meteorological data and time meta). We propose an end-to-end deep learning framework to address the above problems. Our model comprises three components in pipeline: 1) a cascade graph convolutional recurrent neural network to accurately extract the spatial-temporal correlations within citywide historical passenger demand data; 2) two multi-layer LSTM networks to represent the external meteorological data and time meta, respectively; 3) an encoder-decoder module to fuse the above two parts and decode the representation to predict over multi-steps into the future. The experimental results on three real-world datasets demonstrate that our model can achieve accurate prediction and outperform the most discriminative state-of-the-art methods.
Lei Bai 0001, Lina Yao 0001, Salil S. Kanhere, Xianzhi Wang 0001, Wei Liu 0101, Zheng Yang 0002
CIKM1
2019 Passenger Demand Forecasting with Multi-Task Convolutional Recurrent Neural Networks
Lei Bai 0001, Lina Yao 0001, Salil S. Kanhere, Zheng Yang 0002, Jing Chu, Xianzhi Wang 0001
PAKDD (2)1