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

Pengkun Wang 0001

dblp:292/0809-1 · DBLP profile ↗
← Back
49ranked-venue papers
4as first author
49since 2021 · last 2026
0000-0002-2680-4563ORCID · verified

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

Artificial intelligence and machine learning · 29 · 2 first-author · 29 since 2021Databases, data management, data science and information retrieval · 17 · 3 first-author · 17 since 2021Graphics, computer vision, multimedia, augmented reality and games · 14 · 14 since 2021Applied, interdisciplinary, general and emerging computing · 4 · 4 since 2021Computer networks · 2 · 2 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
30 papers
Graph learning · 34% Learning paradigms · 14% Time series and sequential data · 10%
Databases, data mining, and information retrieval
5 papers
Data mining · 96% Machine learning and data management · 4%
Interdisciplinary, comprehensive, and emerging computing
5 papers
Smart cities and intelligent transportation · 53% Computational science and engineering · 19% Environmental and earth informatics · 16%
Computer networks
1 paper
Cellular and mobile networks · 67% Network measurement and analytics · 33%

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

TopicWeightPapersLastEvidence papers
Machine learning › Graph learning
graph neural network
3.252025
RayE-Sub: Countering Subgraph Degradation via Perfect Reconstruction · IEEE Trans. Knowl. Data Eng. 2025
STONE: A Spatio-temporal OOD Learning Framework Kills Both Spatial and Temporal Shifts · KDD 2024
A Twist for Graph Classification: Optimizing Causal Information Flow in Graph Neural Networks · AAAI 2024
Machine learning › Graph learning › graph neural network › dynamic graph neural network
spatio-temporal graph neural network
3.242025
Less but More: Linear Adaptive Graph Learning Empowering Spatiotemporal Forecasting · NeurIPS 2025
Robust Spatio-Temporal Centralized Interaction for OOD Learning · ICML 2025
STONE: A Spatio-temporal OOD Learning Framework Kills Both Spatial and Temporal Shifts · KDD 2024
Machine learning › Learning paradigms › class imbalance
long-tailed learning
3.142025
Balancing Model Efficiency and Performance: Adaptive Pruner for Long-tailed Data · ICML 2025
LLM-AutoDA: Large Language Model-Driven Automatic Data Augmentation for Long-tailed Problems · NeurIPS 2024
Two Fists, One Heart: Multi-Objective Optimization Based Strategy Fusion for Long-tailed Learning · ICML 2024
Machine learning › Time series and sequential data › time series analysis
time series forecasting
2.632025
Less but More: Linear Adaptive Graph Learning Empowering Spatiotemporal Forecasting · NeurIPS 2025
MoFo: Empowering Long-term Time Series Forecasting with Periodic Pattern Modeling · NeurIPS 2025
STEM-LTS: Integrating Semantic-Temporal Dynamics in LLM-driven Time Series Analysis · AAAI 2025
Machine learning › Graph learning
graph representation learning
2.532026
U2B: Scale-unbiased Representation Converter for Graph Classification with Imbalanced and Balanced Scale Distributions · AAAI 2026
Delayed Bottlenecking: Alleviating Forgetting in Pre-trained Graph Neural Networks · IEEE Trans. Knowl. Data Eng. 2025
EXTRACT and REFINE: Finding a Support Subgraph Set for Graph Representation · KDD 2023
Machine learning › Graph learning
graph classification
2.532026
U2B: Scale-unbiased Representation Converter for Graph Classification with Imbalanced and Balanced Scale Distributions · AAAI 2026
When Imbalance Meets Imbalance: Structure-driven Learning for Imbalanced Graph Classification · WWW 2024
A Twist for Graph Classification: Optimizing Causal Information Flow in Graph Neural Networks · AAAI 2024
Machine learning › Learning paradigms
class imbalance
1.822026
Learning Dynamics as Feedback: An Adaptive Entropy Flow Dynamics Framework for Long-tailed Human Action Recognition · AAAI 2026
Kill Two Birds with One Stone: Rethinking Data Augmentation for Deep Long-tailed Learning · ICLR 2024
Machine learning › Trustworthy machine learning
fairness
1.622025
Balancing Model Efficiency and Performance: Adaptive Pruner for Long-tailed Data · ICML 2025
Kill Two Birds with One Stone: Rethinking Data Augmentation for Deep Long-tailed Learning · ICLR 2024
Machine learning › Learning paradigms
long-tailed recognition
1.622025
Deciphering the Extremes: A Novel Approach for Pathological Long-tailed Recognition in Scientific Discovery · NeurIPS 2025
Breaking Long-Tailed Learning Bottlenecks: A Controllable Paradigm with Hypernetwork-Generated Diverse Experts · NeurIPS 2024
Machine learning › Efficient and distributed learning
model compression
1.522025
Balancing Model Efficiency and Performance: Adaptive Pruner for Long-tailed Data · ICML 2025
Searching Lottery Tickets in Graph Neural Networks: A Dual Perspective · ICLR 2023
Machine learning › Time series and sequential data
spatiotemporal forecasting
1.522025
Less but More: Linear Adaptive Graph Learning Empowering Spatiotemporal Forecasting · NeurIPS 2025
A Multi-graph Fusion Based Spatiotemporal Dynamic Learning Framework · WSDM 2023
Machine learning › Graph learning › graph neural network
subgraph learning
1.522025
RayE-Sub: Countering Subgraph Degradation via Perfect Reconstruction · IEEE Trans. Knowl. Data Eng. 2025
EXTRACT and REFINE: Finding a Support Subgraph Set for Graph Representation · KDD 2023
Data mining
time series analysis
1.522025
STEM-LTS: Integrating Semantic-Temporal Dynamics in LLM-driven Time Series Analysis · AAAI 2025
An Observed Value Consistent Diffusion Model for Imputing Missing Values in Multivariate Time Series · KDD 2023
Machine learning › Deep learning architectures and training
data augmentation
1.522024
LLM-AutoDA: Large Language Model-Driven Automatic Data Augmentation for Long-tailed Problems · NeurIPS 2024
Kill Two Birds with One Stone: Rethinking Data Augmentation for Deep Long-tailed Learning · ICLR 2024
Machine learning › Graph learning
spatio-temporal graph learning
1.522024
Make Bricks with a Little Straw: Large-Scale Spatio-Temporal Graph Learning with Restricted GPU-Memory Capacity · IJCAI 2024
Towards Dynamic Spatial-Temporal Graph Learning: A Decoupled Perspective · AAAI 2024
Machine learning › Trustworthy machine learning
out-of-distribution generalization
1.122025
Robust Spatio-Temporal Centralized Interaction for OOD Learning · ICML 2025
STONE: A Spatio-temporal OOD Learning Framework Kills Both Spatial and Temporal Shifts · KDD 2024
Data mining › time series analysis
time series forecasting
1.122025
STEM-LTS: Integrating Semantic-Temporal Dynamics in LLM-driven Time Series Analysis · AAAI 2025
An Observed Value Consistent Diffusion Model for Imputing Missing Values in Multivariate Time Series · KDD 2023
Machine learning › Deep learning architectures and training
attention mechanism
1.022025
MoFo: Empowering Long-term Time Series Forecasting with Periodic Pattern Modeling · NeurIPS 2025
Countering Modal Redundancy and Heterogeneity: A Self-Correcting Multimodal Fusion · ICDM 2022
Computer vision › Video understanding and tracking
action recognition
1.012026
Learning Dynamics as Feedback: An Adaptive Entropy Flow Dynamics Framework for Long-tailed Human Action Recognition · AAAI 2026
Computer vision › Video understanding and tracking › action recognition
long-tailed action recognition
1.012026
Learning Dynamics as Feedback: An Adaptive Entropy Flow Dynamics Framework for Long-tailed Human Action Recognition · AAAI 2026
Computational science and engineering
materials science
1.012026
Rethinking Crystal Symmetry Prediction: A Decoupled Perspective · AAAI 2026
Machine learning › Trustworthy machine learning
robustness
1.022024
Two Fists, One Heart: Multi-Objective Optimization Based Strategy Fusion for Long-tailed Learning · ICML 2024
STONE: A Spatio-temporal OOD Learning Framework Kills Both Spatial and Temporal Shifts · KDD 2024
Machine learning › Graph learning › graph structure learning
adaptive graph learning
0.912025
Less but More: Linear Adaptive Graph Learning Empowering Spatiotemporal Forecasting · NeurIPS 2025
Machine learning › Efficient and distributed learning › model compression › pruning › dynamic pruning
adaptive pruning
0.912025
Balancing Model Efficiency and Performance: Adaptive Pruner for Long-tailed Data · ICML 2025
Machine learning › Graph learning › graph neural network
expressive power
0.912025
RayE-Sub: Countering Subgraph Degradation via Perfect Reconstruction · IEEE Trans. Knowl. Data Eng. 2025
Machine learning › Generative modeling
generative flow networks
0.912025
COFlowNet: Conservative Constraints on Flows Enable High-Quality Candidate Generation · ICLR 2025
Machine learning › Graph learning › graph pre-training
graph neural network pre-training
0.912025
Delayed Bottlenecking: Alleviating Forgetting in Pre-trained Graph Neural Networks · IEEE Trans. Knowl. Data Eng. 2025
Machine learning › Representation and self-supervised learning
information bottleneck
0.912025
Delayed Bottlenecking: Alleviating Forgetting in Pre-trained Graph Neural Networks · IEEE Trans. Knowl. Data Eng. 2025
Machine learning › Efficient and distributed learning › model compression
knowledge distillation
0.912025
Multi-label Self Knowledge Distillation · AAAI 2025
Machine learning › Time series and sequential data
large language model for time series
0.912025
STEM-LTS: Integrating Semantic-Temporal Dynamics in LLM-driven Time Series Analysis · AAAI 2025

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

