Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data. A simulation from public publication records: the people listed were not invited and did not agree to review.
? Tutorial for this page

Huaiyu Wan

dblp:07/9988 · DBLP profile ↗
← Back
86ranked-venue papers
6as first author
73since 2021 · last 2026
0000-0002-0501-9363ORCID · verified

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 48 · 2 first-author · 41 since 2021Databases, data management, data science and information retrieval · 35 · 2 first-author · 34 since 2021Graphics, computer vision, multimedia, augmented reality and games · 14 · 12 since 2021Applied, interdisciplinary, general and emerging computing · 12 · 3 first-author · 6 since 2021Computer networks · 1 · 1 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
34 papers
Graph learning · 15% Reinforcement learning · 14% Representation and self-supervised learning · 14%
Databases, data mining, and information retrieval
28 papers
Data mining · 31% Spatial and temporal data management · 27% Knowledge graphs · 26%
Interdisciplinary, comprehensive, and emerging computing
14 papers
Smart cities and intelligent transportation · 100%

Topics — the 30 heaviest of 93, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Data mining
spatiotemporal data mining
4.572025
TrajCogn: Leveraging LLMs for Cognizing Movement Patterns and Travel Purposes from Trajectories · IJCAI 2025
STD-PLM: Understanding Both Spatial and Temporal Properties of Spatial-Temporal Data with PLM · AAAI 2025
LaDe: The First Comprehensive Last-mile Express Dataset from Industry · KDD 2024
Smart cities and intelligent transportation
traffic prediction
3.662026
RIPCN: A Road Impedance Principal Component Network for Probabilistic Traffic Flow Forecasting · KDD (1) 2026
Proactive-XLight: Proactive Traffic Signal Control With Pluggable and Reliable Traffic Prediction · IEEE Trans. Mob. Comput. 2025
Self-Supervised Spatial-Temporal Bottleneck Attentive Network for Efficient Long-term Traffic Forecasting · ICDE 2023
Knowledge graphs
temporal knowledge graph
2.732026
STK-Adapter: Incorporating Evolving Graph and Event Chain for Temporal Knowledge Graph Extrapolation · ACL (1) 2026
A Generative Adaptive Replay Continual Learning Model for Temporal Knowledge Graph Reasoning · ACL (1) 2025
CognTKE: A Cognitive Temporal Knowledge Extrapolation Framework · AAAI 2025
Machine learning › Graph learning
graph neural network
2.442026
FaST: Efficient and Effective Long-Horizon Forecasting for Large-Scale Spatial-Temporal Graphs via Mixture-of-Experts · KDD (1) 2026
GMDNet: A Graph-Based Mixture Density Network for Estimating Packages' Multimodal Travel Time Distribution · AAAI 2023
Graph2Route: A Dynamic Spatial-Temporal Graph Neural Network for Pick-up and Delivery Route Prediction · KDD 2022
Spatial and temporal data management
trajectory data management
2.432025
UVTM: Universal Vehicle Trajectory Modeling With ST Feature Domain Generation · IEEE Trans. Knowl. Data Eng. 2025
Micro-Macro Spatial-Temporal Graph-Based Encoder-Decoder for Map-Constrained Trajectory Recovery · IEEE Trans. Knowl. Data Eng. 2024
Diff-RNTraj: A Structure-Aware Diffusion Model for Road Network-Constrained Trajectory Generation · IEEE Trans. Knowl. Data Eng. 2024
Spatial and temporal data management
trajectory data
2.242025
TransferTraj: A Vehicle Trajectory Learning Model for Region and Task Transferability · NeurIPS 2025
PLMTrajRec: A Scalable and Generalizable Trajectory Recovery Method with Pre-trained Language Models · NeurIPS 2025
UniTE: A Survey and Unified Pipeline for Pre-Training Spatiotemporal Trajectory Embeddings · IEEE Trans. Knowl. Data Eng. 2025
Machine learning › Representation and self-supervised learning
pre-training
2.232025
UniTE: A Survey and Unified Pipeline for Pre-Training Spatiotemporal Trajectory Embeddings · IEEE Trans. Knowl. Data Eng. 2025
Pre-Training General Trajectory Embeddings With Maximum Multi-View Entropy Coding · IEEE Trans. Knowl. Data Eng. 2024
Pre-Training Time-Aware Location Embeddings from Spatial-Temporal Trajectories · IEEE Trans. Knowl. Data Eng. 2022
Data mining › spatiotemporal data mining
trajectory data mining
2.132025
TrajCogn: Leveraging LLMs for Cognizing Movement Patterns and Travel Purposes from Trajectories · IJCAI 2025
Origin-Destination Travel Time Oracle for Map-based Services · Proc. ACM Manag. Data 2023
Graph2Route: A Dynamic Spatial-Temporal Graph Neural Network for Pick-up and Delivery Route Prediction · KDD 2022
Machine learning › Generative modeling
diffusion model
1.932026
STCDM: Spatio-Temporal Contrastive Diffusion Model for Check-In Sequence Generation · IEEE Trans. Knowl. Data Eng. 2025
DiffLight: A Partial Rewards Conditioned Diffusion Model for Traffic Signal Control with Missing Data · NeurIPS 2024
Spatial-Temporal Feedback Diffusion Guidance for Controlled Traffic Imputation · AAAI 2026
Knowledge graphs › temporal knowledge graph
spatial-temporal knowledge graph
1.722025
Next-POI Recommendation via Spatial-Temporal Knowledge Graph Contrastive Learning and Trajectory Prompt · IEEE Trans. Knowl. Data Eng. 2025
Spatial-Temporal Knowledge Distillation for Takeaway Recommendation · AAAI 2025
Machine learning › Representation and self-supervised learning › pre-training › spatio-temporal pre-training
trajectory embedding pre-training
1.622025
UniTE: A Survey and Unified Pipeline for Pre-Training Spatiotemporal Trajectory Embeddings · IEEE Trans. Knowl. Data Eng. 2025
Pre-Training General Trajectory Embeddings With Maximum Multi-View Entropy Coding · IEEE Trans. Knowl. Data Eng. 2024
Knowledge graphs › temporal knowledge graph
temporal knowledge graph reasoning
1.622025
A Generative Adaptive Replay Continual Learning Model for Temporal Knowledge Graph Reasoning · ACL (1) 2025
Local-Global History-Aware Contrastive Learning for Temporal Knowledge Graph Reasoning · ICDE 2024
Spatial and temporal data management › trajectory data management
trajectory recovery
1.622025
PLMTrajRec: A Scalable and Generalizable Trajectory Recovery Method with Pre-trained Language Models · NeurIPS 2025
Micro-Macro Spatial-Temporal Graph-Based Encoder-Decoder for Map-Constrained Trajectory Recovery · IEEE Trans. Knowl. Data Eng. 2024
Spatial and temporal data management › trajectory analysis
travel time estimation
1.522025
DutyTTE: Deciphering Uncertainty in Origin-Destination Travel Time Estimation · AAAI 2025
Origin-Destination Travel Time Oracle for Map-based Services · Proc. ACM Manag. Data 2023
Machine learning › Representation and self-supervised learning
contrastive learning
1.422024
Spatial-Temporal Cross-View Contrastive Pre-Training for Check-in Sequence Representation Learning · IEEE Trans. Knowl. Data Eng. 2024
Contrastive Pre-training with Adversarial Perturbations for Check-In Sequence Representation Learning · AAAI 2023
Machine learning › Time series and sequential data › time series analysis › time series forecasting
long-term time series forecasting
1.422024
PGN: The RNN's New Successor is Effective for Long-Range Time Series Forecasting · NeurIPS 2024
WITRAN: Water-wave Information Transmission and Recurrent Acceleration Network for Long-range Time Series Forecasting · NeurIPS 2023
Machine learning › Deep learning architectures and training
recurrent neural network
1.422024
PGN: The RNN's New Successor is Effective for Long-Range Time Series Forecasting · NeurIPS 2024
WITRAN: Water-wave Information Transmission and Recurrent Acceleration Network for Long-range Time Series Forecasting · NeurIPS 2023
Machine learning › Time series and sequential data › time series analysis
time series forecasting
1.422024
PGN: The RNN's New Successor is Effective for Long-Range Time Series Forecasting · NeurIPS 2024
WITRAN: Water-wave Information Transmission and Recurrent Acceleration Network for Long-range Time Series Forecasting · NeurIPS 2023
Machine learning › Graph learning
spatio-temporal graph learning
1.332023
M2G4RTP: A Multi-Level and Multi-Task Graph Model for Instant-Logistics Route and Time Joint Prediction · ICDE 2023
Spatial-Temporal Synchronous Graph Convolutional Networks: A New Framework for Spatial-Temporal Network Data Forecasting · AAAI 2020
Learning Dynamics and Heterogeneity of Spatial-Temporal Graph Data for Traffic Forecasting · IEEE Trans. Knowl. Data Eng. 2022
Machine learning and data management
uncertainty quantification
1.122025
DutyTTE: Deciphering Uncertainty in Origin-Destination Travel Time Estimation · AAAI 2025
Proactive-XLight: Proactive Traffic Signal Control With Pluggable and Reliable Traffic Prediction · IEEE Trans. Mob. Comput. 2025
Spatial and temporal data management › trajectory prediction
route prediction
1.122022
Graph2Route: A Dynamic Spatial-Temporal Graph Neural Network for Pick-up and Delivery Route Prediction · KDD 2022
Package Pick-up Route Prediction via Modeling Couriers' Spatial-Temporal Behaviors · ICDE 2021
Knowledge, reasoning and agents › Multi-agent systems › multi-agent coordination
implicit coordination
1.012026
Think How Your Teammates Think: Active Inference Can Benefit Decentralized Execution · AAAI 2026
Machine learning › Deep learning architectures and training
mixture of experts
1.012026
FaST: Efficient and Effective Long-Horizon Forecasting for Large-Scale Spatial-Temporal Graphs via Mixture-of-Experts · KDD (1) 2026
Machine learning › Reinforcement learning
multi-agent reinforcement learning
1.012026
Think How Your Teammates Think: Active Inference Can Benefit Decentralized Execution · AAAI 2026
Machine learning › Graph learning › spatio-temporal graph learning
spatio-temporal graph forecasting
1.012026
FaST: Efficient and Effective Long-Horizon Forecasting for Large-Scale Spatial-Temporal Graphs via Mixture-of-Experts · KDD (1) 2026
Knowledge, reasoning and agents › Multi-agent systems › human-agent interaction
teammate modeling
1.012026
Think How Your Teammates Think: Active Inference Can Benefit Decentralized Execution · AAAI 2026
Knowledge graphs › temporal knowledge graph
knowledge graph extrapolation
1.012026
STK-Adapter: Incorporating Evolving Graph and Event Chain for Temporal Knowledge Graph Extrapolation · ACL (1) 2026
Knowledge graphs
link prediction
1.012026
STK-Adapter: Incorporating Evolving Graph and Event Chain for Temporal Knowledge Graph Extrapolation · ACL (1) 2026
Machine learning › Graph learning › graph neural network › dynamic graph neural network
spatio-temporal graph neural network
1.022022
Graph2Route: A Dynamic Spatial-Temporal Graph Neural Network for Pick-up and Delivery Route Prediction · KDD 2022
Attention Based Spatial-Temporal Graph Convolutional Networks for Traffic Flow Forecasting · AAAI 2019
Machine learning › Reinforcement learning
exploration
0.912025
Enhancing Efficiency and Exploration in Reinforcement Learning for LLMs · EMNLP 2025

Methods — techniques the papers use, named apart from their topics