diffusion model · 3.6large language model · 2.5graph neural network · 2.4knowledge distillation · 1.9attention · 1.5superclass guidance · 1.0multi-objective optimization · 1.0mixup · 1.0hierarchical PXRD pattern learning · 1.0entropy flow dynamics · 1.0consistency alignment · 1.0confidence-guided symbiosis · 1.0transformer · 0.9time series decomposition · 0.9subgraph partition · 0.9spatiotemporal decoupling · 0.9spatial decoupling · 0.9smooth objective regularization · 0.9
YearPublicationVenuePosition
2026 Learning Dynamics as Feedback: An Adaptive Entropy Flow Dynamics Framework for Long-tailed Human Action Recognition
abstract
Deep human action recognition models trained on real-world data are often challenged by long-tailed distributions, where performance on rare classes is severely degraded. Current solutions typically apply static or heuristic interventions that are disconnected from the model's evolving internal state. To overcome this limitation, we reconceptualize long-tailed human action recognition as a closed-loop, self-regulating system, inspired by ecological theory. We further introduce an Adaptive Ecological Entropy Dynamics (AEED) framework, which is built upon three synergistic components. First, AEED perceives the learning state through entropy flow, providing a robust and directional signal of learning progress. Second, this signal drives an adaptation mechanism, which dynamically adjusts class-specific loss weights to allocate more learning resources to underperforming classes. Finally, AEED facilitates intelligent knowledge transfer via Confidence-Guided Symbiosis (CS-Mix). Extensive experiments demonstrate that AEED achieves state-of-the-art performance on challenging skeleton-based action recognition benchmarks, including NTU-60-LT and Kinetics-400-LT.
Zhe Zhao 0008, Liheng Yu, Di Wu 0057, Pengkun Wang 0001
AAAI5
2026 U2B: Scale-unbiased Representation Converter for Graph Classification with Imbalanced and Balanced Scale Distributions
abstract
Graph classification is a critical task in analyzing graph data, with applications across various domains. While graph neural networks (GNNs) have achieved remarkable results, their ability to generalize across graphs of varying scales remains a challenge. Conventional models often perform well on large-scale graphs but struggle with distributions that are skewed towards small scales. Conversely, models tailored to address scale imbalances frequently prioritize small-scale graphs, leading to diminished performance in more balanced scenarios. To overcome these limitations, we introduce a Unbalanced-Balanced Representation Converter (U2B), which exhibits no explicit bias toward graph scales. U2B employs a two-step workflow: a distillation phase to extract base features from both node-level and graph-level representations, followed by a refinement phase to generate unbiased representations for improved balance. In the distillation phase, a static constraint guides node-level adjustments, improving the representation of nodes in small graphs. Simultaneously, a dynamic constraint in the graph-level process mitigates biases toward features from large graphs. To ensure harmony between the representations, a consistency alignment loss is introduced, aligning node-level and graph-level features to create more cohesive and balanced graph representations. Extensive experiments on multiple datasets show that U2B achieves competitive performance.
Jiaming Ma, Pengkun Wang 0001, Zhengyang Zhou, Binwu Wang, Yang Wang 0015
AAAI5
2026 Rethinking Crystal Symmetry Prediction: A Decoupled Perspective
abstract
Efficiently and accurately determining the symmetry is a crucial step in the structural analysis of crystalline materials. Existing methods usually mindlessly apply deep learning models while ignoring the underlying chemical rules. More importantly, experiments show that they face a serious sub-property confusion SPC problem. To address the above challenges, from a decoupled perspective, we introduce the XRDecoupler framework, a problem-solving arsenal specifically designed to tackle the SPC problem. Imitating the thinking process of chemists, we innovatively incorporate multidimensional crystal symmetry information as superclass guidance to ensure that the model's prediction process aligns with chemical intuition. We further design a hierarchical PXRD pattern learning model and a multi-objective optimization approach to achieve high-quality representation and balanced optimization. Comprehensive evaluations on three mainstream databases (e.g., CCDC, CoREMOF, and InorganicData) demonstrate that XRDecoupler excels in performance, interpretability, and generalization.
Liheng Yu, Zhe Zhao 0008, Xucong Wang, Di Wu 0057, Pengkun Wang 0001
AAAI5
2025 Multi-label Self Knowledge Distillation
abstract
Self-Knowledge Distillation (SKD) leverages the student's own knowledge to create a virtual teacher for distillation when the pre-trained bulky teacher is not available. Whilst existing SKD approaches demonstrate gorgeous efficiency in single-label learning, to directly apply them to multi-label learning would suffer from dramatic degradation due to the following inherent imbalance: \textit{targets with unified labels but multifarious visual scales are crammed into one image, resulting in biased learning of major targets and disequilibrium of precision-recall}. To address this issue, this paper proposes a novel SKD method for multi-label learning named Multi-label Self-knowledge Distillation (MSKD), incorporating three Spatial Decoupling mechanisms (i.e. Locality-SD (L-SD), Reconstruction-SD (R-SD), and Step-SD (S-SD)). L-SD exploits relational dark knowledge from regional outputs to amplify the model's perception of visual details. R-SD reconstructs global semantics by integrating regional outputs from local patches and leverages it to guide the model. S-SD aligns outputs of the same input at different steps, aiming to find a synthetical optimizing direction and avoid the overconfidence. In addition, MSKD combines our tailored loss named MBD for balanced distillation. Exhaustive experiments demonstrate that MSKD not only outperforms previous approaches but also effectively mitigates biased learning and equips the model with more robustness.
Xucong Wang, Pengkun Wang 0001, Yang Wang 0015
AAAI2
2025 STEM-LTS: Integrating Semantic-Temporal Dynamics in LLM-driven Time Series Analysis
abstract
Time series forecasting plays a crucial role in domains such as finance, healthcare, and climate science. However, as modern time series data become increasingly complex, featuring high dimensionality, intricate spatiotemporal dependencies, and multi-scale evolutionary patterns, traditional analytical methods and existing predictive models face significant challenges. Although Large Language Models (LLMs) excel in capturing long-range dependencies, they still struggle with multi-scale dynamics and seasonal patterns. Moreover, while LLMs' semantic representation capabilities are rich, they often lack explicit alignment with the numerical patterns and temporal structures of time series data, leading to limitations in predictive accuracy and interpretability. To address these challenges, this paper proposes a novel framework, STEM-LTS (Semantic-TEmporal Modeling for Large-scale Time Series). STEM-LTS enhances the ability to capture complex spatiotemporal dependencies by integrating time series decomposition techniques with LLM-based modeling. The semantic-temporal alignment mechanism within the framework significantly improves LLMs' ability to interpret and forecast time series data. Additionally, we develop an adaptive multi-task learning strategy to optimize the model's performance across multiple dimensions. Through extensive experiments on various real-world datasets, we demonstrate that STEM-LTS achieves significant improvements in prediction accuracy, robustness to noise, and interpretability. Our work not only advances LLM-based time series analysis but also offers new perspectives on handling complex temporal data.
Zhe Zhao 0008, Pengkun Wang 0001, Haibin Wen, Liheng Yu, Yang Wang 0015
AAAI2
2025 Spatiotemporal Causal Decoupling Model for Air Quality Forecasting
abstract
Due to the profound impact of air pollution on human health, livelihoods, and economic development, air quality forecasting is of paramount significance. Initially, we employ the causal graph method to scrutinize the constraints of existing research in comprehensively modeling the causal relationships between the air quality index (AQI) and meteorological features. In order to enhance prediction accuracy, we introduce a novel air quality forecasting model, AirCade, which incorporates a causal decoupling approach. AirCade leverages a spatiotemporal module in conjunction with knowledge embedding techniques to capture the internal dynamics of AQI. Subsequently, a causal decoupling module is proposed to disentangle synchronous causality from past AQI and meteorological features, followed by the dissemination of acquired knowledge to future time steps to enhance performance. Additionally, we introduce a causal intervention mechanism to explicitly represent the uncertainty of future meteorological features, thereby bolstering the model’s robustness. Our evaluation of AirCade on an open-source air quality dataset demonstrates over 20% relative improvement over state-of-the-art models. Our source code is available at https://github.com/PoorOtterBob/AirCade.
Jiaming Ma, Kuo Yang 0002, Binwu Wang, Pengkun Wang 0001, Yang Wang 0015
ICASSP6
2025 Embedding Enhanced MLP Enables Simple and Extensible Spatiotemporal Forecasting
abstract
Spatiotemporal forecasting facilitates many real world intelligent systems. Combining graph learning with temporal models has recently become popular in spatiotemporal forecasting. Although graph convolution enhances the modeling of spatial correlations, it results in unsatisfactory efficiency and poor extensibility in existing models. Consequently, existing graph-learning-based methods struggle to handle emerging nodes and are difficult to deploy on large-scale datasets. In this paper, we argue that graph learning in spatiotemporal forecasting is to discriminate different nodes thus generating node-specific features. To achieve the same effect of graph learning, we propose to use node embeddings to learn node-specific features and introduce a simple yet well-performing Embedding Enhanced MLP (E2MLP) spatiotemporal forecasting model. A hierarchical MLP framework is proposed, where node embeddings are introduced in each procedure of the framework. E2MLP outperforms existing SOTA spatiotemporal forecasting models on five real-world widely-used datasets, and can be easily scaled to newly emerging nodes without any performance influence on existing nodes. The source codes are publicly available from https://github.com/Ziyan2019/E2MLP
Yudong Zhang 0005, Pengkun Wang 0001, Xu Wang 0029, Yang Wang 0015
ICASSP3
2025 COFlowNet: Conservative Constraints on Flows Enable High-Quality Candidate Generation
abstract
Generative flow networks (GFlowNets) have been considered as powerful tools for generating candidates with desired properties. Given that evaluating the property of candidates can be complex and time-consuming, existing GFlowNets train proxy models for efficient online evaluation. However, the performance of proxy models is heavily dependent on the amount of data and is of considerable uncertainty. Therefore, it is of great interest that how to develop an offline GFlowNet that does not rely on online evaluation. Under the offline setting, the limited data results in an insufficient exploration of state space. The insufficient exploration means that offline GFlowNets can hardly generate satisfying candidates out of the distribution of training data. Therefore, it is critical to restrict the offline model to act in the distribution of training data. The distinctive training goal of GFlownets poses a unique challenge for making such restrictions. Tackling the challenge, we propose Conservative Offline GFlowNet (COFlowNet) in this paper. We define unsupported flow, edges containing unseen states in training data. Models can learn extremely little knowledge about unsupported flow from training data. By constraining the model from exploring unsupported flows, we restrict COFlowNet to explore as optimal trajectories on the training set as possible, thus generating better candidates. In order to improve the diversity of candidates, we further introduce a quantile version of unsupported flow restriction. Experimental results on several widely-used datasets validate the effectiveness of COFlowNet in generating high-scored and diverse candidates. All implementations are available at https://github.com/yuxuan9982/COflownet.
Yudong Zhang 0005, Xu Wang 0029, Zhaoyang Sun, Chen Zhang 0007, Pengkun Wang 0001, Yang Wang 0015
ICLR6
2025 Balancing Model Efficiency and Performance: Adaptive Pruner for Long-tailed Data
abstract
Long-tailed distribution datasets are prevalent in many machine learning tasks, yet existing neural network models still face significant challenges when handling such data. This paper proposes a novel adaptive pruning strategy, LTAP (Long-Tailed Adaptive Pruner), aimed at balancing model efficiency and performance to better address the challenges posed by long-tailed data distributions. LTAP introduces multi-dimensional importance scoring criteria and designs a dynamic weight adjustment mechanism to adaptively determine the pruning priority of parameters for different classes. By focusing on protecting parameters critical for tail classes, LTAP significantly enhances computational efficiency while maintaining model performance. This method combines the strengths of long-tailed learning and neural network pruning, overcoming the limitations of existing approaches in handling imbalanced data. Extensive experiments demonstrate that LTAP outperforms existing methods on various long-tailed datasets, achieving a good balance between model compression rate, computational efficiency, and classification accuracy. This research provides new insights into solving model optimization problems in long-tailed learning and is significant for improving the performance of neural networks on imbalanced datasets. The code is available at https://github.com/DataLab-atom/LT-VOTE.
Zhe Zhao 0008, Haibin Wen, Pengkun Wang 0001, Zhenkun Wang 0001, Qingfu Zhang 0001, Yang Wang 0015
ICML3
2025 Robust Spatio-Temporal Centralized Interaction for OOD Learning
abstract
Recently, spatiotemporal graph convolutional networks have achieved dominant performance in spatiotemporal prediction tasks. However, most models relying on node-to-node messaging interaction exhibit sensitivity to spatiotemporal shifts, encountering out-of-distribution (OOD) challenges. To address these issues, we introduce \textbf{\underline{S}}patio-\textbf{\underline{T}}emporal \textbf{\underline{O}}OD \textbf{\underline{P}}rocessor (STOP), which employs a centralized messaging mechanism along with a message perturbation mechanism to facilitate robust spatiotemporal interactions. Specifically, the centralized messaging mechanism integrates Context-Aware Units for coarse-grained spatiotemporal feature interactions with nodes, effectively blocking traditional node-to-node messages. We also implement a message perturbation mechanism to disrupt this messaging process, compelling the model to extract generalizable contextual features from generated variant environments. Finally, we customize a spatiotemporal distributionally robust optimization approach that exposes the model to challenging environments, thereby further enhancing its generalization capabilities. Compared with 14 baselines across six datasets, STOP achieves up to \textbf{17.01\%} improvement in generalization performance and \textbf{18.44\%} improvement in inductive learning performance. The code is available at https://github.com/PoorOtterBob/STOP.
Jiaming Ma, Binwu Wang, Pengkun Wang 0001, Zhengyang Zhou, Xu Wang 0029, Yang Wang 0015
ICML3
2025 Causal Learning Meet Covariates: Empowering Lightweight and Effective Nationwide Air Quality Forecasting
abstract
Air quality prediction plays a crucial role in the development of smart cities, garnering significant attention from both academia and industry. Current air quality prediction models encounter two major limitations: their high computational complexity limits scalability to nationwide datasets, and they often regard weather covariates as optional auxiliary information. In reality, weather covariates can have a substantial impact on air quality indices (AQI), exhibiting a significant causal association. In this paper, we first present a nationwide air quality dataset to address the lack of open-source, large-scale datasets in this field. Then we propose a causal learning model, CauAir, for air quality prediction that harnesses the powerful representation capabilities of the Transformer to explicitly model the causal association between weather covariates and AQI. To address the high complexity of traditional Transformers, we design CachLormer, which features two key innovations: a simplified architecture with redundant components removed, and a cache-attention mechanism that employs learnable embeddings for perceiving causal association between AQI and weather covariates in a coarsegrained perspective. We use information theory to illustrate the superiority of the proposed model. Finally, experimental results on three datasets with 28 as the baseline demonstrate that our model achieves competitive performance, while maintaining high training efficiency and low memory consumption. The source code is available at CauAir Official Repository.
Jiaming Ma, Zhiqing Cui, Binwu Wang, Pengkun Wang 0001, Zhengyang Zhou, Zhe Zhao 0008, Yang Wang 0015
IJCAI4
2025 Delving into Imbalance in Imbalance: Attribute Rebalancing for Generalizing Long-tailed Learning
abstract
Real-world data tends to follow a long-tailed distribution, which greatly challenges the robustness of algorithms. Current long-tailed learning methods usually focus on devising exquisite re-sampling strategies or innovative loss functions to address the class-wise imbalance, yet rarely pay attention to the pervasive problem of attribute imbalance within classes. That is to say, attribute imbalance naturally exists in class imbalance, which brings more complex challenges to current long-tailed learning. In this work, we propose an attribute rebalancing strategy to address imbalance in imbalance. First, we propose attribute-balanced re-sampling, which potentially re-samples the attributes within the class during up/down-sampling to avoid damage to rare attributes. Then, we design an attribute-constrained loss to trade off the intertwined imbalances in class and attribute levels. Comprehensive experiments show that attribute rebalancing can consistently perform well on generalized long-tailed benchmark datasets.
Shuru Zhang, Pengkun Wang 0001, Jason Zeng, Michael Heinrich, Ji Qi 0005
IJCNN2
2025 MoFo: Empowering Long-term Time Series Forecasting with Periodic Pattern Modeling
abstract
The stable periodic patterns present in the time series data serve as the foundation for long-term forecasting. However, existing models suffer from limitations such as continuous and chaotic input partitioning, as well as weak inductive biases, which restrict their ability to capture such recurring structures. In this paper, we propose MoFo, which interprets periodicity as both the correlation of period-aligned time steps and the trend of period-offset time steps. We first design period-structured patches—2D tensors generated through discrete sampling—where each row contains only period-aligned time steps, enabling direct modeling of periodic correlations. Period-offset time steps within a period are aligned in columns. To capture trends across these offset time steps, we introduce a period-aware modulator. This modulator introduces an adaptive strong inductive bias through a regulated relaxation function, encouraging the model to generate attention coefficients that align with periodic trends. This function is end-to-end trainable, enabling the model to adaptively capture the distinct periodic patterns across diverse datasets. Extensive empirical results on widely used benchmark datasets demonstrate that MoFo achieves competitive performance while maintaining high memory efficiency and fast training speed.
Jiaming Ma, Binwu Wang, Qihe Huang, Pengkun Wang 0001, Zhengyang Zhou, Yang Wang 0015
NeurIPS5
2025 Less but More: Linear Adaptive Graph Learning Empowering Spatiotemporal Forecasting
abstract
The effectiveness of Spatiotemporal Graph Neural Networks (STGNNs) critically hinges on the quality of the underlying graph topology. While end-to-end adaptive graph learning methods have demonstrated promising results in capturing latent spatiotemporal dependencies, they often suffer from high computational complexity and limited expressive capacity. In this paper, we propose MAGE for efficient spatiotemporal forecasting. We first conduct a theoretical analysis demonstrating that the ReLU activation function employed in existing methods amplifies edge-level noise during graph topology learning, thereby compromising the fidelity of the learned graph structures. To enhance model expressiveness, we introduce a sparse yet balanced mixture-of-experts strategy, where each expert perceives the unique underlying graph through kernel-based functions and operates with linear complexity relative to the number of nodes. The sparsity mechanism ensures that each node interacts exclusively with compatible experts, while the balancing mechanism promotes uniform activation across all experts, enabling diverse and adaptive graph representations. Furthermore, we theoretically establish that a single graph convolution using the learned graph in MAGE is mathematically equivalent to multiple convolutional steps under conventional graphs. We evaluate MAGE against advanced baselines on multiple real-world spatiotemporal datasets. MAGE achieves competitive performance while maintaining strong computational efficiency.
Jiaming Ma, Binwu Wang, Kuo Yang 0002, Zhengyang Zhou, Pengkun Wang 0001, Xu Wang 0029, Yang Wang 0015
NeurIPS6
2025 TS-MOF: Two-Stage Multi-Objective Fine-tuning for Long-Tailed Recognition
abstract
Long-Tailed Recognition (LTR) presents a significant challenge due to extreme class imbalance, where existing methods often struggle to balance performance across head and tail classes. Directly applying multi-objective optimization (MOO) to leverage multiple LTR strategies can be complex and unstable. To address this, we propose TS-MOF (Two-Stage Multi-Objective Fine-tuning), a novel framework that strategically decouples feature learning from classifier adaptation. After standard pre-training, TS-MOF freezes the feature backbone and focuses on an efficient multi-objective fine-tuning of specialized classifier heads. The core of TS-MOF's second stage lies in two innovations: Refined Performance Level Agreement for adaptive task weighting based on real-time per-class performance, and Robust Deterministic Projective Conflict Gradient for stable gradient conflict resolution and constructive fusion. This approach enables effective synergy between diverse LTR strategies, leading to significant and balanced performance improvements. Extensive experiments on CIFAR100-LT, ImageNet-LT, and iNaturalist 2018 demonstrate that TS-MOF achieves state-of-the-art results, particularly enhancing tail class accuracy (e.g., +3.3\% on CIFAR100-LT IR=100 tail) while improving head class performance, all within a remarkably short fine-tuning period of 20 epochs.
Zhe Zhao 0008, Zhiheng Gong, Pengkun Wang 0001, Haibin Wen, Cankun Guo, Bo Xue 0004, Xi Lin 0001, Zhenkun Wang 0001, Qingfu Zhang 0001, Yang Wang 0015
NeurIPS3
2025 Deciphering the Extremes: A Novel Approach for Pathological Long-tailed Recognition in Scientific Discovery
abstract
Scientific discovery across diverse fields increasingly grapples with datasets exhibiting pathological long-tailed distributions: a few common phenomena overshadow a multitude of rare yet scientifically critical instances. Unlike standard benchmarks, these scientific datasets often feature extreme imbalance coupled with a modest number of classes and limited overall sample volume, rendering existing long-tailed recognition (LTR) techniques ineffective. Such methods, biased by majority classes or prone to overfitting on scarce tail data, frequently fail to identify the very instances—novel materials, rare disease biomarkers, faint astronomical signals—that drive scientific breakthroughs. This paper introduces a novel, end-to-end framework explicitly designed to address pathological long-tailed recognition in scientific contexts. Our approach synergizes a Balanced Supervised Contrastive Learning (B-SCL) mechanism, which enhances the representation of tail classes by dynamically re-weighting their contributions, with a Smooth Objective Regularization (SOR) strategy that manages the inherent tension between tail-class focus and overall classification performance. We introduce and analyze the real-world ZincFluor chemical dataset ($\mathcal{T}=137.54$) and synthetic benchmarks with controllable extreme imbalances (CIFAR-LT variants). Extensive evaluations demonstrate our method's superior ability to decipher these extremes. Notably, on ZincFluor, our approach achieves a Tail Top-2 accuracy of $66.84\%$, significantly outperforming existing techniques. On CIFAR-10-LT with an imbalance ratio of $1000$ ($\mathcal{T}=100$), our method achieves a tail-class accuracy of $38.99\%$, substantially leading the next best. These results underscore our framework's potential to unlock novel insights from complex, imbalanced scientific datasets, thereby accelerating discovery.
Zhe Zhao 0008, Haibin Wen, Xianfu Liu, Pengkun Wang 0001, Liheng Yu, Linjiang Chen, Bo An 0001, Qingfu Zhang 0001, Yang Wang 0015
NeurIPS5
2025 Soft causal learning for generalized molecule property prediction: An environment modeling perspective
Zhengyang Zhou, Kuo Yang 0002, Wenjie Du 0003, Pengkun Wang 0001, Yang Wang 0015
Knowl. Inf. Syst.5
2025 BiST: A Lightweight and Efficient Bi-directional Model for Spatiotemporal Prediction
abstract
While existing spatiotemporal prediction models have shown promising performance, they often rely on the assumption of input-label spatiotemporal consistency, and their high complexity raises concerns about scalability. To enhance both efficiency and performance, we integrate label information into the learning process and propose a spatiotemporal dynamic theory that outlines a bi-directional learning paradigm. Building on this paradigm, we design BiST, a lightweight yet effective Bi -directional S patio -T emporal prediction model. BiST incorporates two key processes: a forward spatiotemporal learning process and a backward correction process. The forward process utilizes MLP layers exclusively to model input correlations and generate base prediction. In the backward process, we implement a spatiotemporal decoupling module, which can learn the residual modeling deviation between input and label representations from a decoupled perspective. After smoothing the residual with a diffusion module, we can obtain the correction term to correct the base predictions. This innovative design enables BiST to achieve competitive performance while remaining lightweight. We evaluate BiST against 26 baselines across 13 datasets, including a large-scale dataset with ten thousand nodes and a longrange dataset spanning 20 years. An impressive experimental result demonstrates that BiST achieves a 8.13% improvement in performance compared to state-of-the-art models while consuming only 1.86% of the training time and 7.36% of the memory usage.
Jiaming Ma, Binwu Wang, Pengkun Wang 0001, Zhengyang Zhou, Xu Wang 0029, Yang Wang 0015
Proc. VLDB Endow.3
2025 RayE-Sub: Countering Subgraph Degradation via Perfect Reconstruction
abstract
Subgraph learning has dominated most practices of improving the expressive power of Message Passing Neural Networks (MPNNs). Existing subgraph discovery policies can be classified into node-based and partition-based, which both achieve impressive performance in most scenarios. However, both mainstream solutions still face a subgraph degradation trap. Subgraph degradation is reflected in the phenomenon that the subgraph-level methods fail to offer any benefits over node-level MPNNs. In this work, we empirically investigate the existence of the subgraph degradation issue and introduce a unified perspective, perfect reconstruction, to provide insights for improving two lines of methods. We further propose a subgraph learning strategy guided by the principle of perfect reconstruction. To achieve this, two major issues should be well-addressed, i.e.,(i) how to ensure the subgraphs to possess with ‘perfect’ information? (ii) how to guarantee the ‘reconstruction’ power of obtained subgraphs?First, we propose a subgraph partition strategyRayleigh-resistanceto extract non-overlap subgraphs by leveraging the graph spectral theory. Second, we put forward aQuerymechanism to achieve subgraph-level equivariant learning, which guarantees subgraph reconstruction ability. These two parts,perfect subgraph partitionandequivariant subgraph learningare seamlessly unified as a novelRayleigh-resistanceEquivariantSubgraph learningarchitecture (RayE-Sub). Comprehensive experiments on both synthetic and real datasets demonstrate that our approach can consistently outperform previous subgraph learning architectures.
Kuo Yang 0002, Zhengyang Zhou, Xu Wang 0029, Pengkun Wang 0001, Yang Wang 0015
IEEE Trans. Knowl. Data Eng.4
2025 Delayed Bottlenecking: Alleviating Forgetting in Pre-trained Graph Neural Networks
abstract
Pre-training GNNs to extract transferable knowledge and apply it to downstream tasks has become the de facto standard of graph representation learning. Recent works focused on designing self-supervised pre-training tasks to extract useful and universal transferable knowledge from large-scale unlabeled data. However, they have to face an inevitable question: traditional pre-training strategies that aim at extracting useful information about pre-training tasks, may not extract all useful information about the downstream task. In this paper, we reexamine the pre-training process within traditional pre-training and fine-tuning frameworks from the perspective of Information Bottleneck (IB) and confirm that the forgetting phenomenon in pre-training phase may cause detrimental effects on downstream tasks. Therefore, we propose a novelDelayedBottleneckingPre-training (DBP) framework which maintains as much as possible mutual information between latent representations and training data during pre-training phase by suppressing the compression operation and delays the compression operation to fine-tuning phase to make sure the compression can be guided with labeled fine-tuning data and downstream tasks. To achieve this, we design two information control objectives that can be directly optimized and further integrate them into the actual model design. Extensive experiments on both chemistry and biology domains demonstrate the effectiveness of DBP.
Zhe Zhao 0008, Pengkun Wang 0001, Xu Wang 0029, Haibin Wen, Xiaolong Xie, Zhengyang Zhou, Qingfu Zhang 0001, Yang Wang 0015
IEEE Trans. Knowl. Data Eng.2
2025 MobiMixer: A Multi-Scale Spatiotemporal Mixing Model for Mobile Traffic Prediction
abstract
Understanding mobile traffic data and predicting future trends are essential for wireless operators and service providers to allocate resources efficiently and manage energy effectively. Despite the strong performance of existing models, accurately forecasting mobile traffic remains a challenge due to limited spatial and temporal modeling capabilities and high computational complexity. This paper introduces MobiMixer, a lightweight and efficient multi-scale spatiotemporal mixing model. Its core concept is to integrate multi-scale information from both spatial and temporal dimensions to improve performance on mobile traffic data. We develop a hierarchical interaction module that incorporates super nodes to enable global high-level feature interactions among nodes with common patterns. Additionally, we employ a dynamic time warping strategy to decouple mobile traffic sequences into stable and seasonal components, which are then modeled at different scales using a multi-scale temporal mixing module. We conduct extensive experiments on mobile traffic datasets collected from four international cities. Compared with 21 state-of-the-art benchmark models, MobiMixer demonstrates highly competitive performance, achieving a maximum improvement of 48.49% on the Milan mobile dataset. The model achieves an improvement in training efficiency of up to 10.69 times and reduces memory usage by 33.01%. The source code is available athttps://github.com/PoorOtterBob/Submitted_Code.
Jiaming Ma, Binwu Wang, Pengkun Wang 0001, Zhengyang Zhou, Yudong Zhang 0005, Xu Wang 0029, Yang Wang 0015
IEEE Trans. Mob. Comput.3
2024 Towards Dynamic Spatial-Temporal Graph Learning: A Decoupled Perspective
abstract
With the progress of urban transportation systems, a significant amount of high-quality traffic data is continuously collected through streaming manners, which has propelled the prosperity of the field of spatial-temporal graph prediction. In this paper, rather than solely focusing on designing powerful models for static graphs, we shift our focus to spatial-temporal graph prediction in the dynamic scenario, which involves a continuously expanding and evolving underlying graph. To address inherent challenges, a decoupled learning framework (DLF) is proposed in this paper, which consists of a spatial-temporal graph learning network (DSTG) with a specialized decoupling training strategy. Incorporating inductive biases of time-series structures, DSTG can interpret time dependencies into latent trend and seasonal terms. To enable prompt adaptation to the evolving distribution of the dynamic graph, our decoupling training strategy is devised to iteratively update these two types of patterns. Specifically, for learning seasonal patterns, we conduct thorough training for the model using a long time series (e.g., three months of data). To enhance the learning ability of the model, we also introduce the masked auto-encoding mechanism. During this period, we frequently update trend patterns to expand new information from dynamic graphs. Considering both effectiveness and efficiency, we develop a subnet sampling strategy to select a few representative nodes for fine-tuning the weights of the model. These sampled nodes cover unseen patterns and previously learned patterns. Experiments on dynamic spatial-temporal graph datasets further demonstrate the competitive performance, superior efficiency, and strong scalability of the proposed framework.
Binwu Wang, Pengkun Wang 0001, Yudong Zhang 0005, Xu Wang 0029, Zhengyang Zhou, Lei Bai 0001, Yang Wang 0015
AAAI2
2024 A Twist for Graph Classification: Optimizing Causal Information Flow in Graph Neural Networks
abstract
Graph neural networks (GNNs) have achieved state-of-the-art results on many graph representation learning tasks by exploiting statistical correlations. However, numerous observations have shown that such correlations may not reflect the true causal mechanisms underlying the data and thus may hamper the ability of the model to generalize beyond the observed distribution. To address this problem, we propose an Information-based Causal Learning (ICL) framework that combines information theory and causality to analyze and improve graph representation learning to transform information relevance to causal dependence. Specifically, we first introduce a multi-objective mutual information optimization objective derived from information-theoretic analysis and causal learning principles to simultaneously extract invariant and interpretable causal information and reduce reliance on non-causal information in correlations. To optimize this multi-objective objective, we enable a causal disentanglement layer that effectively decouples the causal and non-causal information in the graph representations. Moreover, due to the intractability of mutual information estimation, we derive variational bounds that enable us to transform the above objective into a tractable loss function. To balance the multiple information objectives and avoid optimization conflicts, we leverage multi-objective gradient descent to achieve a stable and efficient transformation from informational correlation to causal dependency. Our approach provides important insights into modulating the information flow in GNNs to enhance their reliability and generalization. Extensive experiments demonstrate that our approach significantly improves the robustness and interpretability of GNNs across different distribution shifts. Visual analysis demonstrates how our method converts informative dependencies in representations into causal dependencies.
Zhe Zhao 0008, Pengkun Wang 0001, Haibin Wen, Yudong Zhang 0005, Zhengyang Zhou, Yang Wang 0015
AAAI2
2024 XRDMamba: Large-scale Crystal Material Space Group Identification with Selective State Space Model
abstract
In material science, the properties of crystalline materials largely depend on their structures, and space group is a key descriptor of crystal structure. With the rapid advancement of deep learning, the traditional artificial structure analysis method based on X-ray diffraction (XRD) has become cumbersome and is being gradually supplanted by neural networks. However, existing models are too simplistic and lack a comprehensive understanding of material structure. Our approach XRDMamba integrates chemical knowledge and presents a fresh crystal planes perspective on XRD data. We also introduce a knowledge-driven model for space group identification tasks. We have thoroughly analyzed our approach through numerous experiments, observing its SOTA performance and excellent generalization capabilities. The code is available in ~https://github.com/baigeiguai/XRDMamba.
Liheng Yu, Pengkun Wang 0001, Zhe Zhao 0008, Zhongchao Yi, Sun Nan, Di Wu 0057, Yang Wang 0015
CIKM2
2024 Gradient Reactivation Enhanced Causal Attention for Out-Of-Distribution Generalizable Graph Classification
abstract
Seeking for generalizable graph representations becomes hot spot in the area of graph learning. Recently, causality theory has been applied for extracting the causal relations between graph data and labels, which are generalizable under distribution shift and result in better OOD generalization. In this paper, for more accurately capturing causal representation of graph data, we propose a gradient reactivation enhanced causal subgraph extraction method. The proposed model utilizes attention mechanism to extract the causal features and attenuates the confounding effect of shortcut features. For ensuring stability of extracted causal features, we propose a novel gradient reactivation method to filter features with greater effect on making prediction. Extensively experimental result proves the effectiveness of the proposed model.
Xu Wang 0029, Pengfei Gu, Yudong Zhang 0005, Binwu Wang, Pengkun Wang 0001, Yang Wang 0015
ICASSP5
2024 Graph Networks Stand Strong: Enhancing Robustness via Stability Constraints
abstract
Graph neural networks (GNNs) have achieved great success in graph classification tasks across many domains. However, the varying quality of real-world graph data leads to stability and reliability issues for real-world applications of graph neural networks (GNNs). Improving the robustness of GNNs would help enhance the quality and safety of GNNs in real-world applications. Recently, there have been studies that incorporate insights from information theory, causal theory, etc. into graph classification tasks to improve robustness. However, these strategies rely on extensive task-specific designs that increase model complexity and limit the scope of the methods. In this work, we leverage the interdependence between model stability and robustness by introducing stability constraints to graph neural network models through two different consistency regularization methods. To balance the trade-off between stability constraints and classification performance, we adaptively adjust the strength of the constraints dynamically using multi-objective optimization, making our method applicable to graph classification tasks of varying scales and domains. Extensive experiments on graph datasets from different domains demonstrate the superiority of our proposed method.
Zhe Zhao 0008, Pengkun Wang 0001, Haibin Wen, Yudong Zhang 0005, Binwu Wang, Yang Wang 0015
ICASSP2
2024 Kill Two Birds with One Stone: Rethinking Data Augmentation for Deep Long-tailed Learning
abstract
Real-world tasks are universally associated with training samples that exhibit a long-tailed class distribution, and traditional deep learning models are not suitable for fitting this distribution, thus resulting in a biased trained model. To surmount this dilemma, massive deep long-tailed learning studies have been proposed to achieve inter-class fairness models by designing sophisticated sampling strategies or improving existing model structures and loss functions. Habitually, these studies tend to apply data augmentation strategies to improve the generalization performance of their models. However, this augmentation strategy applied to balanced distributions may not be the best option for long-tailed distributions. For a profound understanding of data augmentation, we first theoretically analyze the gains of traditional augmentation strategies in long-tailed learning, and observe that augmentation methods cause the long-tailed distribution to be imbalanced again, resulting in an intertwined imbalance: inherent data-wise imbalance and extrinsic augmentation-wise imbalance, i.e., two 'birds' co-exist in long-tailed learning. Motivated by this observation, we propose an adaptive Dynamic Optional Data Augmentation (DODA) to address this intertwined imbalance, i.e., one 'stone' simultaneously 'kills' two 'birds', which allows each class to choose appropriate augmentation methods by maintaining a corresponding augmentation probability distribution for each class during training. Extensive experiments across mainstream long-tailed recognition benchmarks (e.g., CIFAR-100-LT, ImageNet-LT, and iNaturalist 2018) prove the effectiveness and flexibility of the DODA in overcoming the intertwined imbalance.
Binwu Wang, Pengkun Wang 0001, Wei Xu 0055, Xu Wang 0029, Yudong Zhang 0005, Kun Wang 0056, Yang Wang 0015
ICLR2
2024 Two Fists, One Heart: Multi-Objective Optimization Based Strategy Fusion for Long-tailed Learning
abstract
Real-world data generally follows a long-tailed distribution, which makes traditional high-performance training strategies unable to show their usual effects. Various insights have been proposed to alleviate this challenging distribution. However, some observations indicate that models trained on long-tailed distributions always show a trade-off between the performance of head and tail classes. For a profound understanding of the trade-off, we first theoretically analyze the trade-off problem in long-tailed learning and creatively transform the trade-off problem in long-tailed learning into a multi-objective optimization (MOO) problem. Motivated by these analyses, we propose the idea of strategy fusion for MOO long-tailed learning and point out the potential conflict problem. We further design a Multi-Objective Optimization based Strategy Fusion (MOOSF), which effectively resolves conflicts, and achieves an efficient fusion of heterogeneous strategies. Comprehensive experiments on mainstream datasets show that even the simplest strategy fusion can outperform complex long-tailed strategies. More importantly, it provides a new perspective for generalized long-tailed learning. The code is available in the accompanying supplementary materials.
Zhe Zhao 0008, Pengkun Wang 0001, Haibin Wen, Wei Xu 0055, Song Lai 0001, Qingfu Zhang 0001, Yang Wang 0015
ICML2
2024 Make Bricks with a Little Straw: Large-Scale Spatio-Temporal Graph Learning with Restricted GPU-Memory Capacity
Binwu Wang, Pengkun Wang 0001, Zhengyang Zhou, Zhe Zhao 0008, Wei Xu 0055, Yang Wang 0015
IJCAI2
2024 STONE: A Spatio-temporal OOD Learning Framework Kills Both Spatial and Temporal Shifts
abstract
Traffic prediction is a crucial task in the Intelligent Transportation System (ITS), receiving significant attention from both industry and academia. Numerous spatio-temporal graph convolutional networks have emerged for traffic prediction and achieved remarkable success. However, these models have limitations in terms of generalization and scalability when dealing with Out-of-Distribution (OOD) graph data with both structural and temporal shifts. To tackle the challenges of spatio-temporal shift, we propose a framework called STONE by learning invariable node dependencies, which achieve stable performance in variable environments. STONE initially employs gated-transformers to extract spatial and temporal semantic graphs. These two kinds of graphs represent spatial and temporal dependencies, respectively. Then we design three techniques to address spatio-temporal shifts. Firstly, we introduce a Fréchet embedding method that is insensitive to structural shifts, and this embedding space can integrate loose position dependencies of nodes within the graph. Secondly, we propose a graph intervention mechanism to generate multiple variant environments by perturbing two kinds of semantic graphs without any data augmentations, and STONE can explore invariant node representation from environments. Finally, we further introduce an explore-to-extrapolate risk objective to enhance the variety of generated environments. We conduct experiments on multiple traffic datasets, and the results demonstrate that our proposed model exhibits competitive performance in terms of generalization and scalability.
Binwu Wang, Jiaming Ma, Pengkun Wang 0001, Xu Wang 0029, Yudong Zhang 0005, Zhengyang Zhou, Yang Wang 0015
KDD3
2024 LLM-AutoDA: Large Language Model-Driven Automatic Data Augmentation for Long-tailed Problems
abstract
The long-tailed distribution is the underlying nature of real-world data, and it presents unprecedented challenges for training deep learning models. Existing long-tailed learning paradigms based on re-balancing or data augmentation have partially alleviated the long-tailed problem. However, they still have limitations, such as relying on manually designed augmentation strategies, having a limited search space, and using fixed augmentation strategies. To address these limitations, this paper proposes a novel LLM-based long-tailed data augmentation framework called LLM-AutoDA, which leverages large-scale pretrained models to automatically search for the optimal augmentation strategies suitable for long-tailed data distributions. In addition, it applies this strategy to the original imbalanced data to create an augmented dataset and fine-tune the underlying long-tailed learning model. The performance improvement on the validation set serves as a reward signal to update the generation model, enabling the generation of more effective augmentation strategies in the next iteration. We conducted extensive experiments on multiple mainstream long-tailed learning benchmarks. The results show that LLM-AutoDA outperforms state-of-the-art data augmentation methods and other re-balancing methods significantly.
Pengkun Wang 0001, Zhe Zhao 0008, Haibin Wen, Fanfu Wang, Binwu Wang, Qingfu Zhang 0001, Yang Wang 0015
NeurIPS1
2024 Breaking Long-Tailed Learning Bottlenecks: A Controllable Paradigm with Hypernetwork-Generated Diverse Experts
abstract
Traditional long-tailed learning methods often perform poorly when dealing with inconsistencies between training and test data distributions, and they cannot flexibly adapt to different user preferences for trade-offs between head and tail classes. To address this issue, we propose a novel long-tailed learning paradigm that aims to tackle distribution shift in real-world scenarios and accommodate different user preferences for the trade-off between head and tail classes. We generate a set of diverse expert models via hypernetworks to cover all possible distribution scenarios, and optimize the model ensemble to adapt to any test distribution. Crucially, in any distribution scenario, we can flexibly output a dedicated model solution that matches the user's preference. Extensive experiments demonstrate that our method not only achieves higher performance ceilings but also effectively overcomes distribution shift while allowing controllable adjustments according to user preferences. We provide new insights and a paradigm for the long-tailed learning problem, greatly expanding its applicability in practical scenarios. The code can be found here: https://github.com/DataLab-atom/PRL.
Zhe Zhao 0008, Haibin Wen, Zikang Wang, Pengkun Wang 0001, Fanfu Wang, Song Lai 0001, Qingfu Zhang 0001, Yang Wang 0015
NeurIPS4
2024 When Imbalance Meets Imbalance: Structure-driven Learning for Imbalanced Graph Classification
abstract
Graph Neural Networks (GNNs) can learn representative graph-level features to achieve efficient graph classification. But GNNs usually assume an environment where both class and structure distribution are balanced. Although previous works have considered the graph classification problem under the scenario of class imbalance or structure imbalance, they habitually ignored the obvious fact that class imbalance and structural imbalance are often intertwined in the real world. In this paper, we propose a carefully designed structure-driven learning framework called ImbGNN to address the potential intertwined class imbalance and structural imbalance in graph classification. Specifically, we find that feature-oriented augmentation (e.g., feature masking) and structure-oriented augmentation (e.g., edge perturbation) will have differential impacts when applied to different graphs. Therefore, we design optional augmentation based on the average degree distribution to alleviate structural imbalance. Furthermore, based on the imbalance of graph size distribution, we utilize a similarity-friendly graph random walk to extract a core subgraph to improve the accuracy of graph kernel similarity calculation, and then construct a more reasonable kernel-based graph of graphs, thereby alleviating the class imbalance and size imbalance. Extensive experiments on multiple benchmark datasets demonstrate that our proposed ImbGNN framework outperforms previous baselines on imbalanced graph classification tasks. The code of ImbGNN is available in~https://github.com/Xiaovy/ImbGNN.
Wei Xu 0055, Pengkun Wang 0001, Zhe Zhao 0008, Binwu Wang, Xu Wang 0029, Yang Wang 0015
WWW2
2024 Meta Koopman decomposition for time series forecasting under temporal distribution shifts
Yudong Zhang 0005, Xu Wang 0029, Zhaoyang Sun, Pengkun Wang 0001, Binwu Wang, Yang Wang 0015
Adv. Eng. Informatics4
2024 Face Anti-Spoofing with Unknown Attacks: A Comprehensive Feature Extraction and Representation Perspective
Li-Min Li, Binwu Wang, Xu Wang 0029, Pengkun Wang 0001, Yudong Zhang 0005, Yang Wang 0015
J. Comput. Sci. Technol.4
2024 Adaptive and Interactive Multi-Level Spatio-Temporal Network for Traffic Forecasting
abstract
Traffic forecasting is a challenging research topic due to the complex spatial and temporal dependencies among different roads. Though great efforts have been made on traffic forecasting, existing works still have the following shortcomings: i) Most methods only directly perform on the original road network topology which cannot accommodate the diverse traffic patterns and multi-granularity traffic forecasting requirements driven by the natural multi-level urban structure and layout, ii) The existing studies based on the spatio-temporal multi-granularity perspective ignore the interactions between the fine-grained information and coarse-grained information, resulting in the spatio-temporal correlation under multi-granularity inaccurately modeled. To solve the problems, we propose an Adaptive and Interactive Multi-level Spatio-Temporal network (AIMST) for traffic forecasting. Specifically, we first devise a learnable adaptive hierarchical clustering method to automatically generate more coarse-grained graphs from the initial road networks and the traffic data. Then, the spatio-temporal graph convolutional networks are executed on the constructed hierarchical traffic graph of each level correspondingly to capture the spatio-temporal patterns. Furthermore, a multi-level bidirectional interaction module is designed to emphasize the multi-grained interaction patterns among different levels. Extensive experiments on two real-world traffic datasets demonstrate that our framework is superior to several state-of-the-art baselines.
Yudong Zhang 0005, Pengkun Wang 0001, Binwu Wang, Xu Wang 0029, Zhe Zhao 0008, Zhengyang Zhou, Lei Bai 0001, Yang Wang 0015
IEEE Trans. Intell. Transp. Syst.2
2024 Modeling Spatio-Temporal Mobility Across Data Silos via Personalized Federated Learning
abstract
Spatio-temporal mobility modeling plays a pivotal role in the advancement of mobile computing. Nowadays, data is frequently held by various distributed silos, which are isolated from each other and confront limitations on data sharing. Given this, there have been some attempts to introduce federated learning into spatio-temporal mobility modeling. Meanwhile, the distributional heterogeneity inherent in the spatio-temporal data also puts forward requirements for model personalization. However, the existing methods tackle personalization in a model-centric manner and fail to explore the data characteristics in various data silos, thus ignoring the fact that the fundamental cause of insufficient personalization in the model is the heterogeneous distribution of data. In this paper, we propose a novel distribution-oriented personalizedFederated learning framework forCross-siloSpatio-Temporal mobility modeling (namedFedCroST), that leverages learnable spatio-temporal prompts to implicitly represent the local data distribution patterns of data silos and guide the local models to learn the personalized information. Specifically, we focus on the potential characteristics within temporal distribution and devise a conditional diffusion module to generate temporal prompts that serve as guidance for the evolution of the time series. Simultaneously, we emphasize the structure distribution inherent in node neighborhoods and propose adaptive spatial structure partition to construct the spatial prompts, augmenting the spatial information representation. Furthermore, we introduce a denoising autoencoder to effectively harness the learned multi-view spatio-temporal features and obtain personalized representations adapted to local tasks. Our proposal highlights the significance of latent spatio-temporal data distributions in enabling personalized federated spatio-temporal learning, providing new insights into modeling spatio-temporal mobility in data silo scenarios. Extensive experiments conducted on real-world datasets demonstrate that FedCroST outperforms the advanced baselines by a large margin in diverse cross-silo spatio-temporal mobility modeling tasks.
Yudong Zhang 0005, Xu Wang 0029, Pengkun Wang 0001, Binwu Wang, Zhengyang Zhou, Yang Wang 0015
IEEE Trans. Mob. Comput.3
2023 Long-Tailed Time Series Classification via Feature Space Rebalancing
Pengkun Wang 0001, Xu Wang 0029, Binwu Wang, Yudong Zhang 0005, Lei Bai 0001, Yang Wang 0015
DASFAA (1)1
2023 A Knowledge-Driven Memory System for Traffic Flow Prediction
Binwu Wang, Yudong Zhang 0005, Pengkun Wang 0001, Xu Wang 0029, Lei Bai 0001, Yang Wang 0015
DASFAA (4)3
2023 Pondering About Task Spatial Misalignment: Classification-Localization Equilibrated Object Detection
abstract
Object detection is a fundamental task in computer vision, consisting of both classification and localization tasks. Previous works mostly perform classification and localization with shared feature extractor like Convolution Neural Network. However, the tasks of classification and localization exhibit different sensitivities with regard to the same feature, hence the "task spatial misalignment" issue. This issue can result in a hedge issue between the performances of localizer and classifier. To address these issues, we first propose a novel Dynamic Coefficient Loss to simultaneously consider and balance the performances of classification and localization tasks. To well address anchor label misjudgment issue in irregular- shaped object detection, we define a new classification-aware IoU metric to assign anchors intelligently. Finally, we further introduce the localization factor into NMS by proposing a Classification-Localization balanced NMS. Extensive experiments on MS COCO and PASCAL VOC demonstrate that our proposals can improve RetinaNet by around 1.5% AP with various backbones.
Yudong Zhang 0005, Xu Wang 0029, Pengkun Wang 0001, Yang Wang 0015
ICASSP4
2023 Searching Lottery Tickets in Graph Neural Networks: A Dual Perspective
Kun Wang 0056, Yuxuan Liang 0002, Pengkun Wang 0001, Xu Wang 0029, Pengfei Gu, Junfeng Fang, Yang Wang 0015
ICLR3
2023 Pattern Expansion and Consolidation on Evolving Graphs for Continual Traffic Prediction
abstract
Recently, spatiotemporal graph convolutional networks are becoming popular in the field of traffic flow prediction and significantly improve prediction accuracy. However, the majority of existing traffic flow prediction models are tailored to static traffic networks and fail to model the continuous evolution and expansion of traffic networks. In this work, we move to investigate the challenge of traffic flow prediction on an expanding traffic network. And we propose an efficient and effective continual learning framework to achieve continuous traffic flow prediction without the access to historical graph data, namely Pattern Expansion and Consolidation based on Pattern Matching based (PECPM). Specifically, we first design a pattern bank based on pattern matching to store representative patterns of the road network. With the expansion of the road network, the model configured with such a bank module can achieve continuous traffic prediction by effectively managing patterns stored in the bank. The core idea is to continuously update new patterns while consolidating learned ones. Specifically, we design a pattern expansion mechanism that can detect evolved and new patterns from the updated network, then these unknown patterns are expanded into the pattern bank to adapt to the updated road network. Additionally, we propose a pattern consolidation mechanism that includes both a bank preservation mechanism and a pattern traceability mechanism. This can effectively consolidate the learned patterns in the bank without requiring access to detailed historical graph data. We construct experiments on real-world traffic datasets to demonstrate the competitive performance, superior efficiency, and strong generalization ability of PECPM.
Binwu Wang, Yudong Zhang 0005, Xu Wang 0029, Pengkun Wang 0001, Zhengyang Zhou, Lei Bai 0001, Yang Wang 0015
KDD4
2023 An Observed Value Consistent Diffusion Model for Imputing Missing Values in Multivariate Time Series
abstract
Missing values, which are common in multivariate time series, is most important obstacle towards the utilization and interpretation of those data. Great efforts have been employed on how to accurately impute missing values in multivariate time series, and existing works either use deep learning networks to achieve deterministic imputations or aim at generating different plausible imputations by sampling multiple noises from a same distribution and then denoising them. However, these models either fall short of modeling the uncertainties of imputations due to their deterministic nature or perform poorly in terms of interpretability and imputation accuracy due to their ignorance of the correlations between the latent representations of both observed and missing values which are parts of samples from a same distribution. To this end, in this paper, we explicitly take the correlations between observed and missing values into account, and theoretically re-derive the Evidence Lower BOund (ELBO) of conditional diffusion model in the scenario of multivariate time series imputation. Based on the newly derived ELBO, we further propose a novel multivariate imputation diffusion model (MIDM) which is equipped with novel noise sampling, adding and denoising mechanisms for multivariate time series imputation, and the series of newly designed technologies jointly ensure the involving of the consistency between observed and missing values. Extensive experiments on both the tasks of multivariate time series imputation and forecasting witness the superiority of our proposed MIDM model on generating conditional estimations.
Xu Wang 0029, Pengkun Wang 0001, Yudong Zhang 0005, Binwu Wang, Zhengyang Zhou, Yang Wang 0015
KDD3
2023 EXTRACT and REFINE: Finding a Support Subgraph Set for Graph Representation
abstract
Subgraph learning has received considerable attention in its capacity of interpreting important structural information for predictions. Existing subgraph learning usually exploits statistics on predefined structures e.g., node degrees, occurrence frequency, to extract subgraphs, or refine the contents via only capturing label-relevant information with node-level sampling. Given diverse subgraph patterns, and mutual independence with local correlations on graphs, current solutions on subgraph learning still have two limitations in extraction and refinement stages. 1) The universality of extracting substructure patterns across domains is still lacking, 2) node-level sampling in refinement will distort the original local topology and none explicit guidance eliminating redundant information contribute to inefficiency issue. In this paper, we propose a unified subgraph learning scheme, Poly-Pivot Graph Neural Network (P2GNN) where we designate the centric node of each subgraph as the pivot. In the extraction stage, we present a general subgraph extraction principle, i.e., Local; Asymmetry between the centric and affiliated nodes. To this end, we asymmetrically model the similarity between each pair of nodes with random walk and quantify mutual affiliations in Affinity Propagation architecture, to extract subgraph structures. In the refinement, we devise a subgraph-level exclusion regularization to squash the target-independent information by considering mutual relations across subgraphs, cooperatively preserving a support set of subgraphs and facilitating the refinement process for graph representation. Empirical experiments on diverse web and biological graphs reveal 1.1%~7.3% improvements against best baselines, and visualized case studies prove the universality and interpretability of our P2GNN.
Kuo Yang 0002, Zhengyang Zhou, Pengkun Wang 0001, Xu Wang 0029, Yang Wang 0015
KDD4
2023 A Multi-graph Fusion Based Spatiotemporal Dynamic Learning Framework
abstract
Spatiotemporal data forecasting is a fundamental task in the field of graph data mining. Typical spatiotemporal data prediction methods usually capture spatial dependencies by directly aggregating features of local neighboring vertices in a fixed graph. However, this kind of aggregators can only capture localized correlations between vertices, and while been stacked for larger receptive field, they fall into the dilemma of over-smoothing. Additional, in temporal perspective, traditional methods focus on fixed graphs, while the correlations among vertexes can be dynamic. And time series components integrated strategies in traditional spatiotemporal learning methods can hardly handle frequently and drastically changed sequences. To overcome those limitations of existing works, in this paper, we propose a novel multi-graph based dynamic learning framework. First, a novel Dynamic Neighbor Search (DNS) mechanism is introduced to model global dynamic correlations between vertices by constructing a feature graph (FG), where the adjacency matrix is dynamically determined by DNS. Then we further alleviate the over-smoothing issue with our newly designed Adaptive Heterogeneous Representation (AHR) module. Both FG and origin graph (OG) are fed into the AHR modules and fused in our proposed Multi-graph Fusion block. Additionally, we design a Differential Vertex Representation (DVR) module which takes advantage of differential information to model temporal trends. Extensive experiments illustrate the superior forecasting performances of our proposed multi-graph based dynamic learning framework on six real-world spatiotemporal datasets from different cities and domains, and this corroborates the solid effectiveness of our proposed framework and its superior generalization ability.
Xu Wang 0029, Lianliang Chen, Pengkun Wang 0001, Zhengyang Zhou, Yang Wang 0015
WSDM4
2023 Knowledge Expansion and Consolidation for Continual Traffic Prediction With Expanding Graphs
abstract
Accurate traffic prediction plays a vital role in intelligent transport managements and applications. However, in the vast majority of existing works, the focus is mainly on modeling spatiotemporal correlations in static traffic networks. Thus, the continuous expansion and evolution of traffic networks are ignored. In this work, we study the problem of traffic prediction with expanding road network structures under the continual learning paradigm. Considering the model prediction performance, efficiency, and data accessibility, a SpatioTemporal Knowledge Expansion and Consolidation (STKEC) framework is proposed. This framework contains an influence-based knowledge expansion strategy to help the spatiotemporal learning model integrate new spatiotemporal traffic patterns and a memory-augmented knowledge consolidation mechanism to preserve the learned spatiotemporal patterns without accessing the data in previous graphs. Extensive experiments are conducted on a large-scale dataset and verify the superior performance of STKEC in continual traffic prediction.
Binwu Wang, Yudong Zhang 0005, Pengkun Wang 0001, Xu Wang 0029, Lei Bai 0001, Yang Wang 0015
IEEE Trans. Intell. Transp. Syst.4
2023 Joint Gated Co-Attention Based Multi-Modal Networks for Subregion House Price Prediction
abstract
Urban housing price is widely accepted as an economic indicator which is of both business and research interest in urban computing. However, due to the complex nature of influencing factors and the sparse property of transaction records, to implement such a model is still challenging. To address these challenges, in this work, we study an effective and fine-grained model for urban subregion housing price predictions. Compared to existing works, our proposal improves the forecasting granularity from city-level to mile-level, with only publicly released transaction data. We employ a feature selection mechanism to select more relevant features. Then, we propose an integrated model, JGC MMN (Joint Gated Co-attention Based Multi-modal Network), to learn all-level features and capture spatiotemporal correlations in all-time stages with a modified densely connected convolutional network as well as current ingredients and future expectations. Next, we devise a novel JGC based fusion method to better fuse the heterogeneous data of multi-stage models by considering their interactions in temporal dimension. Finally, extensive empirical studies on real datasets demonstrate the effectiveness of our proposal, and this fine-grained housing price forecasting has the potential to support a broad scope of applications, ranging from urban planning to housing market recommendations.
Pengkun Wang 0001, Chuancai Ge, Zhengyang Zhou, Xu Wang 0029, Yuantao Li, Yang Wang 0015
IEEE Trans. Knowl. Data Eng.1
2023 A2DJP: A Two Graph-Based Component Fused Learning Framework for Urban Anomaly Distribution and Duration Joint-Prediction
abstract
Modern intelligent transportation system (ITS) has greatly benefitted people's daily life. However, the chanciness and suddenness of urban anomalies may greatly restrict the trouble-free operations of ITS. To be aware of future urban anomalies and their possible influences, great efforts have been achieved on these two aspects, but comprehensive predictions of urban anomalies including the predictions of distributions and durations, are still beingless. And the spatiotemporal cascade self/mutual exciting influences among anomalies have never been considered in previous studies. In this paper, we propose a novel Anomaly Distribution and Duration Joint-Prediction (A2DJP) algorithm to simultaneously filtrate urban subregions and estimate the duration of corresponding potential anomalies in the future. To capture the spatiotemporal correlations between urban traffics and anomalies, we use a modified Graph Convolution Network and Long Short-Term Memory integrated network. To learn the cascade correlations among anomalies themselves, we devise a novel Spatiotemporal neural Hawkes Process model, which contains a Hawkes Process (HP) based GCN and HP-based LSTM to extract the anomaly-wise spatiotemporal cascading correlations. By fusing the spatiotemporal correlations between traffics and anomalies, we then simultaneously predict the distributions and durations of future anomalies. Extensive experiments on real-world datasets demonstrate that our proposed method significantly outperforms state-of-the-art solutions.
Kun Wang 0056, Zhengyang Zhou, Xu Wang 0029, Pengkun Wang 0001, Yang Wang 0015
IEEE Trans. Knowl. Data Eng.4
2022 Countering Modal Redundancy and Heterogeneity: A Self-Correcting Multimodal Fusion
abstract
Fusing multimodal heterogeneous data plays a vital role in recognition and prediction tasks in various fields, e.g., action recognition and traffic accident forecast. Yet, there remain some key challenges, such as heterogeneous feature interaction and feature redundancies, that significantly affect the performance of multimodal fusion. To tackle these challenges, we first devise a Unified Feature Interaction Module (UFIM) in which a novel orthogonal attention component is designed to obtain fine-grained inter-modal interaction information among heterogeneous features. Then, we propose a novel Self-Correcting Transformer Module (SCTM) which employs a modified transformer to obtain the one-to-many correlation information between the current modal feature and the merged features of other modalities to alleviate the redundancy problem. Extensive experiments on four cross-domain tasks demonstrate the effectiveness and generalization ability of our proposed method.
Pengkun Wang 0001, Xu Wang 0029, Binwu Wang, Yudong Zhang 0005, Lei Bai 0001, Yang Wang 0015
ICDM1