graph neural network · 5.5pre-trained language model · 5.2mixture of experts · 4.5contrastive learning · 4.2pre-training · 3.5large language model · 3.3deep reinforcement learning · 3.0multi-task learning · 2.6diffusion model · 2.3encoder-decoder · 1.3score-based diffusion · 1.0principal component network · 1.0feedback control · 1.0event chain modeling · 1.0attention-based clustering · 1.0graph convolution · 1.0transfer learning · 0.9multi-task joint optimization · 0.9
YearPublicationVenuePosition
2026 Spatial-Temporal Feedback Diffusion Guidance for Controlled Traffic Imputation
abstract
Imputing missing values in spatial-temporal traffic data is essential for intelligent transportation systems. Among advanced imputation methods, score-based diffusion models have demonstrated competitive performance. These models generate data by reversing a noising process, using observed values as conditional guidance. However, existing diffusion models typically apply a uniform guidance scale across both spatial and temporal dimensions, which is inadequate for nodes with high missing data rates. Sparse observations provide insufficient conditional guidance, causing the generative process to drift toward the learned prior distribution rather than closely following the conditional observations, resulting in suboptimal imputation performance. To address this, we propose FENCE, a spatial-temporal feedback diffusion guidance method designed to adaptively control guidance scales during imputation. First, FENCE introduces a dynamic feedback mechanism that adjusts the guidance scale based on the posterior likelihood approximations. The guidance scale is increased when generated values diverge from observations and reduced when alignment improves, preventing overcorrection. Second, because alignment to observations varies across nodes and denoising steps, a global guidance scale for all nodes is suboptimal. FENCE computes guidance scales at the cluster level by grouping nodes based on their attention scores, leveraging spatial-temporal correlations to provide more accurate guidance. Experimental results on real-world traffic datasets show that FENCE significantly enhances imputation accuracy.
Xiaowei Mao, Huihu Ding, Yan Lin 0006, Tingrui Wu, Shengnan Guo 0001, Dazhuo Qiu, Feiling Fang, Jilin Hu, Huaiyu Wan
AAAI9
2026 Think How Your Teammates Think: Active Inference Can Benefit Decentralized Execution
abstract
In multi-agent systems, explicit cognition of teammates' decision logic serves as a critical factor in facilitating coordination. Communication (i.e., "Tell") can assist in the cognitive development process by information dissemination, yet it is inevitably subject to real-world constraints such as noise, latency, and attacks. Therefore, building the understanding of teammates' decisions without communication remains challenging. To address this, we propose a novel non-communication MARL framework that realizes the construction of cognition through local observation-based modeling (i.e., "Think"). Our framework enables agents to model teammates' active inference process. At first, the proposed method produces three teammate portraits: perception-belief-action. Specifically, we model the teammate's decision process as follows: 1) Perception: observing environments; 2) Belief: forming beliefs; 3) Action: making decisions. Then, we selectively integrate the belief portrait into the decision process based on the accuracy and relevance of the perception portrait. This enables the selection of cooperative teammates and facilitates effective collaboration. Extensive experiments on the SMAC, SMACv2, MPE, and GRF benchmarks demonstrate the superior performance of our method.
Hao Wu 0010, Shoucheng Song, Sheng Han 0001, Huaiyu Wan, Youfang Lin, Kai Lv 0002
AAAI5
2026 STK-Adapter: Incorporating Evolving Graph and Event Chain for Temporal Knowledge Graph Extrapolation
abstract
Shuyuan Zhao, Wei Chen, Weijie Zhang, Xinrui Hou, Junfeng Shen, Boyan Shi, Shengnan Guo, Youfang Lin, Huaiyu Wan. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026.
Shuyuan Zhao 0001, Wei Chen 0105, Xinrui Hou, Junfeng Shen, Boyan Shi, Shengnan Guo 0001, Youfang Lin, Huaiyu Wan
ACL (1)9
2026 ForceTraj: Modeling Realistic Intention and Heterogeneous Interactions for Multi-modal Trajectory Prediction
Guanyu Yao, Yan Lin 0006, Letian Gong, Yiwei Shuang, Shengnan Guo 0001, Huaiyu Wan
DASFAA (5)7
2026 RIPCN: A Road Impedance Principal Component Network for Probabilistic Traffic Flow Forecasting
Haochen Lv 0001, Yan Lin 0006, Shengnan Guo 0001, Xiaowei Mao, Letian Gong, Youfang Lin, Huaiyu Wan
KDD (1)8
2026 FaST: Efficient and Effective Long-Horizon Forecasting for Large-Scale Spatial-Temporal Graphs via Mixture-of-Experts
abstract
Spatial-Temporal Graph (STG) forecasting on large-scale networks has garnered significant attention. However, existing models predominantly focus on short-horizon predictions and suffer from notorious computational costs and memory consumption when scaling to long-horizon predictions and large graphs. Targeting the above challenges, we present FaST, an effective and efficient framework based on heterogeneity-aware Mixture-of-Experts (MoEs) for long-horizon and large-scale STG forecasting, which unlocks one-week-ahead (672 steps at a 15-minute granularity) prediction with thousands of nodes. FaST is underpinned by two key innovations. First, an adaptive graph agent attention mechanism is proposed to alleviate the computational burden inherent in conventional graph convolution and self-attention modules when applied to large-scale graphs. Second, we propose a new parallel MoE module that replaces traditional feed-forward networks with Gated Linear Units (GLUs), enabling an efficient and scalable parallel structure. Extensive experiments on real-world datasets demonstrate that FaST not only delivers superior long-horizon predictive accuracy but also achieves remarkable computational efficiency compared to state-of-the-art baselines. Our source code is available at: https://github.com/yijizhao/FaST.
Yiji Zhao, Zihao Zhong, Haomin Wen, Ming Jin 0005, Yuxuan Liang 0002, Huaiyu Wan, Hao Wu 0010
KDD (1)7
2026 TARP: An Effective and Efficient Route Prediction Method via Uncertainty-Aware Traffic Imputation and Divide-and-Conquer Inference
Xingyu Xian, Xiaowei Mao, Yang Jiang 0005, Tingrui Wu, Shengnan Guo 0001, Youfang Lin, Huaiyu Wan
PAKDD (1)7
2026 Multi-faceted dynamic flexible job shop scheduling via heterogeneous graph transformer and deep reinforcement learning
Mengqi Liao, Wei Chen 0105, Huaiyu Wan, Youfang Lin
Expert Syst. Appl.5
2026 Trajectory anomaly detection via spatial-temporal complementary reconstruction diffusion model
Tonglong Wei, Tianzi Yan, Jingsong Xu, Qisen Xu, Shengnan Guo 0001, Huaiyu Wan, Youfang Lin
GeoInformatica7
2026 DRL4AOI: A DRL Framework for Semantic-aware AOI Segmentation in Location-based Services
abstract
In Location-Based Services (LBS), such as food delivery, a fundamental task is segmenting Areas of Interest (AOIs), aiming at partitioning the urban geographical spaces into non-overlapping regions. Traditional AOI segmentation algorithms primarily rely on road networks to partition urban areas. While promising in modeling the geo-semantics, road network-based models overlooked the service-semantic goals (e.g., workload equality) in LBS service. In this article, we point out that the AOI segmentation problem can be naturally formulated as a Markov Decision Process (MDP), which gradually chooses a nearby AOI for each grid in the current AOI’s border. Based on the MDP, we present the first attempt to generalize Deep Reinforcement Learning (DRL) for AOI segmentation, leading to a novel DRL-based framework called DRL4AOI. The DRL4AOI framework introduces different service-semantic goals in a flexible way by treating them as rewards that guide the AOI generation. To evaluate the effectiveness of DRL4AOI, we develop and release an AOI segmentation system. We also present a representative implementation of DRL4AOI—TrajRL4AOI—for AOI segmentation in the logistics service. It introduces a Double Deep Q-learning Network (DDQN) to gradually optimize the AOI generation for two specific semantic goals: (i) trajectory modularity, i.e., maximize tightness of the trajectory connections within an AOI and the sparsity of connections between AOIs, (ii) matchness with the road network, i.e., maximizing the matchness between AOIs and the road network. Quantitative and qualitative experiments conducted on synthetic and real-world data demonstrate the effectiveness and superiority of our method. The code and system is publicly available at https://github.com/Kogler7/AoiOpt .
Youfang Lin, Jinji Fu, Haomin Wen, Jiyuan Wang 0001, Zhenjie Wei, Yuting Qiang, Xiaowei Mao, Lixia Wu, Haoyuan Hu, Yuxuan Liang 0002, Huaiyu Wan
ACM Trans. Intell. Syst. Technol.11
2026 Sparse Traffic Accident Risk Forecasting With Spatial-Temporal Knowledge Graphs
Shengnan Guo 0001, Yan Lin 0006, Wei Chen 0105, Weiwen Tang, Haochen Lv 0001, Rongzhi Zhou, Junliang Lin, Youfang Lin, Huaiyu Wan
IEEE Trans. Knowl. Data Eng.9
2025 CognTKE: A Cognitive Temporal Knowledge Extrapolation Framework
abstract
Reasoning future unknowable facts on temporal knowledge graphs (TKGs) is a challenging task, holding significant academic and practical values for various fields. Existing studies exploring explainable reasoning concentrate on modeling comprehensible temporal paths relevant to the query. Yet, these path-based methods primarily focus on local temporal paths appearing in recent times, failing to capture the complex temporal paths in TKG and resulting in the loss of longer historical relations related to the query. Motivated by the Dual Process Theory in cognitive science, we propose a Cognitive Temporal Knowledge Extrapolation framework (CognTKE), which introduces a novel temporal cognitive relation directed graph (TCR-Digraph) and performs interpretable global shallow reasoning and local deep reasoning over the TCR-Digraph. Specifically, the proposed TCR-Digraph is constituted by retrieving significant local and global historical temporal relation paths associated with the query. In addition, CognTKE presents the global shallow reasoner and the local deep reasoner to perform global one-hop temporal relation reasoning (System 1) and local complex multi-hop path reasoning (System 2) over the TCR-Digraph, respectively. The experimental results on four benchmark datasets demonstrate that CognTKE achieves significant improvement in accuracy compared to the state-of-the-art baselines and delivers excellent zero-shot reasoning ability.
Wei Chen 0105, Shuhan Wu, Mengqi Liao, Youfang Lin, Huaiyu Wan
AAAI7
2025 STD-PLM: Understanding Both Spatial and Temporal Properties of Spatial-Temporal Data with PLM
abstract
Spatial-temporal forecasting and imputation are important for real-world intelligent systems. Most existing methods are tailored for individual forecasting or imputation tasks but are not designed for both. Additionally, they are less effective for zero-shot and few-shot learning. While pre-trained language model (PLM) have exhibited strong pattern recognition and reasoning abilities across various tasks, including few-shot and zero-shot learning, their applications in spatial-temporal data understanding has been constrained by insufficient modeling of complex correlations such as the temporal correlations, spatial connectivity, non-pairwise and high-order spatial-temporal correlations within data. In this paper, we propose STD-PLM for understanding both spatial and temporal properties of Spatial-Temporal Data with PLM, which is capable of implementing both spatial-temporal forecasting and imputation tasks. STD-PLM understands spatial-temporal correlations via explicitly designed spatial and temporal tokenizers. Topology-aware node embeddings are designed for PLM to comprehend and exploit the topology structure of data in inductive manner. Furthermore, to mitigate the efficiency issues introduced by the PLM, we design a sandglass attention module(SGA) combined with a specific constrained loss function, which significantly improves the model's efficiency while ensuring performance. Extensive experiments demonstrate that STD-PLM exhibits competitive performance and generalization capabilities across the forecasting and imputation tasks on various datasets. Moreover, STD-PLM achieves promising results on both few-shot and zero-shot tasks.
Xiaowei Mao, Shengnan Guo 0001, Yubin Chen, Junfeng Shen, Tiankuo Li, Youfang Lin, Huaiyu Wan
AAAI8
2025 DutyTTE: Deciphering Uncertainty in Origin-Destination Travel Time Estimation
abstract
Uncertainty quantification in travel time estimation (TTE) aims to estimate the confidence interval for travel time, given the origin (O), destination (D), and departure time (T). Accurately quantifying this uncertainty requires generating the most likely path and assessing travel time uncertainty along the path. This involves two main challenges: 1) Predicting a path that aligns with the ground truth, and 2) modeling the impact of travel time in each segment on overall uncertainty under varying conditions. We propose DutyTTE to address these challenges. For the first challenge, we introduce a deep reinforcement learning method to improve alignment between the predicted path and the ground truth, providing more accurate travel time information from road segments to improve TTE. For the second challenge, we propose a mixture of experts guided uncertainty quantification mechanism to better capture travel time uncertainty for each segment under varying contexts. Extensive experiments on two real-world datasets demonstrate the superiority of our proposed method.
Xiaowei Mao, Yan Lin 0006, Shengnan Guo 0001, Yubin Chen, Xingyu Xian, Haomin Wen, Qisen Xu, Youfang Lin, Huaiyu Wan
AAAI9
2025 Spatial-Temporal Knowledge Distillation for Takeaway Recommendation
abstract
The takeaway recommendation system aims to recommend users' future takeaway purchases based on their historical purchase behaviors, thereby improving user satisfaction and boosting merchant sales. Existing methods focus on incorporating auxiliary information or leveraging knowledge graphs to alleviate the sparsity issue of user purchase sequences. However, two main challenges limit the performance of these approaches: (1) capturing dynamic user preferences on complex geospatial information and (2) efficiently integrating spatial-temporal knowledge from both graphs and sequence data with low computational costs. In this paper, we propose a novel spatial-temporal knowledge distillation model for takeaway recommendation (STKDRec) based on the two-stage training process. Specifically, during the first pre-training stage, a spatial-temporal knowledge graph (STKG) encoder is trained to extract high-order spatial-temporal dependencies and collaborative associations from the STKG. During the second spatial-temporal knowledge distillation (STKD) stage, a spatial-temporal Transformer (ST-Transformer) is employed to comprehensively model dynamic user preferences on various types of fine-grained geospatial information from a sequential perspective. Furthermore, the STKD strategy is introduced to transfer graph-based spatial-temporal knowledge to the ST-Transformer, facilitating the adaptive fusion of rich knowledge derived from both the STKG and sequence data while reducing computational overhead. Extensive experiments on three real-world datasets show that STKDRec significantly outperforms the state-of-the-art baselines.
Shuyuan Zhao 0001, Wei Chen 0105, Boyan Shi, Liyong Zhou, Shuohao Lin, Huaiyu Wan
AAAI6
2025 A Generative Adaptive Replay Continual Learning Model for Temporal Knowledge Graph Reasoning
abstract
Recent Continual Learning (CL)-based Temporal Knowledge Graph Reasoning (TKGR) methods focus on significantly reducing computational cost and mitigating catastrophic forgetting caused by fine-tuning models with new data. However, existing CL-based TKGR methods still face two key limitations: (1) They usually one-sidedly reorganize individual historical facts, while overlooking the historical context essential for accurately understanding the historical semantics of these facts; (2) They preserve historical knowledge by simply replaying historical facts, while ignoring the potential conflicts between historical and emerging facts. In this paper, we propose a \textbf{D}eep \textbf{G}enerative \textbf{A}daptive \textbf{R}eplay (DGAR) method, which can generate and adaptively replay historical entity distribution representations from the whole historical context. To address the first challenge, historical context prompts as sampling units are built to preserve the whole historical context information. To overcome the second challenge, a pre-trained diffusion model is adopted to generate the historical distribution. During the generation process, the common features between the historical and current distributions are enhanced under the guidance of the TKGR model. In addition, a layer-by-layer adaptive replay mechanism is designed to effectively integrate historical and current distributions. Experimental results demonstrate that DGAR significantly outperforms baselines in reasoning and mitigating forgetting.
Wei Chen 0105, Youfang Lin, Huaiyu Wan
ACL (1)4
2025 Towards An Efficient and Effective En Route Travel Time Estimation Framework
Zekai Shen 0001, Haitao Yuan 0002, Xiaowei Mao, Congkang Lv, Shengnan Guo 0001, Youfang Lin, Huaiyu Wan
DASFAA (2)7
2025 Enhancing Efficiency and Exploration in Reinforcement Learning for LLMs
abstract
Reasoning large language models (LLMs) excel in complex tasks, which has drawn significant attention to reinforcement learning (RL) for LLMs. However, existing approaches allocate an equal number of rollouts to all questions during the RL process, which is inefficient. This inefficiency stems from the fact that training on simple questions yields limited gains, whereas more rollouts are needed for challenging questions to sample correct answers. Furthermore, while RL improves response precision, it limits the model’s exploration ability, potentially resulting in a performance cap below that of the base model prior to RL. To address these issues, we propose a mechanism for dynamically allocating rollout budgets based on the difficulty of the problems, enabling more efficient RL training. Additionally, we introduce an adaptive dynamic temperature adjustment strategy to maintain the entropy at a stable level, thereby encouraging sufficient exploration. This enables LLMs to improve response precision while preserving their exploratory ability to uncover potential correct pathways. The code and data is available on: https://anonymous.4open.science/r/E3-RL4LLMs-DB28
Mengqi Liao, Xiangyu Xi, Ruinian Chen, Jia Leng, Yangen Hu, Huaiyu Wan
EMNLP8
2025 HMoRA: Making LLMs More Effective with Hierarchical Mixture of LoRA Experts
abstract
Recent studies have combined Mixture of Experts (MoE) and Parameter-Efficient Fine-tuning (PEFT) to fine-tune large language models (LLMs), holding excellent performance in multi-task scenarios while remaining resource-efficient. However, existing MoE approaches still exhibit the following limitations: (1) Current methods fail to consider that different LLM layers capture features at varying levels of granularity, leading to suboptimal performance. (2) Task-level routing methods lack generalizability to unseen tasks. (3) The uncertainty introduced by load imbalance loss undermines the effective specialization of the experts. To address these challenges, we propose HMoRA, a Hierarchical fine-tuning method that combines MoE and LoRA, employing hybrid routing that integrates token-level and task-level routing in a hierarchical manner. This hierarchical hybrid routing allows the model to more efficiently capture both fine-grained token information and broader task contexts. To improve the certainty of expert selection, a novel routing auxiliary loss is introduced. This auxiliary function also enhances the task router's ability to differentiate tasks and its generalization to unseen tasks. Additionally, several optional lightweight designs have been proposed to significantly reduce both the number of trainable parameters and computational costs. Experimental results demonstrate that HMoRA outperforms full fine-tuning across multiple NLP benchmarks, while fine-tuning only 3.9\% of the parameters. The code is available on: https://github.com/LiaoMengqi/HMoRA.
Mengqi Liao, Wei Chen 0105, Junfeng Shen, Shengnan Guo 0001, Huaiyu Wan
ICLR5
2025 TrajCogn: Leveraging LLMs for Cognizing Movement Patterns and Travel Purposes from Trajectories
abstract
Spatio-temporal trajectories are crucial for data mining tasks, requiring versatile learning methods that can accurately extract movement patterns and travel purposes. While large language models (LLMs) have shown remarkable versatility through training on extensive datasets, and trajectories share similarities with natural language, standard LLMs cannot directly handle spatio-temporal features or extract trajectory-specific information. We propose TrajCogn, a model that effectively adapts LLMs for trajectory learning. TrajCogn incorporates a novel trajectory semantic embedder to process spatio-temporal features and extract movement patterns and travel purposes, along with a trajectory prompt that integrates this information into LLMs for various downstream tasks. Experiments on three real-world datasets and four representative tasks demonstrate TrajCogn's effectiveness.
Zeyu Zhou 0001, Yan Lin 0006, Haomin Wen, Shengnan Guo 0001, Jilin Hu, Youfang Lin, Huaiyu Wan
IJCAI7
2025 Balancing Imbalance: Data-Scarce Urban Flow Prediction via Spatio-Temporal Balanced Transfer Learning
abstract
Advanced deep spatio-temporal networks have become the mainstream for traffic prediction, but the widespread adoption of these models is impeded by the prevalent scarcity of available data. Despite cross-city transfer learning emerging as a common strategy to address this issue, it overlooks the inherent distribution imbalances within each city, which could potentially hinder the generalization capabilities of pre-trained models. To overcome this limitation, we propose a Spatio-Temporal Balanced Transfer Learning (STBaT) framework to enhance existing spatio-temporal prediction networks, ensuring both universality and precision in predictions for new urban environments. A Regional Imbalance Acquisition Module is designed to model the regional imbalances of source cities. Besides, to promote generalizable knowledge acquisition, a Spatio-Temporal Balanced Learning Module is devised to balance the predictive learning process. Extensive experiments on real-world datasets validate the efficacy of our proposed approach compared with several state-of-the-art methods.
Xinyan Hao, Huaiyu Wan, Shengnan Guo 0001, Youfang Lin
IJCAI2
2025 TrajMamba: An Efficient and Semantic-rich Vehicle Trajectory Pre-training Model
abstract
Vehicle GPS trajectories record how vehicles move over time, storing valuable travel semantics, including movement patterns and travel purposes. Learning travel semantics effectively and efficiently is crucial for real-world applications of trajectory data, which is hindered by two major challenges. First, travel purposes are tied to the functions of the roads and points-of-interest (POIs) involved in a trip. Such information is encoded in textual addresses and descriptions and introduces heavy computational burden to modeling. Second, real-world trajectories often contain redundant points, which harm both computational efficiency and trajectory embedding quality. To address these challenges, we propose TrajMamba, a novel approach for efficient and semantically rich vehicle trajectory learning. TrajMamba introduces a Traj-Mamba Encoder that captures movement patterns by jointly modeling both GPS and road perspectives of trajectories, enabling robust representations of continuous travel behaviors. It also incorporates a Travel Purpose-aware Pre-training procedure to integrate travel purposes into the learned embeddings without introducing extra overhead to embedding calculation. To reduce redundancy in trajectories, TrajMamba features a Knowledge Distillation Pre-training scheme to identify key trajectory points through a learnable mask generator and obtain effective compressed trajectory embeddings. Extensive experiments on two real-world datasets and three downstream tasks show that TrajMamba outperforms state-of-the-art baselines in both efficiency and accuracy.
Yichen Liu 0003, Yan Lin 0006, Shengnan Guo 0001, Zeyu Zhou 0001, Youfang Lin, Huaiyu Wan
NeurIPS6
2025 PLMTrajRec: A Scalable and Generalizable Trajectory Recovery Method with Pre-trained Language Models
abstract
Spatiotemporal trajectory data is crucial for various traffic-related applications. However, issues such as device malfunctions and network instability often result in sparse trajectories that lose detailed movement information compared to their dense counterparts. Recovering missing points in sparse trajectories is thus essential. Despite recent progress, three challenges remain. First, the lack of large-scale dense trajectory datasets hinders the training of a trajectory recovery model. Second, the varying spatiotemporal correlations in sparse trajectories make it hard to generalize across different sampling intervals. Third, extracting road conditions for missing points is non-trivial. To address these challenges, we propose $\textit{PLMTrajRec}$, a novel trajectory recovery model. It leverages the scalability of a pre-trained language model (PLM) and can effectively recover trajectories by fine-tuning with small-scale dense trajectory datasets. To handle different sampling intervals in sparse trajectories, we first convert sampling intervals and movement features into prompts for the PLM to understand. We then introduce a trajectory encoder to unify trajectories of varying intervals into a single interval. To extract road conditions for missing points, we propose an area flow-guided implicit trajectory prompt that represents traffic conditions in each region, and a road condition passing mechanism that infers the road conditions of missing points using the observed ones. Experiments on four public trajectory datasets with three sampling intervals demonstrate the effectiveness, scalability, and generalization ability of PLMTrajRec. Code is available at https://github.com/wtl52656/PLMTrajRec.
Tonglong Wei, Yan Lin 0006, Youfang Lin, Shengnan Guo 0001, Jilin Hu, Haitao Yuan 0002, Gao Cong, Huaiyu Wan
NeurIPS8
2025 TransferTraj: A Vehicle Trajectory Learning Model for Region and Task Transferability
abstract
Vehicle GPS trajectories provide valuable movement information that supports various downstream tasks and applications. A desirable trajectory learning model should be able to transfer across regions and tasks without retraining, avoiding the need to maintain multiple specialized models and subpar performance with limited training data. However, each region has its unique spatial features and contexts, which are reflected in vehicle movement patterns and are difficult to generalize. Additionally, transferring across different tasks faces technical challenges due to the varying input-output structures required for each task. Existing efforts towards transferability primarily involve learning embedding vectors for trajectories, which perform poorly in region transfer and require retraining of prediction modules for task transfer. To address these challenges, we propose $\textit{TransferTraj}$, a vehicle GPS trajectory learning model that excels in both region and task transferability. For region transferability, we introduce RTTE as the main learnable module within TransferTraj. It integrates spatial, temporal, POI, and road network modalities of trajectories to effectively manage variations in spatial context distribution across regions. It also introduces a TRIE module for incorporating relative information of spatial features and a spatial context MoE module for handling movement patterns in diverse contexts. For task transferability, we propose a task-transferable input-output scheme that unifies the input-output structure of different tasks into the masking and recovery of modalities and trajectory points. This approach allows TransferTraj to be pre-trained once and transferred to different tasks without retraining. We conduct extensive experiments on three real-world vehicle trajectory datasets under various transfer settings, including task transfer, zero-shot region transfer, and few-shot region transfer. Experimental results demonstrate that TransferTraj significantly outperforms state-of-the-art baselines in different scenarios, validating its effectiveness in region and task transfer. Code is available at https://github.com/wtl52656/TransferTraj.
Tonglong Wei, Yan Lin 0006, Zeyu Zhou 0001, Haomin Wen, Jilin Hu, Shengnan Guo 0001, Youfang Lin, Gao Cong, Huaiyu Wan
NeurIPS9
2025 UFN: User-Friendly Navigation Framework Based on Systematic Utilization of User-Friendliness Features from Historical Traffic Data
Shien Huang, Baixi Sun, Ling-Jun Fan, Xinyu Chen 0008, Jiannan Tian, Qianwen Shen, Huaiyu Wan, Dingwen Tao, Ergude Bao
J. Comput. Sci. Technol.7
2025 Dual-view temporal knowledge graph reasoning
Wei Chen 0105, Shengnan Guo 0001, Shuhan Wu, Zhishu Jiang, Youfang Lin, Huaiyu Wan
Knowl. Based Syst.7
2025 Next-POI Recommendation via Spatial-Temporal Knowledge Graph Contrastive Learning and Trajectory Prompt
abstract
Next POI (Point-of-Interest) recommendation aims to forecast users’ future movements based on their historical check-in trajectories, holding significant value in location-based services. Existing methods address trajectory data sparsity by integrating rich auxiliary information or using spatial-temporal knowledge graphs (STKGs), showing promising results. Yet, they face two main challenges: i) Due to the difficulty of transforming structured trajectory data into trajectory text describing users’ spatial-temporal mobility, the powerful reasoning ability of pre-trained language models is rarely explored to enhance recommendation performance. ii) Methods based on STKG can introduce external knowledge inconsistent with user preferences, leading to the knowledge noise generated hampering the accuracy of recommendations. To this end, we propose a novel approach called STKG-PLM that integratesSTKGcontrastive learning andprompt pre-trainedlanguagemodel (PLM) to enhance the next POI recommendation. Specifically, we design a spatial-temporal trajectory prompt template that transforms structured trajectories into text corpus based on STKG, serving as the input of PLM to understand the movement pattern of users from coarse-grained and fine-grained perspectives. Additionally, we propose an STKG contrastive learning framework to mitigate the introduced knowledge noise. Extensive experiments on three real-world datasets demonstrate that STKG-PLM exhibits notable performance improvements over the state-of-the-art baseline methods.
Wei Chen 0105, Youfang Lin, Liang Chang 0003, Huaiyu Wan
IEEE Trans. Knowl. Data Eng.7
2025 STCDM: Spatio-Temporal Contrastive Diffusion Model for Check-In Sequence Generation
abstract
Analyzing and comprehending check-in sequences is crucial for various applications in smart cities. However, publicly available check-in datasets are often limited in scale due to privacy concerns. This poses a significant obstacle to academic research and downstream applications. Thus, it is urgent to generate realistic check-in datasets. The denoising diffusion probabilistic model (DDPM) as one of the most capable generation methods is a good choice to achieve this goal. However, generating check-in sequences using DDPM is not an easy feat. The difficulties lie in handling check-in sequences of variable lengths and capturing the correlation from check-in sequences' distinct characteristics. This paper addresses the challenges by proposing a Spatio-Temporal Contrastive Diffusion Model (STCDM). This model introduces a novel spatio-temporal lossless encoding method that effectively encodes check-in sequences into a suitable format with equal length. Furthermore, we capture the spatio-temporal correlations with two disentangled diffusion modules to reduce the impact of the difference between spatial and temporal characteristics. Finally, we incorporate contrastive learning to enhance the relationship between diffusion modules. We generate four realistic datasets in different scenarios using STCDM and design four metrics for comparison. Experiments demonstrate that our generated datasets are more realistic and free of privacy leakage.
Letian Gong, Shengnan Guo 0001, Yan Lin 0006, Yichen Liu 0003, Erwen Zheng, Yiwei Shuang, Youfang Lin, Jilin Hu, Huaiyu Wan
IEEE Trans. Knowl. Data Eng.9
2025 UVTM: Universal Vehicle Trajectory Modeling With ST Feature Domain Generation
abstract
Vehicle movement is frequently captured in the form of GPS trajectories, i.e., sequences of timestamped GPS locations. Such data is widely used for various tasks such as travel-time estimation, trajectory recovery, and trajectory prediction. A universal vehicle trajectory model could be applied to different tasks, removing the need to maintain multiple specialized models, thereby reducing computational and storage costs. However, creating such a model is challenging when the integrity of trajectory features is compromised, i.e., in scenarios where only partial features are available or the trajectories are sparse. To address these challenges, we propose the Universal Vehicle Trajectory Model (UVTM), which can effectively adapt to different tasks without excessive retraining. UVTM incorporates two specialized designs. First, it divides trajectory features into three distinct domains. Each domain can be masked and generated independently to accommodate tasks with only partially available features. Second, UVTM is pre-trained by reconstructing dense, feature-complete trajectories from sparse, feature-incomplete counterparts, enabling strong performance even when the integrity of trajectory features is compromised. Experiments involving four representative trajectory-related tasks on three real-world vehicle trajectory datasets provide insight into the performance of UVTM and offer evidence that it is capable of meeting its objectives.
Yan Lin 0006, Jilin Hu, Shengnan Guo 0001, Bin Yang 0002, Christian S. Jensen, Youfang Lin, Huaiyu Wan
IEEE Trans. Knowl. Data Eng.7
2025 UniTE: A Survey and Unified Pipeline for Pre-Training Spatiotemporal Trajectory Embeddings
abstract
Spatiotemporal trajectories are sequences of timestamped locations, which enable a variety of analyses that in turn enable important real-world applications. It is common to map trajectories to vectors, called embeddings, before subsequent analyses. Thus, the qualities of embeddings are very important. Methods for pre-training embeddings, which leverage unlabeled trajectories for training universal embeddings, have shown promising applicability across different tasks, thus attracting considerable interest. However, research progress on this topic faces two key challenges: a lack of a comprehensive overview of existing methods, resulting in several related methods not being well-recognized, and the absence of a unified pipeline, complicating the development of new methods and the analysis of methods. We present UniTE, a survey and a unified pipeline for this domain. In doing so, we present a comprehensive list of existing methods for pre-training trajectory embeddings, which includes methods that either explicitly or implicitly employ pre-training techniques. Further, we present a unified and modular pipeline with publicly available underlying code, simplifying the process of constructing and evaluating methods for pre-training trajectory embeddings. Additionally, we contribute a selection of experimental results using the proposed pipeline on real-world datasets.
Yan Lin 0006, Zeyu Zhou 0001, Yichen Liu 0003, Haochen Lv 0001, Haomin Wen, Tianyi Li 0005, Yushuai Li, Christian S. Jensen, Shengnan Guo 0001, Youfang Lin, Huaiyu Wan
IEEE Trans. Knowl. Data Eng.11
2025 Proactive-XLight: Proactive Traffic Signal Control With Pluggable and Reliable Traffic Prediction
abstract
Traffic signal control (TSC) plays a crucial role in the intelligent transportation system. Among existing TSC approaches, Proactive Traffic Signal Control (PTSC) predicts future traffic states at intersections and proactively adjusts control policies. It is evident that PTSC methods are highly effective in alleviating both current and future traffic congestion at intersections. However, existing PTSC methods focus on point estimation prediction while neglecting prediction reliability. Additionally, they fail to adaptively coordinate between current and future traffic states for optimal control. To address these limitations, we propose an innovative Proactive-Plugin that can be combined with existing TSC methods to enhance the accuracy and robustness of traffic signal control policies. This plugin enhances two critical aspects: 1) Prediction reliability is achieved through Fine-grained Traffic Uncertainty Quantification. This module generates probabilistic forecasts along with confidence intervals to explicitly indicate the credibility of the predictions. 2) Coordination adaptiveness is enabled by a Current-Future Tradeoff Integration mechanism. This mechanism dynamically adjusts the relative influence of current traffic states and probabilistic forecasts on control policies. To further ensure robustness, we design a multi-task joint optimization to reduce the negative impact of inaccurate predictions during training. Experimental results on six real-world datasets demonstrate consistent improvements in traffic efficiency, validating the effectiveness of our approach.
Yang Jiang 0005, Shengnan Guo 0001, Hanyang Chen, Xiaowei Mao, Youfang Lin, Huaiyu Wan
IEEE Trans. Mob. Comput.6
2024 KPatch: Knowledge Patch to Pre-trained Language Model for Zero-Shot Stance Detection on Social Media
abstract
Zero-shot stance detection on social media (ZSSD-SM) aims to distinguish the attitude in tweets towards an unseen target. Previous work capture latent variables between source and target domains to perform this task, but the lack of context knowledge hinders the detection performance. Recent studies have been devoted to obtaining the accurate representation of tweets by bringing additional facts from Knowledge Graph (KG), showing promising performance. However, these knowledge injection methods still suffer from two challenges: (i) The pipeline of knowledge injection causes error accumulation and (ii) irrelevant knowledge makes them fail to understand the semantics. In this paper, we propose a novel knowledge injection method for ZSSD-SM, which adopts two training stages, namely knowledge compression and task guidance, to flexibly inject knowledge into the pre-trained language model (PLM) and adaptively expand tweets context. Specifically, in the knowledge compression stage, the latent representation of KG is reconstructed by the triplet denoising task and compressed into external matrices; while in the task guidance stage, the frozen matrices are employed to guide the PLM to adaptively extract its own context-related knowledge, and then complete the fine-tuning of the ZSSD-SM task. Extensive experiments on multiple datasets show the effectiveness of our proposed method. The code is available at: https://github.com/ShuohaoLin/KPatch.
Shuohao Lin, Wei Chen 0105, Zhishu Jiang, Mengqi Liao, Shuyuan Zhao 0001, Huaiyu Wan
LREC/COLING8
2024 Local-Global History-Aware Contrastive Learning for Temporal Knowledge Graph Reasoning
abstract
Temporal knowledge graphs (TKGs) have been identified as a promising approach to represent the dynamics of facts along the timeline. The extrapolation of TKG is to predict unknowable facts happening in the future, holding significant practical value across diverse fields. Most extrapolation studies in TKGs focus on modeling global historical fact repeating and cyclic patterns, as well as local historical adjacent fact evolution patterns, showing promising performance in predicting future un-known facts. Yet, existing methods still face two major challenges: (1) They usually neglect the importance of historical information in KG snapshots related to the queries when encoding the local and global historical information; (2) They exhibit weak anti-noise capabilities, which hinders their performance when the inputs are contaminated with noise. To this end, we propose a novel Local-global history-aware Contrastive Learning model (LogCL) for TKG reasoning, which adopts contrastive learning to better guide the fusion of local and global historical information and enhance the ability to resist interference. Specifically, for the first challenge, LogCL proposes an entity-aware attention mechanism applied to the local and global historical facts encoder, which captures the key historical information related to queries. For the latter issue, LogCL designs a local-global query contrast module, effectively improving the robustness of the model. The experimental results on four benchmark datasets demonstrate that LogCL delivers better and more robust performance than the state-of-the-art baselines. The code of LogCL is available at https://eithub.com/WeiChen3690/LoeCL.
Wei Chen 0105, Huaiyu Wan, Shuyuan Zhao 0001, Jiayaqi Cheng, Youfang Lin
ICDE2
2024 LaDe: The First Comprehensive Last-mile Express Dataset from Industry
abstract
Real-world last-mile express datasets are crucial for research in logistics, supply chain management, and spatio-temporal data mining. Despite a plethora of algorithms developed to date, no widely accepted, publicly available last-mile express dataset exists to support research in this field. In this paper, we introduce LaDe, the first publicly available last-mile express dataset with millions of packages from the industry. LaDe has three unique characteristics: (1)Large-scale. It involves 10,677k packages of 21k couriers over 6 months of real-world operation. (2)Comprehensive information. It offers original package information, task-event information, as well as couriers' detailed trajecotries and road networks. (3)Diversity. The dataset includes data from various scenarios, including package pick-up and delivery, and from multiple cities, each with its unique spatio-temporal patterns due to their distinct characteristics such as populations. We verify LaDe on three tasks by running several classical baseline models per task. We believe that the large-scale, comprehensive, diverse feature of LaDe can offer unparalleled opportunities to researchers in the supply chain community, data mining community, and beyond. The dataset and code is publicly available at https://huggingface.co/datasets/Cainiao-AI/LaDe.
Lixia Wu, Haomin Wen, Haoyuan Hu, Xiaowei Mao, Yutong Xia, Ergang Shan, Jianbin Zheng 0003, Junhong Lou, Yuxuan Liang 0002, Liuqing Yang 0001, Roger Zimmermann, Youfang Lin, Huaiyu Wan
KDD13
2024 DiffLight: A Partial Rewards Conditioned Diffusion Model for Traffic Signal Control with Missing Data
abstract
The application of reinforcement learning in traffic signal control (TSC) has been extensively researched and yielded notable achievements. However, most existing works for TSC assume that traffic data from all surrounding intersections is fully and continuously available through sensors. In real-world applications, this assumption often fails due to sensor malfunctions or data loss, making TSC with missing data a critical challenge. To meet the needs of practical applications, we introduce DiffLight, a novel conditional diffusion model for TSC under data-missing scenarios in the offline setting. Specifically, we integrate two essential sub-tasks, i.e., traffic data imputation and decision-making, by leveraging a Partial Rewards Conditioned Diffusion (PRCD) model to prevent missing rewards from interfering with the learning process. Meanwhile, to effectively capture the spatial-temporal dependencies among intersections, we design a Spatial-Temporal transFormer (STFormer) architecture. In addition, we propose a Diffusion Communication Mechanism (DCM) to promote better communication and control performance under data-missing scenarios. Extensive experiments on five datasets with various data-missing scenarios demonstrate that DiffLight is an effective controller to address TSC with missing data. The code of DiffLight is released at https://github.com/lokol5579/DiffLight-release.
Hanyang Chen, Yang Jiang 0005, Shengnan Guo 0001, Xiaowei Mao, Youfang Lin, Huaiyu Wan
NeurIPS6
2024 Mobility-LLM: Learning Visiting Intentions and Travel Preference from Human Mobility Data with Large Language Models
abstract
Location-based services (LBS) have accumulated extensive human mobility data on diverse behaviors through check-in sequences. These sequences offer valuable insights into users’ intentions and preferences. Yet, existing models analyzing check-in sequences fail to consider the semantics contained in these sequences, which closely reflect human visiting intentions and travel preferences, leading to an incomplete comprehension. Drawing inspiration from the exceptional semantic understanding and contextual information processing capabilities of large language models (LLMs) across various domains, we present Mobility-LLM, a novel framework that leverages LLMs to analyze check-in sequences for multiple tasks. Since LLMs cannot directly interpret check-ins, we reprogram these sequences to help LLMs comprehensively understand the semantics of human visiting intentions and travel preferences. Specifically, we introduce a visiting intention memory network (VIMN) to capture the visiting intentions at each record, along with a shared pool of human travel preference prompts (HTPP) to guide the LLM in understanding users’ travel preferences. These components enhance the model’s ability to extract and leverage semantic information from human mobility data effectively. Extensive experiments on four benchmark datasets and three downstream tasks demonstrate that our approach significantly outperforms existing models, underscoring the effectiveness of Mobility-LLM in advancing our understanding of human mobility data within LBS contexts.
Letian Gong, Yan Lin 0006, Xuedi Han, Yichen Liu 0003, Shengnan Guo 0001, Youfang Lin, Huaiyu Wan
NeurIPS9
2024 PGN: The RNN's New Successor is Effective for Long-Range Time Series Forecasting
abstract
Due to the recurrent structure of RNN, the long information propagation path poses limitations in capturing long-term dependencies, gradient explosion/vanishing issues, and inefficient sequential execution. Based on this, we propose a novel paradigm called Parallel Gated Network (PGN) as the new successor to RNN. PGN directly captures information from previous time steps through the designed Historical Information Extraction (HIE) layer and leverages gated mechanisms to select and fuse it with the current time step information. This reduces the information propagation path to $\mathcal{O}(1)$, effectively addressing the limitations of RNN. To enhance PGN's performance in long-range time series forecasting tasks, we propose a novel temporal modeling framework called Temporal PGN (TPGN). TPGN incorporates two branches to comprehensively capture the semantic information of time series. One branch utilizes PGN to capture long-term periodic patterns while preserving their local characteristics. The other branch employs patches to capture short-term information and aggregate the global representation of the series. TPGN achieves a theoretical complexity of $\mathcal{O}(\sqrt{L})$, ensuring efficiency in its operations. Experimental results on five benchmark datasets demonstrate the state-of-the-art (SOTA) performance and high efficiency of TPGN, further confirming the effectiveness of PGN as the new successor to RNN in long-range time series forecasting. The code is available in this repository: https://github.com/Water2sea/TPGN.
Yuxin Jia, Youfang Lin, Shuo Wang 0031, Huaiyu Wan
NeurIPS6
2024 PVII: A pedestrian-vehicle interactive and iterative prediction framework for pedestrian's trajectory
Qianwen Shen, Shien Huang, Baixi Sun, Xinyu Chen 0008, Dingwen Tao, Huaiyu Wan, Ergude Bao
Appl. Intell.6
2024 Spatial-temporal uncertainty-aware graph networks for promoting accuracy and reliability of traffic forecastin
Xiyuan Jin, Jing Wang 0060, Shengnan Guo 0001, Tonglong Wei, Yiji Zhao, Youfang Lin, Huaiyu Wan
Expert Syst. Appl.7
2024 Inductive and adaptive graph convolution networks equipped with constraint task for spatial-temporal traffic data kriging
Tonglong Wei, Youfang Lin, Shengnan Guo 0001, Yan Lin 0006, Yiji Zhao, Xiyuan Jin, Zhihao Wu 0001, Huaiyu Wan
Knowl. Based Syst.8
2024 Spatial-Temporal Cross-View Contrastive Pre-Training for Check-in Sequence Representation Learning
abstract
The rapid growth of location-based services (LBS) has yielded massive amounts of data on human mobility. Effectively extracting meaningful representations for user-generated check-in sequences is pivotal for facilitating various downstream services. However, the user-generated check-in data are simultaneously influenced by the surrounding objective circumstances and the user's subjective intention. Specifically, the temporal uncertainty and spatial diversity exhibited in check-in data make it difficult to capture the macroscopic spatial-temporal patterns of users and to understand the semantics of user mobility activities. Furthermore, the distinct characteristics of the temporal and spatial information in check-in sequences call for an effective fusion method to incorporate these two types of information. In this paper, we propose a novel Spatial-Temporal Cross-view Contrastive Representation (STCCR) framework for check-in sequence representation learning. Specifically, STCCR addresses the above challenges by employing self-supervision from “spatial topic” and “temporal intention” views, facilitating effective fusion of spatial and temporal information at the semantic level. Besides, STCCR leverages contrastive clustering to uncover users’ shared spatial topics from diverse mobility activities, while employing angular momentum contrast to mitigate the impact of temporal uncertainty and noise. We extensively evaluate STCCR on three real-world datasets and demonstrate its superior performance across three downstream tasks.
Letian Gong, Huaiyu Wan, Shengnan Guo 0001, Xiucheng Li, Yan Lin 0006, Erwen Zheng, Zeyu Zhou 0001, Youfang Lin
IEEE Trans. Knowl. Data Eng.2
2024 Pre-Training General Trajectory Embeddings With Maximum Multi-View Entropy Coding
abstract
Spatio-temporal trajectories provide valuable information about movement and travel behavior, enabling various downstream tasks that in turn power real-world applications. Learning trajectory embeddings can improve task performance but may incur high computational costs and face limited training data availability. Pre-training learns generic embeddings by means of specially constructed pretext tasks that enable learning from unlabeled data. Existing pre-training methods face (i) difficulties in learning general embeddings due to biases towards certain downstream tasks incurred by the pretext tasks, (ii) limitations in capturing both travel semantics and spatio-temporal correlations, and (iii) the complexity of long, irregularly sampled trajectories. To tackle these challenges, we propose Maximum Multi-view Trajectory Entropy Coding (MMTEC) for learning general and comprehensive trajectory embeddings. We introduce a pretext task that reduces biases in pre-trained trajectory embeddings, yielding embeddings that are useful for a wide variety of downstream tasks. We also propose an attention-based discrete encoder and a NeuralCDE-based continuous encoder that extract and represent travel behavior and continuous spatio-temporal correlations from trajectories in embeddings, respectively. Extensive experiments on two real-world datasets and three downstream tasks offer insight into the design properties of our proposal and indicate that it is capable of outperforming existing trajectory embedding methods.
Yan Lin 0006, Huaiyu Wan, Shengnan Guo 0001, Jilin Hu, Christian S. Jensen, Youfang Lin
IEEE Trans. Knowl. Data Eng.2
2024 Diff-RNTraj: A Structure-Aware Diffusion Model for Road Network-Constrained Trajectory Generation
abstract
Trajectory data is essential for various applications. However, publicly available trajectory datasets remain limited in scale due to privacy concerns, which hinders the development of trajectory mining and applications. Although some trajectory generation methods have been proposed to expand dataset scale, they generate trajectories in the geographical coordinate system, posing two limitations for practical applications: 1) failing to ensure that the generated trajectories are road-constrained. 2) lacking road-related information. In this paper, we propose a new problem, road network-constrained trajectory (RNTraj) generation, which can directly generate trajectories on the road network with road-related information. Specifically, RNTraj is a hybrid type of data, in which each point is represented by a discrete road segment and a continuous moving rate. To generate RNTraj, we design a diffusion model called Diff-RNTraj, which can effectively handle the hybrid RNTraj using a continuous diffusion framework by incorporating a pre-training strategy to embed hybrid RNTraj into continuous representations. During the sampling stage, a RNTraj decoder is designed to map the continuous representation generated by the diffusion model back to the hybrid RNTraj format. Furthermore, Diff-RNTraj introduces a novel loss function to enhance trajectory’s spatial validity. Extensive experiments conducted on two datasets demonstrate the effectiveness of Diff-RNTraj.
Tonglong Wei, Youfang Lin, Shengnan Guo 0001, Yan Lin 0006, Chenyang Xiang, Yuqing Bai, Huaiyu Wan
IEEE Trans. Knowl. Data Eng.8
2024 Micro-Macro Spatial-Temporal Graph-Based Encoder-Decoder for Map-Constrained Trajectory Recovery
abstract
Recovering intermediate missing GPS points in a sparse trajectory, while adhering to the constraints of the road network, could offer deep insights into users' moving behaviors in intelligent transportation systems. Although recent studies have demonstrated the advantages of achieving map-constrained trajectory recovery via an end-to-end manner, they still face two significant challenges. Firstly, existing methods are mostly sequence-based models. It is extremely hard for them to comprehensively capture the micro-semantics of individual trajectory, including the information of each GPS point and the movement between two GPS points. Secondly, existing approaches ignore the impact of the macro-semantics,i.e., the road conditions and the people's shared travel preferences reflected by a group of trajectories. To address the above challenges, we propose a Micro-Macro Spatial-Temporal Graph-based Encoder-Decoder (MM-STGED). Specifically, we model each trajectory as a graph to efficiently describe the micro-semantics of trajectory and design a novel message-passing mechanism to learn trajectory representations. Additionally, we extract the macro-semantics of trajectories and further incorporate them into a well-designed graph-based decoder to guide trajectory recovery. Extensive experiments conducted on sparse trajectories with three different sampling intervals that are respectively constructed from two real-world trajectory datasets demonstrate the superiority of our proposed model.
Tonglong Wei, Youfang Lin, Yan Lin 0006, Shengnan Guo 0001, Huaiyu Wan
IEEE Trans. Knowl. Data Eng.6
2024 A Survey on Service Route and Time Prediction in Instant Delivery: Taxonomy, Progress, and Prospects
abstract
Instant delivery services, such as food delivery and package delivery, have achieved explosive growth in recent years by providing customers with daily-life convenience. An emerging research area within these services is service Route&Time Prediction (RTP), which aims to estimate the future service route as well as the arrival time of a given worker. As one of the most crucial tasks in those service platforms, RTP stands central to enhancing user satisfaction and trimming operational expenditures on these platforms. Despite a plethora of algorithms developed to date, there is no systematic, comprehensive survey to guide researchers in this domain. To fill this gap, our work presents the first comprehensive survey that methodically categorizes recent advances in service route and time prediction. We start by defining the RTP challenge and then delve into the metrics that are often employed. Following that, we scrutinize the existing RTP methodologies, presenting a novel taxonomy of them. We categorize these methods based on three criteria: (i) type of task, subdivided into only-route prediction, only-time prediction, and joint route&time prediction; (ii) model architecture, which encompasses sequence-based and graph-based models; and (iii) learning paradigm, including Supervised Learning (SL) and Deep Reinforcement Learning (DRL). Conclusively, we highlight the limitations of current research and suggest prospective avenues. We believe that the taxonomy, progress, and prospects introduced in this paper can significantly promote the development of this field.
Haomin Wen, Youfang Lin, Lixia Wu, Xiaowei Mao, Tianyue Cai, Yunfeng Hou, Shengnan Guo 0001, Yuxuan Liang 0002, Guangyin Jin, Yiji Zhao, Roger Zimmermann, Jieping Ye, Huaiyu Wan
IEEE Trans. Knowl. Data Eng.13
2023 Contrastive Pre-training with Adversarial Perturbations for Check-In Sequence Representation Learning
abstract
A core step of mining human mobility data is to learn accurate representations for user-generated check-in sequences. The learned representations should be able to fully describe the spatial-temporal mobility patterns of users and the high-level semantics of traveling. However, existing check-in sequence representation learning is usually implicitly achieved by end-to-end models designed for specific downstream tasks, resulting in unsatisfactory generalizable abilities and poor performance. Besides, although the sequence representation learning models that follow the contrastive learning pre-training paradigm have achieved breakthroughs in many fields like NLP, they fail to simultaneously consider the unique spatial-temporal characteristics of check-in sequences and need manual adjustments on the data augmentation strategies. So, directly applying them to check-in sequences cannot yield a meaningful pretext task. To this end, in this paper we propose a contrastive pre-training model with adversarial perturbations for check-in sequence representation learning (CACSR). Firstly, we design a novel spatial-temporal augmentation block for disturbing the spatial-temporal features of check-in sequences in the latent space to relieve the stress of designing manual data augmentation strategies. Secondly, to construct an effective contrastive pretext task, we generate “hard” positive and negative pairs for the check-in sequence by adversarial training. These two designs encourage the model to capture the high-level spatial-temporal patterns and semantics of check-in sequences while ignoring the noisy and unimportant details. We demonstrate the effectiveness and versatility of CACSR on two kinds of downstream tasks using three real-world datasets. The results show that our model outperforms both the state-of-the-art pre-training methods and the end-to-end models.
Letian Gong, Youfang Lin, Shengnan Guo 0001, Yan Lin 0006, Erwen Zheng, Zeyu Zhou 0001, Huaiyu Wan
AAAI8
2023 GMDNet: A Graph-Based Mixture Density Network for Estimating Packages' Multimodal Travel Time Distribution
abstract
In the logistics network, accurately estimating packages' Travel Time Distribution (TTD) given the routes greatly benefits both consumers and platforms. Although recent works perform well in predicting an expected time or a time distribution in a road network, they could not be well applied to estimate TTD in logistics networks. Because TTD prediction in the logistics network requires modeling packages' multimodal TTD (MTTD, i.e., there can be more than one likely output with a given input) while leveraging the complex correlations in the logistics network. To this end, this work opens appealing research opportunities in studying MTTD learning conditioned on graph-structure data by investigating packages' travel time distribution in the logistics network. We propose a Graph-based Mixture Density Network, named GMDNet, which takes the benefits of both graph neural network and mixture density network for estimating MTTD conditioned on graph-structure data (i.e., the logistics network). Furthermore, we adopt the Expectation-Maximization (EM) framework in the training process to guarantee local convergence and thus obtain more stable results than gradient descent. Extensive experiments on two real-world datasets demonstrate the superiority of our proposed model.
Xiaowei Mao, Huaiyu Wan, Haomin Wen, Jianbin Zheng 0003, Yuting Qiang, Shengnan Guo 0001, Lixia Wu, Haoyuan Hu, Youfang Lin
AAAI2
2023 DiffSTG: Probabilistic Spatio-Temporal Graph Forecasting with Denoising Diffusion Models
abstract
Spatio-temporal graph neural networks (STGNN) have emerged as the dominant model for spatio-temporal graph (STG) forecasting. Despite their success, they fail to model intrinsic uncertainties within STG data, which cripples their practicality in downstream tasks for decision-making. To this end, this paper focuses on probabilistic STG forecasting, which is challenging due to the difficulty in modeling uncertainties and complex ST dependencies. In this study, we present the first attempt to generalize the popular de-noising diffusion probabilistic models to STGs, leading to a novel non-autoregressive framework called DiffSTG, along with the first denoising network UGnet for STG in the framework. Our approach combines the spatio-temporal learning capabilities of STGNNs with the uncertainty measurements of diffusion models. Extensive experiments validate that DiffSTG reduces the Continuous Ranked Probability Score (CRPS) by 4%-14%, and Root Mean Squared Error (RMSE) by 2%-7% over existing methods on three real-world datasets.
Haomin Wen, Youfang Lin, Yutong Xia, Huaiyu Wan, Qingsong Wen, Roger Zimmermann, Yuxuan Liang 0002
SIGSPATIAL/GIS4
2023 Self-Supervised Spatial-Temporal Bottleneck Attentive Network for Efficient Long-term Traffic Forecasting
abstract
In intelligent transportation systems, accurate long-term traffic forecasting is informative for administrators and travelers to make wise decisions in advance. Recently proposed spatial-temporal forecasting models perform well for short-term traffic forecasting, but two challenges hinder their applications for long-term forecasting in practice. Firstly, existing traffic forecasting models do not have satisfactory scalability on effectiveness and efficiency, i.e., as the prediction time spans extend, existing models either cannot capture the long-term spatial-temporal dynamics of traffic data or equip global receptive fields at the cost of quadratic computational complexity. Secondly, the dilemma between the models’ strong appetite for high-quality training data and their generalization ability is also a challenge we have to face. Thus how to improve data utilization efficiency deserves thoughtful thinking. Aiming at solving the long-term traffic forecasting problem and facilitating the deployment of traffic forecasting models in practice, this paper proposes an efficient and effective Self-supervised Spatial-Temporal Bottleneck Attentive Network (SSTBAN). Specifically, SSTBAN follows a multi-task framework by incorporating a self-supervised learner to produce robust latent representations for historical traffic data, so as to improve its generalization performance and robustness for forecasting. Besides, we design a spatial-temporal bottleneck attention mechanism, reducing the computational complexity meanwhile encoding global spatial-temporal dynamics. Extensive experiments on real-world long-term traffic forecasting tasks, including traffic speed forecasting and traffic flow forecasting under nine scenarios, demonstrate that SSTBAN not only achieves the overall best performance but also has good computation efficiency and data utilization efficiency.
Shengnan Guo 0001, Youfang Lin, Letian Gong, Zeyu Zhou 0001, Zekai Shen 0001, Huaiyu Wan
ICDE8
2023 M2G4RTP: A Multi-Level and Multi-Task Graph Model for Instant-Logistics Route and Time Joint Prediction
abstract
Instant-logistics (e.g., food delivery and package pick-up) is increasingly calling for Route and Time Prediction (RTP), which aims to predict both future route and arrival time of a courier’s unvisited locations. Accurate RTP can greatly benefit the platform, such as optimizing order dispatching and improving user experience. Although recent years have witnessed various works for solving the RTP problem, they still suffer from the following three limitations: i) Failing to consider the high-level transfer mode of couriers between AOIs (Areas Of Interest, such as residential quarters or office buildings), which can help to build more accurate RTP. ii) Failing to simultaneously make the route and time prediction. Existing works either separately predict route/time or predict them in a two-step way. However, since route and time are strongly correlated (nearby locations in the route should have similar arrival times), jointly predicting them should be more effective. iii) The widely adopted tree-based or sequence-based architecture fails to fully encode the spatial relationship between different locations. To address the above limitations, we propose a multi-level and multi-task graph model, named M2G4RTP, for instant-logistics route and time joint prediction. Specifically, we propose a multi-level graph encoder equipped with a newly-designed GAT-e encoding module to capture couriers’ both high-level transfer modes between AOIs and low-level transfer modes between locations. Moreover, a multi-task decoder is presented to jointly predict the route and time at different levels. Finally, a loss weighting method based on homoscedastic uncertainty is designed to balance the two tasks adaptively. Extensive experiments on an industry-scale real-world dataset, as well as the online deployment on Cainiao Alibaba, demonstrate the superiority of our proposed model.
Tianyue Cai, Huaiyu Wan, Haomin Wen, Shengnan Guo 0001, Lixia Wu, Haoyuan Hu, Youfang Lin
ICDE2
2023 Modeling Intra- and Inter-community Information for Route and Time Prediction in Last-mile Delivery
abstract
Last-mile delivery, which refers to delivering packages from the depot to customers, is a crucial step for logistics service. The Route and Time Prediction (RTP) in last-mile package delivery is beneficial to improve customers’ experience and supervise couriers’ behavior. However, the limited raw information brings great challenges to accurately predict the route and delivery time. In this paper, we propose a deep model named I2RTP, which explores the heterogeneous representation of the package’s community to help predict the delivery route and estimate the arrival time of each package. Specifically, for the entire delivery route prediction, we model the inter- and intra-community information to learn the route features from global and local perspectives. Besides, by integrating the community representation with package features, our model could make more accurate predictions of the next-delivery package and its time duration. Experiments on the offline dataset and the online deployment on Cainiao’s Delivery System demonstrate the effectiveness of our proposed method, as well as validate the rationality of the global and local prediction pipeline.
Yuting Qiang, Haomin Wen, Lixia Wu, Xiaowei Mao, Huaiyu Wan, Haoyuan Hu
ICDE6
2023 DRL4Route: A Deep Reinforcement Learning Framework for Pick-up and Delivery Route Prediction
abstract
Pick-up and Delivery Route Prediction (PDRP), which aims to estimate the future service route of a worker given his current task pool, has received rising attention in recent years. Deep neural networks based on supervised learning have emerged as the dominant model for the task because of their powerful ability to capture workers' behavior patterns from massive historical data. Though promising, they fail to introduce the non-differentiable test criteria into the training process, leading to a mismatch in training and test criteria. Which considerably trims down their performance when applied in practical systems. To tackle the above issue, we present the first attempt to generalize Reinforcement Learning (RL) to the route prediction task, leading to a novel RL-based framework called DRL4Route. It combines the behavior-learning abilities of previous deep learning models with the non-differentiable objective optimization ability of reinforcement learning. DRL4Route can serve as a plug-and-play component to boost the existing deep learning models. Based on the framework, we further implement a model named DRL4Route-GAE for PDRP in logistic service. It follows the actor-critic architecture which is equipped with a Generalized Advantage Estimator that can balance the bias and variance of the policy gradient estimates, thus achieving a more optimal policy. Extensive offline experiments and the online deployment show that DRL4Route-GAE improves Location Square Deviation (LSD) by 0.9%-2.7%, and Accuracy@3 (ACC@3) by 2.4%-3.2% over existing methods on the real-world dataset.
Xiaowei Mao, Haomin Wen, Huaiyu Wan, Lixia Wu, Jianbin Zheng 0003, Haoyuan Hu, Youfang Lin
KDD4
2023 WITRAN: Water-wave Information Transmission and Recurrent Acceleration Network for Long-range Time Series Forecasting
abstract
Capturing semantic information is crucial for accurate long-range time series forecasting, which involves modeling global and local correlations, as well as discovering long- and short-term repetitive patterns. Previous works have partially addressed these issues separately, but have not been able to address all of them simultaneously. Meanwhile, their time and memory complexities are still not sufficiently low for long-range forecasting. To address the challenge of capturing different types of semantic information, we propose a novel Water-wave Information Transmission (WIT) framework. This framework captures both long- and short-term repetitive patterns through bi-granular information transmission. It also models global and local correlations by recursively fusing and selecting information using Horizontal Vertical Gated Selective Unit (HVGSU). In addition, to improve the computing efficiency, we propose a generic Recurrent Acceleration Network (RAN) which reduces the time complexity to $\mathcal{O}(\sqrt{L})$ while maintaining the memory complexity at $\mathcal{O}(L)$. Our proposed method, called Water-wave Information Transmission and Recurrent Acceleration Network (WITRAN), outperforms the state-of-the-art methods by 5.80% and 14.28% on long-range and ultra-long-range time series forecasting tasks respectively, as demonstrated by experiments on four benchmark datasets. The code is available at: https://github.com/Water2sea/WITRAN.
Yuxin Jia, Youfang Lin, Xinyan Hao, Yan Lin 0006, Shengnan Guo 0001, Huaiyu Wan
NeurIPS6
2023 Generative-Contrastive-Attentive Spatial-Temporal Network for Traffic Data Imputation
Wenchuang Peng, Youfang Lin, Shengnan Guo 0001, Weiwen Tang, Huaiyu Wan
PAKDD (4)6
2023 Multi-scale adaptive attention-based time-variant neural networks for multi-step time series forecasting
Changxia Gao, Youru Li, Yan Lin 0006, Huaiyu Wan
Appl. Intell.5
2023 Adversarial self-attentive time-variant neural networks for multi-step time series forecasting
abstract
Accurate forecasting of time series mitigates the uncertainty of future outlooks and is a great help in reducing errors in decisions. Despite years of researches, there are still some challenges to accurate forecasting of time series, including the difficulty of dynamic modeling, the problem of capturing short-term correlations, and the conundrum of long-term forecasting. This paper offers an A dversarial Truncated Cauchy S elf- A ttentive T ime- V ariant Neural N etwork (ASATVN) for multi-step ahead time series forecasting. Specifically, the proposed model builds on Generative Adversarial Networks, in which the generator is composed of a novel time-variant model. The time-variant model contributes to learning dynamic time-series changes with its time-variant architecture and employs a newly proposed Truncated Cauchy Self-Attention block to capture the local sequential dependencies better. For the discriminator , two self-attentive discriminators are presented to regularize predictions with fidelity and continuity, which is beneficial to predicting sequence over longer time horizons. Our proposed ASATVN model outperforms the state-of-the-art predictive models on eleven real-world benchmark datasets, demonstrating its effectiveness.
Changxia Gao, Youru Li, Yan Lin 0006, Huaiyu Wan
Expert Syst. Appl.5
2023 Exploiting global context and external knowledge for distantly supervised relation extraction
Jianwei Gao, Huaiyu Wan, Youfang Lin
Knowl. Based Syst.2
2023 Origin-Destination Travel Time Oracle for Map-based Services
abstract
Given an origin (O), a destination (D), and a departure time (T), an Origin-Destination (OD) travel time oracle~(ODT-Oracle) returns an estimate of the time it takes to travel from O to D when departing at T. ODT-Oracles serve important purposes in map-based services. To enable the construction of such oracles, we provide a travel-time estimation (TTE) solution that leverages historical trajectories to estimate time-varying travel times for OD pairs. The problem is complicated by the fact that multiple historical trajectories with different travel times may connect an OD pair, while trajectories may vary from one another. To solve the problem, it is crucial to remove outlier trajectories when doing travel time estimation for future queries. We propose a novel, two-stage framework called Diffusion-based Origin-destination Travel Time Estimation (DOT), that solves the problem. First, DOT employs a conditioned Pixelated Trajectories (PiT) denoiser that enables building a diffusion-based PiT inference process by learning correlations between OD pairs and historical trajectories. Specifically, given an OD pair and a departure time, we aim to infer a PiT. Next, DOT encompasses a Masked Vision Transformer~(MViT) that effectively and efficiently estimates a travel time based on the inferred PiT. We report on extensive experiments on two real-world datasets that offer evidence that DOT is capable of outperforming baseline methods in terms of accuracy, scalability, and explainability.
Yan Lin 0006, Huaiyu Wan, Jilin Hu, Shengnan Guo 0001, Bin Yang 0002, Youfang Lin, Christian S. Jensen
Proc. ACM Manag. Data2
2023 Enough Waiting for the Couriers: Learning to Estimate Package Pick-up Arrival Time from Couriers' Spatial-Temporal Behaviors
abstract
In intelligent logistics systems, predicting the Estimated Time of Pick-up Arrival (ETPA) of packages is a crucial task, which aims to predict the courier’s arrival time to all the unpicked-up packages at any time. Accurate prediction of ETPA can help systems alleviate customers’ waiting anxiety and improve their experience. We identify three main challenges of this problem. First, unlike the travel time estimation problem in other fields like ride-hailing, the ETPA task is distinctively a multi-destination and path-free prediction problem. Second, an intuitive idea for solving ETPA is to predict the pick-up route and then the time in two stages. However, it is difficult to accurately and efficiently predict couriers’ future routes in the route prediction step since their behaviors are affected by multiple complex factors. Third, furthermore, in the time prediction step, the requirement for providing a courier’s all unpicked-up packages’ ETPA at once in real time makes the problem even more challenging. To tackle the preceding challenges, we propose RankETPA, which integrates the route inference into the ETPA prediction. First, a learning-based pick-up route predictor is designed to learn the route-ranking strategies of couriers from their massive spatial-temporal behaviors. Then, a spatial-temporal attention-based arrival time predictor is designed for real-time ETPA inference via capturing the spatial-temporal correlations between the unpicked-up packages. Extensive experiments on two real-world datasets and a synthetic dataset demonstrate that RankETPA achieves significant performance improvement against the baseline models.
Haomin Wen, Youfang Lin, Huaiyu Wan, Zhongxiang Sun, Tianyue Cai, Hongyu Liu 0003, Shengnan Guo 0001, Jianbin Zheng 0003, Lixia Wu
ACM Trans. Intell. Syst. Technol.4
2023 Modeling Spatial-Temporal Constraints and Spatial-Transfer Patterns for Couriers' Package Pick-up Route Prediction
abstract
Couriers’ package pick-up route prediction is a fundamental task in the emerging intelligent logistics systems. It is beneficial for order dispatching and arrival-time estimation by leveraging the predicted routes to improve those downstream tasks. However, the package pick-up route prediction problem is challenging since couriers’ behaviors are affected by both strict Spatial–Temporal Constraints (STC) and personalized spatial-transfer patterns (STP). Specifically, couriers have to consider explicitly spatial-temporal requirements such as the locations of packages and the promised pick-up time when selecting future routes. In addition, couriers have personalized mobility patterns between different locations, which are implicit patterns hidden behind couriers’ historical trajectories and cannot be ignored to precisely depict their behaviors. This paper proposes a novel framework, named CP-Route, to predict a specific courier’s future package pick-up route under strict spatial-temporal constraints, and enhance the prediction performance by learning the spatial-transfer patterns (i.e., the routing patterns) of couriers. A sophisticated encoder is designed to capture the STC and STP, and a mixed-distribution-based decoder is designed to simultaneously consider the influence of spatial-temporal constraints and routing patterns on couriers’ final decisions. Extensive experiments conducted on an industry-scale logistics dataset demonstrate the superiority of our proposed framework against the existing baseline methods. The online A/B test shows our contribution to the improvement of arrival time prediction.
Haomin Wen, Youfang Lin, Mingxuan Xia, Lixia Wu, Haoyuan Hu, Huaiyu Wan
IEEE Trans. Intell. Transp. Syst.9
2023 Spatial-Temporal Position-Aware Graph Convolution Networks for Traffic Flow Forecasting
abstract
Recent works demonstrate that capturing correlations between road network nodes is crucial to improving traffic flow forecasting accuracy. In general, there are spatial, temporal, and joint spatial-temporal correlations between two nodes, whose strength is related to spatial and temporal position factors. For example, traffic congestion that occurs at a traffic hub has a wider and stronger impact than that at a branch road. Moreover, the above impacts can vary with temporal position. Although spatial-temporal graph convolution networks have become a popular paradigm for modeling those correlations, there are still three problems with existing models: (i) failing to effectively model joint spatial-temporal correlations; (ii) ignoring spatial and temporal position factors when modeling the aforementioned correlations; and (iii) failing to capture distinct spatial-temporal patterns of each node. To cope with the above issues, this paper proposes a novelSpatial-TemporalPosition-awareGraphConvolutionNetwork (STPGCN) for traffic flow forecasting. Specifically, a trainable embedding module is constructed to represent the spatial and temporal positions of the nodes. Subsequently, a spatial-temporal position-aware relation inference module is proposed to adaptively infer the correlation weights of the three important spatial-temporal relations. Based on this, the generated spatial-temporal relations are integrated into a graph convolution layer for aggregating and updating node features. Finally, we design a spatial-temporal position-aware gated activation unit in the graph convolution, to capture the node-specific pattern features under the guidance of position embedding. Extensive experiments on six real-world datasets demonstrate the superiority of our model in terms of prediction performance and computational efficiency.
Yiji Zhao, Youfang Lin, Haomin Wen, Tonglong Wei, Xiyuan Jin, Huaiyu Wan
IEEE Trans. Intell. Transp. Syst.6
2022 Graph2Route: A Dynamic Spatial-Temporal Graph Neural Network for Pick-up and Delivery Route Prediction
abstract
Pick-up and delivery (P&D) services such as food delivery have achieved explosive growth in recent years by providing customers with daily-life convenience. Though many service providers have invested considerably in routing tools, more and more practitioners realize that significant deviations exist between workers' actual routes and planned ones. So it is not wise to feed "optimal routes" as workers' actual service routes into downstream tasks (e.g., arrival-time prediction and order dispatching), whose performances count on the accuracy of route prediction, i.e., to predict the future service route of a worker's unfinished tasks. Therefore, to meet the rising calling for route prediction models that can capture workers' future routing behaviors, in this paper, we formulate the Pick-up and Delivery Route Prediction task (PDRP task for short) from the graph perspective for the first time, then propose a dynamic spatial-temporal graph-based model, named Graph2Route. Unlike previous sequence-based models, our model leverages the underlying graph structure and features into the encoding and decoding process. Moreover, the dynamic graph-based nature can spontaneously describe the evolving relationship between different problem instances. As a result, abundant decision context information and various spatial-temporal information of node/edge can be fully utilized in Graph2Route to improve the prediction performance. Offline experiments over two real-world industry-scale datasets under different P&D services (i.e., food delivery and package pick-up) and online A/B test demonstrate the superiority of our proposed model.
Haomin Wen, Youfang Lin, Xiaowei Mao, Yiji Zhao, Jianbin Zheng 0003, Lixia Wu, Haoyuan Hu, Huaiyu Wan
KDD10
2022 Building and exploiting spatial-temporal knowledge graph for next POI recommendation
Wei Chen 0105, Huaiyu Wan, Shengnan Guo 0001, Shaojie Zheng, Jiamu Li, Shuohao Lin, Youfang Lin
Knowl. Based Syst.2
2022 Self-attention-based time-variant neural networks for multi-step time series forecasting
Changxia Gao, Youru Li, Feng Bian, Huaiyu Wan
Neural Comput. Appl.5
2022 DeepRoute+: Modeling Couriers' Spatial-temporal Behaviors and Decision Preferences for Package Pick-up Route Prediction
abstract
Over 10 billion packages are picked up every day in China. A fundamental task raised in the emerging intelligent logistics systems is the couriers’ package pick-up route prediction, which is beneficial for package dispatching, arrival-time estimation and overdue-risk evaluation, by leveraging the predicted routes to improve those downstream tasks. In the package pick-up scene, the decision-making of a courier is affected by strict spatial-temporal constraints (e.g., package location, promised pick-up time, current time, and courier’s current location). Furthermore, couriers have different decision preferences on various factors (e.g., time factor, distance factor, and balance of both), based on their own perception of the environments and work experience. In this article, we propose a novel model, named DeepRoute+, to predict couriers’ future package pick-up routes according to the couriers’ decision experience and preference learned from the historical behaviors. Specifically, DeepRoute+ consists of three layers: (1) The representation layer produces experience- and preference-aware representations for the unpicked-up packages, in which a decision preference module can dynamically adjust the importance of factors that affects the courier’s decision under the current situation. (2) The transformer encoder layer encodes the representations of packages while considering the spatial-temporal correlations among them. (3) The attention-based decoder layer uses the attention mechanism to generate the whole pick-up route recurrently. Experiments on a real-world logistics dataset demonstrate the state-of-the-art performance of our model.
Haomin Wen, Youfang Lin, Huaiyu Wan, Shengnan Guo 0001, Lixia Wu
ACM Trans. Intell. Syst. Technol.3
2022 Traffic Inflow and Outflow Forecasting by Modeling Intra- and Inter-Relationship Between Flows
abstract
Forecasting traffic inflows and outflows is crucial for intelligent transportation applications such as traffic management and risk assessment. Recently, deep learning models, which focus on capturing spatio-temporal correlations between stations (locations) by constructing Spatio-Temporal Feature Learners (STFL), have achieved promising performance in traffic inflows and outflows prediction. However, two unresolved issues limit the performance of these models. i) dynamic and heterogeneous intra- and inter-relationships between flows are ignored, and ii) the STFL in these models cannot capture the global information. To address the above issues, we propose a novel deep Spatio-Temporal Network framework based on Multi-Relational learning (MR-STN) for predicting traffic inflows and outflows. Specifically, a multi-relational learning module is designed to comprehensively model three kinds of relationships between flows while extracting diverse spatio-temporal features. In this module, an enhanced STFL is developed to capture both local and global information. Then, a feature fusion module is introduced to extract fused features for inflows and outflows respectively via a gated fusion mechanism. On this basis, the prediction module uses fusion features to generate future inflows and outflows. Finally, we implement the proposed framework with four state-of-the-art graph-based deep spatio-temporal models to demonstrate its generality and superiority. Extensive experiments on three datasets show that the proposed framework can significantly boost the performance of existing models.
Yiji Zhao, Youfang Lin, Yongkai Zhang, Haomin Wen, Yunxiao Liu, Hao Wu 0010, Zhihao Wu 0001, Shuaichao Zhang, Huaiyu Wan
IEEE Trans. Intell. Transp. Syst.9
2022 Learning Dynamics and Heterogeneity of Spatial-Temporal Graph Data for Traffic Forecasting
abstract
Accurate traffic forecasting is critical in improving safety, stability, and efficiency of intelligent transportation systems. Despite years of studies, accurate traffic prediction still faces the following challenges, including modeling the dynamics of traffic data along both temporal and spatial dimensions, and capturing the periodicity and the spatial heterogeneity of traffic data, and the problem is more difficult for long-term forecast. In this paper, we propose an Attention based Spatial-Temporal Graph Neural Network (ASTGNN) for traffic forecasting. Specifically, in the temporal dimension, we design a novel self-attention mechanism that is capable of utilizing the local context, which is specialized for numerical sequence representation transformation. It enables our prediction model to capture the temporal dynamics of traffic data and to enjoy global receptive fields that is beneficial for long-term forecast. In the spatial dimension, we develop a dynamic graph convolution module, employing self-attention to capture the spatial correlations in a dynamic manner. Furthermore, we explicitly model the periodicity and capture the spatial heterogeneity through embedding modules. Experiments on five real-world traffic flow datasets demonstrate that ASTGNN outperforms the state-of-the-art baselines.
Shengnan Guo 0001, Youfang Lin, Huaiyu Wan, Xiucheng Li, Gao Cong
IEEE Trans. Knowl. Data Eng.3
2022 Pre-Training Time-Aware Location Embeddings from Spatial-Temporal Trajectories
abstract
With the increasing accumulation of spatial-temporal trajectory data, location-based data mining has recently been extensively studied. A fundamental research topic in this field is learning the embedding vectors of locations through self-supervised pre-training. Pre-trained embedding vectors can utilize the highly available unlabeled trajectory data, and benefit downstream tasks in multiple aspects. However, most existing methods ignore the temporal information hidden in the visited time of locations in trajectories. Considering that human activities are highly regulated by specific periods of a day, temporal information can reflect some intrinsic characteristics of locations, so it is necessary to fuse them into location embedding vectors. In this paper, we propose a Time-Aware Location Embedding (TALE) pre-training method based on the CBOW framework, which is able to incorporate temporal information into the learned embedding vectors of locations. A novel temporal tree structure is designed to extract temporal information during the calculation of Hierarchical Softmax. In order to verify the effectiveness of TALE, we apply the learned embedding vectors into three downstream location-based prediction tasks, i.e., location classification, location visitor flow prediction and user next location prediction. Experiments are conducted on four real-world user trajectory datasets, and the experimental results demonstrate that our TALE model can obviously help downstream tasks gain better performance.
Huaiyu Wan, Yan Lin 0006, Shengnan Guo 0001, Youfang Lin
IEEE Trans. Knowl. Data Eng.1
2021 Pre-training Context and Time Aware Location Embeddings from Spatial-Temporal Trajectories for User Next Location Prediction
abstract
Pre-training location embeddings from spatial-temporal trajectories is a fundamental procedure and very beneficial for user next location prediction. In the real world, a location usually has variable functionalities under different contextual environments. If the exact functions of a location in the trajectory can be reflected in its embedding, the accuracy of user next location prediction should be improved. Yet, existing location embeddings pre-trained on trajectories are mostly based on distributed word representations, which mix a location's various functionalities into one latent representation vector. To address this problem, we propose a Context and Time aware Location Embedding (CTLE) model, which calculates a location's representation vector with consideration of its specific contextual neighbors in trajectories. In this way, the multi-functional properties of locations can be properly tackled. Furthermore, in order to incorporate temporal information in trajectories into location embeddings, we propose a subtle temporal encoding module and a novel pre-training objective, which further improve the quality of location embeddings. We evaluate our proposed model on two real-world mobile user trajectory datasets. The experimental results demonstrate that, compared with the existing embedding methods, our CTLE model can pre-train higher quality location embeddings and significantly improve the performance of downstream user location prediction models.
Yan Lin 0006, Huaiyu Wan, Shengnan Guo 0001, Youfang Lin
AAAI2
2021 GSNet: Learning Spatial-Temporal Correlations from Geographical and Semantic Aspects for Traffic Accident Risk Forecasting
abstract
Traffic accident forecasting is of great importance to urban public safety, emergency treatment, and construction planning. However, it is very challenging since traffic accidents are affected by multiple factors, and have multi-scale dependencies on both spatial and temporal dimensional features. Meanwhile, traffic accidents are rare events, which leads to the zero-inflated issue. Existing traffic accident forecasting methods cannot deal with all above problems simultaneously. In this paper, we propose a novel model, named GSNet, to learn the spatial-temporal correlations from geographical and semantic aspects for traffic accident risk forecasting. In the model, a Spatial-Temporal Geographical Module is designed to capture the geographical spatial-temporal correlations among regions, while a Spatial-Temporal Semantic Module is proposed to model the semantic spatial-temporal correlations among regions. In addition, a weighted loss function is designed to solve the zero-inflated issue. Extensive experiments on two real-world datasets demonstrate the superiority of GSNet against the state-of-the-art baseline methods.
Youfang Lin, Shengnan Guo 0001, Huaiyu Wan
AAAI4
2021 Estimated Time of Arrival Prediction via Modeling the Spatial-Temporal Interactions between Links and Crosses
abstract
The ACM SIGSPATIAL GIS CUP 2021 focuses on Estimated Time of Arrival (ETA) prediction, which is important to the travel scheduling and decision-making of ride-hailing platforms. Accurate ETA prediction is very challenging since ETA is affected by many heterogeneous influencing factors, including static features (e.g., number of links) and dynamic features (e.g., real-time road conditions). Meanwhile, ETA can also be affected by complex spatial-temporal dependencies between links and crosses in the route. To tackle the above challenges, we propose a deep learning method based on the Wide-Deep-Recurrent (WDR) architecture while modeling the interactions between links and crosses. We adopt Neural Factorization Machines (NFM) to memorize the historical patterns and a multiple layer perceptron (MLP) to integrate various heterogeneous influencing factors. We also model links and crosses jointly to learn their spatial-temporal dependencies in the route. Extensive experiments conducted on a real dataset show that our method achieves a high prediction accuracy. The source code is available at: https://github.com/wanhuaiyu/WDR-LC.
Xiaowei Mao, Tianyue Cai, Wenchuang Peng, Huaiyu Wan
SIGSPATIAL/GIS4
2021 Package Pick-up Route Prediction via Modeling Couriers' Spatial-Temporal Behaviors
abstract
Over 10 billion packages are picked up every day in China. Accurate prediction of couriers' pick-up routes can help the dispatch system to assign packages to couriers more intelligently, which is able to further increase the pick-up efficiency and reduce the overdue rate. In the package pick-up scene, the decision-making of a courier is quite complex since it's affected by strict spatial-temporal constraints (e.g., package location, promised pick-up time, current time and courier's current location). In this paper, we propose a novel model, named DeepRoute, to predict couriers' future package pick-up routes according to the couriers' decision experience learnt from their historical spatial-temporal behaviors. Specifically, DeepRoute consists of three layers: 1) The representation layer produces experience-aware representations for unpicked-up packages. 2) The transformer encoder layer encodes the representations of packages while considering the spatial-temporal correlations among them. 3) The attention-based decoder layer uses the attention mechanism to generate the whole pick-up route recurrently. Experiments on a real-world logistics dataset demonstrate the state-of-the-art performance of our DeepRoute model.
Haomin Wen, Youfang Lin, Huaiyu Wan, Shengnan Guo 0001, Lixia Wu
ICDE4
2021 HScodeNet: Combining Hierarchical Sequential and Global Spatial Information of Text for Commodity HS Code Classification
Shaohua Du, Zhihao Wu 0001, Huaiyu Wan, Youfang Lin
PAKDD (2)3
2020 Spatial-Temporal Synchronous Graph Convolutional Networks: A New Framework for Spatial-Temporal Network Data Forecasting
abstract
Spatial-temporal network data forecasting is of great importance in a huge amount of applications for traffic management and urban planning. However, the underlying complex spatial-temporal correlations and heterogeneities make this problem challenging. Existing methods usually use separate components to capture spatial and temporal correlations and ignore the heterogeneities in spatial-temporal data. In this paper, we propose a novel model, named Spatial-Temporal Synchronous Graph Convolutional Networks (STSGCN), for spatial-temporal network data forecasting. The model is able to effectively capture the complex localized spatial-temporal correlations through an elaborately designed spatial-temporal synchronous modeling mechanism. Meanwhile, multiple modules for different time periods are designed in the model to effectively capture the heterogeneities in localized spatial-temporal graphs. Extensive experiments are conducted on four real-world datasets, which demonstrates that our method achieves the state-of-the-art performance and consistently outperforms other baselines.
Youfang Lin, Shengnan Guo 0001, Huaiyu Wan
AAAI4
2020 CTS-LSTM: LSTM-based neural networks for correlatedtime series prediction
Huaiyu Wan, Shengnan Guo 0001, Xiaohui Liang 0001, Youfang Lin
Knowl. Based Syst.1
2019 Attention Based Spatial-Temporal Graph Convolutional Networks for Traffic Flow Forecasting
abstract
Forecasting the traffic flows is a critical issue for researchers and practitioners in the field of transportation. However, it is very challenging since the traffic flows usually show high nonlinearities and complex patterns. Most existing traffic flow prediction methods, lacking abilities of modeling the dynamic spatial-temporal correlations of traffic data, thus cannot yield satisfactory prediction results. In this paper, we propose a novel attention based spatial-temporal graph convolutional network (ASTGCN) model to solve traffic flow forecasting problem. ASTGCN mainly consists of three independent components to respectively model three temporal properties of traffic flows, i.e., recent, daily-periodic and weekly-periodic dependencies. More specifically, each component contains two major parts: 1) the spatial-temporal attention mechanism to effectively capture the dynamic spatialtemporal correlations in traffic data; 2) the spatial-temporal convolution which simultaneously employs graph convolutions to capture the spatial patterns and common standard convolutions to describe the temporal features. The output of the three components are weighted fused to generate the final prediction results. Experiments on two real-world datasets from the Caltrans Performance Measurement System (PeMS) demonstrate that the proposed ASTGCN model outperforms the state-of-the-art baselines.
Shengnan Guo 0001, Youfang Lin, Ning Feng, Huaiyu Wan
AAAI5
2019 Improving Relation Classification by Entity Pair Graph
abstract
Relation classification is one of the most important tasks in the field of information extraction, and also a key component of systems that require relational understanding of unstructured text. Existing relation classification approaches mainly rely on exploiting external resources and background knowledge to improve the performance and ignore the correlations between entity pairs which are helpful for relation classification. We present the concept of entity pair graph to represent the correlations between entity pairs and propose a novel entity pair graph based neural network (EPGNN) model, relying on graph convolutional network to capture the topological features of an entity pair graph. EPGNN combines sentence semantic features generated by pre-trained BERT model with graph topological features for relation classification. Our proposed model makes full use of a given corpus and forgoes the need of external resources and background knowledge. The experimental results on two widely used dataset: SemEval 2010 Task 8 and ACE 2005, show that our method outperforms the state-of-the-art approaches.
Huaiyu Wan, Jianwei Gao, Youfang Lin
ACML2
2019 Exploiting the Entity Type Sequence to Benefit Event Detection
abstract
Event Detection (ED) is one of the most important task in the field of information extraction. The goal of ED is to find triggers in sentences and classify them into different event types. In previous works, the information of entity types are commonly utilized to benefit event detection. However, the sequential features of entity types have not been well utilized yet in the existing ED methods. In this paper, we propose a novel ED approach which learns sequential features from word sequences and entity type sequences separately, and combines these two types of sequential features with the help of a trigger-entity interaction learning module. The experimental results demonstrate that our proposed approach outperforms the state-of-the-art methods.
Yuze Ji, Youfang Lin, Jianwei Gao, Huaiyu Wan
CoNLL4
2019 High-Value Prioritized Experience Replay for Off-Policy Reinforcement Learning
abstract
In deep reinforcement learning, experience replay has been shown an effective solution to handle sample-inefficiency. Prioritized Experience Replay (PER) uses temporal-difference error (TD error) as replay priority in Deep Q-Networks (DQN), so that agent can learn more effectively from important experiences. However, experiences with large TD error may appear near the edge of state space and these experiences do not help agent learn policy quickly. We present a novel technique called High-Value Prioritized Experience Replay (HVPER), which designs a combination of TD error and value (reward or state-action value) in replay priority. Specifically, we first propose prioritizing replay based on reward and TD error in sparse reward environment. Extendedly, we design prioritizing replay based on state-action value and TD error for more ordinary environment. We design experiments in the gym environment to evaluate the proposed HVPER. First, we verify that the combination of TD error and reward improves the training speed in two problems with sparse rewards compared to DQN algorithm and PER algorithm. In addition, HVPER accelerates the network learning and achieves a better performance in two continuous space problems compared to Deep Deterministic Policy Gradient algorithm.
Huaiyu Wan, Youfang Lin, Sheng Han 0001
ICTAI2
2019 Deep Spatial-Temporal 3D Convolutional Neural Networks for Traffic Data Forecasting
abstract
Reliable traffic prediction is critical to improve safety, stability, and efficiency of intelligent transportation systems. However, traffic prediction is a very challenging problem because traffic data are a typical type of spatio-temporal data, which simultaneously shows correlation and heterogeneity both in space and time. Most existing works can only capture the partial properties of traffic data and even assume that the effect of correlation on traffic prediction is globally invariable, resulting in inadequate modeling and unsatisfactory prediction performance. In this paper, we propose a novel end-to-end deep learning model, called ST-3DNet, for traffic raster data prediction. ST-3DNet introduces 3D convolutions to automatically capture the correlations of traffic data in both spatial and temporal dimensions. A novel recalibration (Rc) block is proposed to explicitly quantify the difference of the contributions of the correlations in space. Considering two kinds of temporal properties of traffic data, i.e., local patterns and long-term patterns, ST-3DNet employs two components consisting of 3D convolutions and Rc blocks to, respectively, model the two kinds of patterns and then aggregates them together in a weighted way for the final prediction. The experiments on several real-world traffic datasets, viz., traffic congestion data and crowd flows data, demonstrate that our ST-3DNet outperforms the state-of-the-art baselines.
Shengnan Guo 0001, Youfang Lin, Huaiyu Wan
IEEE Trans. Intell. Transp. Syst.5
2016 Extracting relations from traditional Chinese medicine literature via heterogeneous entity networks
abstract
OBJECTIVE: Traditional Chinese medicine (TCM) is a unique and complex medical system that has developed over thousands of years. This article studies the problem of automatically extracting meaningful relations of entities from TCM literature, for the purposes of assisting clinical treatment or poly-pharmacology research and promoting the understanding of TCM in Western countries. METHODS: Instead of separately extracting each relation from a single sentence or document, we propose to collectively and globally extract multiple types of relations (eg, herb-syndrome, herb-disease, formula-syndrome, formula-disease, and syndrome-disease relations) from the entire corpus of TCM literature, from the perspective of network mining. In our analysis, we first constructed heterogeneous entity networks from the TCM literature, in which each edge is a candidate relation, then used a heterogeneous factor graph model (HFGM) to simultaneously infer the existence of all the edges. We also employed a semi-supervised learning algorithm estimate the model's parameters. RESULTS: We performed our method to extract relations from a large dataset consisting of more than 100,000 TCM article abstracts. Our results show that the performance of the HFGM at extracting all types of relations from TCM literature was significantly better than a traditional support vector machine (SVM) classifier (increasing the average precision by 11.09%, the recall by 13.83%, and the F1-measure by 12.47% for different types of relations, compared with a traditional SVM classifier). CONCLUSION: This study exploits the power of collective inference and proposes an HFGM based on heterogeneous entity networks, which significantly improved our ability to extract relations from TCM literature.
Huaiyu Wan, Marie-Francine Moens, Walter Luyten, Xuezhong Zhou, Qiaozhu Mei, Lu Liu 0005, Jie Tang 0001
J. Am. Medical Informatics Assoc.1
2015 Discovering Family Groups in Passenger Social Networks
Huaiyu Wan, Youfang Lin, Xuguang Jia, Yuan-Wei Zhou
J. Comput. Sci. Technol.1
2015 Inferring the Travel Purposes of Passenger Groups for Better Understanding of Passengers
abstract
People usually travel to the same destinations and for the same purposes together with other people in groups. Inferring the travel purposes of passenger groups is a very interesting research problem in the field of passenger transport, because it can help us to better understand passengers and should bring about meaningful changes for personalized travel service and decision making of passenger carriers, organizations, and even governments. In this paper, we attempt to solve this problem by utilizing the historical travel records of passengers. To overcome the constraint of the independent and identical distribution assumption of traditional classifiers, we propose a novel iterative classification approach based on the idea of collective inference. First, we construct cotravel networks by extracting social relations between passengers from their historical travel records that are available in carriers' passenger information systems. Then, we generate a series of sophisticated features for each passenger group in the context of cotravel networks to capture the link structure information between passengers and use the overlapping relations between passenger groups to capture the probabilistic dependence relations between their labels. Finally, we collectively infer the labels of all the groups in an iterative way. Experimental results on a real data set of passenger travel records in the field of civil aviation demonstrate that our proposed iterative classification approach can efficiently infer the travel purposes of passenger groups.
Youfang Lin, Huaiyu Wan, Zhihao Wu 0001, Xuguang Jia
IEEE Trans. Intell. Transp. Syst.2
2012 Discovering Typed Communities in Mobile Social Networks
Huaiyu Wan, Youfang Lin, Zhihao Wu 0001, Houkuan Huang
J. Comput. Sci. Technol.1
2012 Balanced Multi-Label Propagation for Overlapping Community Detection in Social Networks
Zhihao Wu 0001, Youfang Lin, Steve Gregory, Huaiyu Wan, Shengfeng Tian
J. Comput. Sci. Technol.4
2011 A Community-Based Pseudolikelihood Approach for Relationship Labeling in Social Networks
Huaiyu Wan, Youfang Lin, Zhihao Wu 0001, Houkuan Huang
ECML/PKDD (3)1