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

Zhixuan Chu

dblp:258/1233 · DBLP profile ↗
← Back
39ranked-venue papers
10as first author
37since 2021 · last 2026
0000-0001-6075-1816ORCID · verified

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

Artificial intelligence and machine learning · 29 · 6 first-author · 27 since 2021Databases, data management, data science and information retrieval · 17 · 8 first-author · 15 since 2021Graphics, computer vision, multimedia, augmented reality and games · 7 · 1 first-author · 7 since 2021Security and privacy · 3 · 1 first-author · 3 since 2021Applied, interdisciplinary, general and emerging computing · 2 · 1 first-author · 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
17 papers
Trustworthy machine learning · 37% Language models and text generation · 19% Probabilistic and Bayesian machine learning · 12%
Network and information security
5 papers
Security and privacy of machine learning · 97% Digital forensics and information hiding · 3%
Databases, data mining, and information retrieval
5 papers
Data mining · 65% Recommender systems · 16% Knowledge graphs · 16%

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

TopicWeightPapersLastEvidence papers
Machine learning › Trustworthy machine learning
fairness
1.522025
Mitigating Social Bias in Large Language Models: A Multi-Objective Approach Within a Multi-Agent Framework · AAAI 2025
Fair Attribute Completion on Graph with Missing Attributes · ICLR 2023
Machine learning › Trustworthy machine learning
interpretability
1.522024
A Causal Explainable Guardrails for Large Language Models · CCS 2024
Task-Driven Causal Feature Distillation: Towards Trustworthy Risk Prediction · AAAI 2024
Machine learning › Graph learning
graph neural network
1.322024
Invariant Graph Learning for Causal Effect Estimation · WWW 2024
Graph Infomax Adversarial Learning for Treatment Effect Estimation with Networked Observational Data · KDD 2021
Machine learning › Probabilistic and Bayesian machine learning
causal inference
1.222024
Invariant Graph Learning for Causal Effect Estimation · WWW 2024
Causal Inference Meets Machine Learning · KDD 2020
Data mining
causal inference
1.222023
Continual Causal Inference with Incremental Observational Data · ICDE 2023
Graph Infomax Adversarial Learning for Treatment Effect Estimation with Networked Observational Data · KDD 2021
Natural language and speech › Language models and text generation › large language model training
continual pre-training
1.012026
Perplexity-Aware Data Scaling Law: Perplexity Landscapes Predict Performance for Continual Pre-training · ACL (1) 2026
Natural language and speech › Language models and text generation
model steering
1.012026
Why Steering Works: Toward a Unified View of Language Model Parameter Dynamics · ACL (1) 2026
Machine learning › Deep learning architectures and training
scaling laws
1.012026
Perplexity-Aware Data Scaling Law: Perplexity Landscapes Predict Performance for Continual Pre-training · ACL (1) 2026
Security and privacy of machine learning › model intellectual property protection
model watermarking
1.012026
SWAP: Towards Copyright Auditing of Soft Prompts via Sequential Watermarking · Int. J. Comput. Vis. 2026
Machine learning › Trustworthy machine learning › fairness › bias mitigation
large language model debiasing
0.912025
Mitigating Social Bias in Large Language Models: A Multi-Objective Approach Within a Multi-Agent Framework · AAAI 2025
Machine learning › Trustworthy machine learning › AI safety › safety alignment
overrefusal
0.912025
Understanding and Mitigating Overrefusal in LLMs from an Unveiling Perspective of Safety Decision Boundary · EMNLP 2025
Machine learning › Trustworthy machine learning › AI safety
safety alignment
0.912025
Understanding and Mitigating Overrefusal in LLMs from an Unveiling Perspective of Safety Decision Boundary · EMNLP 2025
Machine learning › Trustworthy machine learning › fairness › bias mitigation
social bias mitigation
0.912025
Mitigating Social Bias in Large Language Models: A Multi-Objective Approach Within a Multi-Agent Framework · AAAI 2025
Machine learning › Time series and sequential data
time series analysis
0.912025
TimeMixer++: A General Time Series Pattern Machine for Universal Predictive Analysis · ICLR 2025
Machine learning › Representation and self-supervised learning › representation learning › sequence representation learning
time series representation learning
0.912025
TimeMixer++: A General Time Series Pattern Machine for Universal Predictive Analysis · ICLR 2025
Security and privacy of machine learning › adversarial attack › physical adversarial attack
adversarial patch
0.912025
Toward Defending Adversarial Patch Attacks With Mask-Reconstruction-Assisted Adversarial Training · IEEE Trans. Inf. Forensics Secur. 2025
Security and privacy of machine learning
adversarial robustness
0.912025
Toward Defending Adversarial Patch Attacks With Mask-Reconstruction-Assisted Adversarial Training · IEEE Trans. Inf. Forensics Secur. 2025
Security and privacy of machine learning › adversarial robustness
adversarial training
0.912025
Toward Defending Adversarial Patch Attacks With Mask-Reconstruction-Assisted Adversarial Training · IEEE Trans. Inf. Forensics Secur. 2025
Security and privacy of machine learning › adversarial attack › backdoor attack
backdoor defense
0.912025
Probe before You Talk: Towards Black-box Defense against Backdoor Unalignment for Large Language Models · ICLR 2025
Security and privacy of machine learning › adversarial attack › backdoor attack › backdoor defense
backdoor detection
0.912025
Probe before You Talk: Towards Black-box Defense against Backdoor Unalignment for Large Language Models · ICLR 2025
Security and privacy of machine learning › adversarial attack
jailbreak attack
0.912025
Heuristic-Induced Multimodal Risk Distribution Jailbreak Attack for Multimodal Large Language Models · ICCV 2025
Security and privacy of machine learning › large language model safety
jailbreak defense
0.912025
Probe before You Talk: Towards Black-box Defense against Backdoor Unalignment for Large Language Models · ICLR 2025
Security and privacy of machine learning › large language model security
multimodal large language model security
0.912025
Heuristic-Induced Multimodal Risk Distribution Jailbreak Attack for Multimodal Large Language Models · ICCV 2025
Machine learning › Trustworthy machine learning › interpretability
causal analysis
0.812024
A Causal Explainable Guardrails for Large Language Models · CCS 2024
Machine learning › Probabilistic and Bayesian machine learning › causal inference
causal effect estimation
0.812024
Invariant Graph Learning for Causal Effect Estimation · WWW 2024
Machine learning › Trustworthy machine learning › fairness
fairness and robustness
0.812024
Task-Driven Causal Feature Distillation: Towards Trustworthy Risk Prediction · AAAI 2024
Machine learning › Trustworthy machine learning › out-of-distribution generalization
invariant learning
0.812024
Invariant Graph Learning for Causal Effect Estimation · WWW 2024
Natural language and speech › Language models and text generation
large language model
0.812024
Time-LLM: Time Series Forecasting by Reprogramming Large Language Models · ICLR 2024
Machine learning › Transfer learning and domain adaptation › parameter-efficient transfer learning
model reprogramming
0.812024
Time-LLM: Time Series Forecasting by Reprogramming Large Language Models · ICLR 2024
Natural language and speech › Language models and text generation › model steering
representation steering
0.812024
A Causal Explainable Guardrails for Large Language Models · CCS 2024

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

causal inference · 2.0adversarial prompting · 2.0adversarial learning · 1.8neural temporal point process · 1.5large language model · 1.5deep neural network · 1.2watermarking · 1.0scaling law analysis · 1.0mechanistic interpretability · 1.0activation steering · 1.0probe concatenate effect · 0.9output distribution sampling · 0.9multi-objective optimization · 0.9multi-agent framework · 0.9mask-reconstruction · 0.9heuristic-induced multimodal risk distribution · 0.9causal intervention · 0.9adversarial example generation · 0.9
YearPublicationVenuePosition
2026 Perplexity-Aware Data Scaling Law: Perplexity Landscapes Predict Performance for Continual Pre-training
abstract
Lei Liu, Hao Zhu, Xiaoyan Yang, Yue Shen, Zhixuan Chu, Jian Wang, Jinjie Gu, Kui Ren. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026.
Zhixuan Chu, Jian Wang 0108, Jinjie Gu, Kui Ren 0001
ACL (1)5
2026 Why Steering Works: Toward a Unified View of Language Model Parameter Dynamics
abstract
Ziwen Xu, Chenyan WU, Hengyu Sun, Haiwen Hong, Mengru Wang, Yunzhi Yao, Longtao Huang, Hui Xue, Shumin Deng, Zhixuan Chu, Huajun Chen, Ningyu Zhang. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026.
Ziwen Xu, Chenyan Wu, Hengyu Sun, Haiwen Hong, Yunzhi Yao, Longtao Huang, Hui Xue 0001, Shumin Deng, Zhixuan Chu, Huajun Chen, Ningyu Zhang 0001
ACL (1)10
2026 A Causal Perspective for Enhancing Jailbreak Attack and Defense
Licheng Pan, Yunsheng Lu, Jiexi Liu 0005, Jialing Tao, Haozhe Feng, Hui Xue 0001, Zhixuan Chu, Kui Ren 0001
NDSS7
2026 SWAP: Towards Copyright Auditing of Soft Prompts via Sequential Watermarking
Zhixuan Chu, Jiaheng Zhang, Yiming Li 0004, Dacheng Tao
Int. J. Comput. Vis.4
2025 Mitigating Social Bias in Large Language Models: A Multi-Objective Approach Within a Multi-Agent Framework
abstract
Natural language processing (NLP) has seen remarkable advancements with the development of large language models (LLMs). Despite these advancements, LLMs often produce socially biased outputs. Recent studies have mainly addressed this problem by prompting LLMs to behave ethically, but this approach results in unacceptable performance degradation. In this paper, we propose a multi-objective approach within a multi-agent framework (MOMA) to mitigate social bias in LLMs without significantly compromising their performance. The key idea of MOMA involves deploying multiple agents to perform causal interventions on bias-related contents of the input questions, breaking the shortcut connection between these contents and the corresponding answers. Unlike traditional debiasing techniques leading to performance degradation, MOMA substantially reduces bias while maintaining accuracy in downstream tasks. Our experiments conducted in two datasets and two models demonstrate that MOMA reduces bias scores by up to 87.7%, with only a marginal performance degradation of up to 6.8% in the BBQ dataset. Additionally, it significantly enhances the multi-objective metric icat in the StereoSet dataset by up to 58.1%.
Zhenjie Xu, Wenqing Chen, Xuanying Li, Zhixuan Chu, Kui Ren 0001, Zibin Zheng, Zhichao Lu
AAAI6
2025 Entity-Aware Pretraining of Large Language Models for Recommendation Systems
Zhixuan Chu
IEEE Big Data1
2025 Understanding and Mitigating Overrefusal in LLMs from an Unveiling Perspective of Safety Decision Boundary
abstract
Large language models (LLMs) have demonstrated remarkable capabilities across a wide range of tasks, yet they often refuse to answer legitimate queries-a phenomenon known as overrefusal.Overrefusal typically stems from over-conservative safety alignment, causing models to treat many reasonable prompts as potentially risky.To systematically understand this issue, we probe and leverage the models' safety decision boundaries to analyze and mitigate overrefusal.Our findings reveal that overrefusal is closely tied to misalignment at these boundary regions, where models struggle to distinguish subtle differences between benign and harmful content.Building on these insights, we present RASS, an automated framework for prompt generation and selection that strategically targets overrefusal prompts near the safety boundary.By harnessing steering vectors in the representation space, RASS efficiently identifies and curates boundary-aligned prompts, enabling more effective and targeted mitigation of overrefusal.This approach not only provides a more precise and interpretable view of model safety decisions but also seamlessly extends to multilingual scenarios.We have explored the safety decision boundaries of various LLMs and construct the MORBENCH evaluation set to facilitate robust assessment of model safety and helpfulness across multiple languages.
Licheng Pan, Yongqi Tong, Jun Zhou 0011, Zhixuan Chu
EMNLP6
2025 Heuristic-Induced Multimodal Risk Distribution Jailbreak Attack for Multimodal Large Language Models
Xiaojun Jia, Ranjie Duan, Xinfeng Li, Yihao Huang 0001, Xiaoshuang Jia, Zhixuan Chu, Wenqi Ren
ICCV7
2025 TimeMixer++: A General Time Series Pattern Machine for Universal Predictive Analysis
abstract
Time series analysis plays a critical role in numerous applications, supporting tasks such as forecasting, classification, anomaly detection, and imputation. In this work, we present the time series pattern machine (TSPM), a model designed to excel in a broad range of time series tasks through powerful representation and pattern extraction capabilities. Traditional time series models often struggle to capture universal patterns, limiting their effectiveness across diverse tasks. To address this, we define multiple scales in the time domain and various resolutions in the frequency domain, employing various mixing strategies to extract intricate, task-adaptive time series patterns. Specifically, we introduce TimeMixer++, a general-purpose TSPM that processes multi-scale time series using (1) multi-resolution time imaging (MRTI), (2) time image decomposition (TID), (3) multi-scale mixing (MCM), and (4) multi-resolution mixing (MRM) to extract comprehensive temporal patterns. MRTI transforms multi-scale time series into multi-resolution time images, capturing patterns across both temporal and frequency domains. TID leverages dual-axis attention to extract seasonal and trend patterns, while MCM hierarchically aggregates these patterns across scales. MRM adaptively integrates all representations across resolutions. TimeMixer++ achieves state-of-the-art performance across 8 time series analytical tasks, consistently surpassing both general-purpose and task-specific models. Our work marks a promising step toward the next generation of TSPMs, paving the way for further advancements in time series analysis.
Shiyu Wang 0001, Jiawei Li 0017, Xiaoming Shi 0001, Zhou Ye 0001, Baichuan Mo, Wenze Lin, Shengtong Ju, Zhixuan Chu, Ming Jin 0005
ICLR8
2025 Probe before You Talk: Towards Black-box Defense against Backdoor Unalignment for Large Language Models
abstract
Backdoor unalignment attacks against Large Language Models (LLMs) enable the stealthy compromise of safety alignment using a hidden trigger while evading normal safety auditing. These attacks pose significant threats to the applications of LLMs in the real-world Large Language Model as a Service (LLMaaS) setting, where the deployed model is a fully black-box system that can only interact through text. Furthermore, the sample-dependent nature of the attack target exacerbates the threat. Instead of outputting a fixed label, the backdoored LLM follows the semantics of any malicious command with the hidden trigger, significantly expanding the target space. In this paper, we introduce BEAT, a black-box defense that detects triggered samples during inference to deactivate the backdoor. It is motivated by an intriguing observation (dubbed the **probe concatenate effect**), where concatenated triggered samples significantly reduce the refusal rate of the backdoored LLM towards a malicious probe, while non-triggered samples have little effect. Specifically, BEAT identifies whether an input is triggered by measuring the degree of distortion in the output distribution of the probe before and after concatenation with the input. Our method addresses the challenges of sample-dependent targets from an opposite perspective. It captures the impact of the trigger on the refusal signal (which is sample-independent) instead of sample-specific successful attack behaviors. It overcomes black-box access limitations by using multiple sampling to approximate the output distribution. Extensive experiments are conducted on various backdoor attacks and LLMs (including the closed-source GPT-3.5-turbo), verifying the effectiveness and efficiency of our defense. Besides, we also preliminarily verify that BEAT can effectively defend against popular jailbreak attacks, as they can be regarded as "natural backdoors". Our source code is available at https://github.com/clearloveclearlove/BEAT.
Biao Yi, Tiansheng Huang, Sishuo Chen, Tong Li 0011, Zheli Liu, Zhixuan Chu, Yiming Li 0004
ICLR6
2025 Toward Defending Adversarial Patch Attacks With Mask-Reconstruction-Assisted Adversarial Training
abstract
Adversarial patch attacks pose a significant threat to deep learning models in real-world applications, such as autonomous driving, due to their physical feasibility and ease of deployment. Although several defenses exist, they often have limitations, such as requiring prior knowledge of adversarial patches, being tied to specific model architectures, or struggling to balance robustness with accuracy on clean inputs. In this work, we propose a novel Mask-Reconstruction-assisted Adversarial Training (MRAT) framework to enhance model robustness against adversarial patches while preserving accuracy. To achieve this, we first develop an adversarial example generation method that applies masks comprising randomly positioned square blocks with adversarial perturbations to clean images. This technique generates diverse adversarial examples for training, preventing the model from overfitting to a specific type of adversarial patch. To maintain accuracy on clean inputs while enhancing adversarial robustness, we incorporate a mask-reconstruction task into standard adversarial training paradigms. This task utilizes feature representations extracted from the backbone to reconstruct clean versions of adversarial examples. By jointly optimizing the reconstruction network and the classification model, the feature representations of adversarial examples retain essential features of their original images, leading to robust feature extraction and high classification accuracy. Extensive experimental results in both digital and physical domains demonstrate that MRAT significantly enhances the model’s defensive performance against adversarial patch attacks while maintaining high accuracy on clean inputs. Our code is available at: https://github.com/wanggroupAI/AI_Security/tree/main/Adversarial%20Example/MRAT.
Hongshan Yang, Zhibo Wang 0001, Peng Sun 0003, Zhixuan Chu, Feng Lin 0004
IEEE Trans. Inf. Forensics Secur.5
2024 Task-Driven Causal Feature Distillation: Towards Trustworthy Risk Prediction
abstract
Since artificial intelligence has seen tremendous recent successes in many areas, it has sparked great interest in its potential for trustworthy and interpretable risk prediction. However, most models lack causal reasoning and struggle with class imbalance, leading to poor precision and recall. To address this, we propose a Task-Driven Causal Feature Distillation model (TDCFD) to transform original feature values into causal feature attributions for the specific risk prediction task. The causal feature attribution helps describe how much contribution the value of this feature can make to the risk prediction result. After the causal feature distillation, a deep neural network is applied to produce trustworthy prediction results with causal interpretability and high precision/recall. We evaluate the performance of our TDCFD method on several synthetic and real datasets, and the results demonstrate its superiority over the state-of-the-art methods regarding precision, recall, interpretability, and causality.
Zhixuan Chu, Mengxuan Hu, Qing Cui, Sheng Li 0001
AAAI1
2024 LLMRG: Improving Recommendations through Large Language Model Reasoning Graphs
abstract
Recommendation systems aim to provide users with relevant suggestions, but often lack interpretability and fail to capture higher-level semantic relationships between user behaviors and profiles. In this paper, we propose a novel approach that leverages large language models (LLMs) to construct personalized reasoning graphs. These graphs link a user's profile and behavioral sequences through causal and logical inferences, representing the user's interests in an interpretable way. Our approach, LLM reasoning graphs (LLMRG), has four components: chained graph reasoning, divergent extension, self-verification and scoring, and knowledge base self-improvement. The resulting reasoning graph is encoded using graph neural networks, which serves as additional input to improve conventional recommender systems, without requiring extra user or item information. Our approach demonstrates how LLMs can enable more logical and interpretable recommender systems through personalized reasoning graphs. LLMRG allows recommendations to benefit from both engineered recommendation systems and LLM-derived reasoning graphs. We demonstrate the effectiveness of LLMRG on benchmarks and real-world scenarios in enhancing base recommendation models.
Yan Wang 0002, Zhixuan Chu, Xin Ouyang, Simeng Wang, Hongyan Hao, Jinjie Gu, Siqiao Xue, James Zhang, Qing Cui, Jun Zhou 0011, Sheng Li 0001
AAAI2
2024 A Causal Explainable Guardrails for Large Language Models
abstract
Large Language Models (LLMs) have shown impressive performance in natural language tasks, but their outputs can exhibit undesirable attributes or biases. Existing methods for steering LLMs toward desired attributes often assume unbiased representations and rely solely on steering prompts. However, the representations learned from pre-training can introduce semantic biases that influence the steering process, leading to suboptimal results. We propose LLMGuardrail, a novel framework that incorporates causal analysis and adversarial learning to obtain unbiased steering representations in LLMs. LLMGuardrail systematically identifies and blocks the confounding effects of biases, enabling the extraction of unbiased steering representations. Experiments demonstrate LLMGuardrail's effectiveness in steering LLMs toward desired attributes while mitigating biases. Our work contributes to developing safe and reliable LLMs that align with desired attributes.
Zhixuan Chu, Yan Wang 0002, Zhibo Wang 0001, Zhan Qin, Kui Ren 0001
CCS1
2024 Multiscale Representation Enhanced Temporal Flow Fusion Model for Long-Term Workload Forecasting
abstract
Accurate workload forecasting is critical for efficient resource management in cloud computing systems, enabling effective scheduling and autoscaling. Despite recent advances with transformer-based forecasting models, challenges remain due to the non-stationary, nonlinear characteristics of workload time series and the long-term dependencies. In particular, inconsistent performance between long-term history and near-term forecasts hinders long-range predictions. This paper proposes a novel framework leveraging self-supervised multiscale representation learning to capture both long-term and near-term workload patterns. The long-term history is encoded through multiscale representations while the near-term observations are modeled via temporal flow fusion. These representations of different scales are fused using an attention mechanism and characterized with normalizing flows to handle non-Gaussian/non-linear distributions of time series. Extensive experiments on 9 benchmarks demonstrate superiority over existing methods.
Shiyu Wang 0001, Zhixuan Chu, Yinbo Sun, Yu Liu 0071, Yuliang Guo, Huiyang Jian, Lintao Ma, Xingyu Lu 0004, Jun Zhou 0011
CIKM2
2024 Causal Interventional Prediction System for Robust and Explainable Effect Forecasting
abstract
Although the widespread use of AI systems in today's world is growing, many current AI systems are found vulnerable due to hidden bias and missing information, especially in the most commonly used forecasting system. In this work, we explore the robustness and explainability of AI-based forecasting systems. We provide an in-depth analysis of the underlying causality involved in the effect prediction task and further establish a causal graph based on treatment, adjustment variable, confounder, and outcome. Correspondingly, we design a causal interventional prediction system (CIPS) based on a variational autoencoder and fully conditional specification of multiple imputations. Extensive results demonstrate the superiority of our system over state-of-the-art methods and show remarkable versatility and extensibility in practice.
Zhixuan Chu, Guang Zeng 0001, Shiyu Wang 0001, Yiming Li 0004
CIKM1
2024 VMFTransformer: An Angle-Preserving and Auto-Scaling Machine for Multi-Horizon Probabilistic Forecasting
abstract
As deep learning develops, the major research methodologies of time series forecasting can be divided into two categories, i.e., iterative and direct methods. In the iterative methods, since a small amount of error is produced at each time step, the recursive structure can potentially lead to large error accumulations over longer forecasting horizons. Although the direct methods can avoid this puzzle involved in the iterative methods, they face abuse of conditional independence among time points. This impractical assumption can also lead to biased models. To solve these challenges, we propose a direct approach for multi-horizon probabilistic forecasting, which can effectively characterize the dependence across future horizons. Specifically, we consider the multi-horizon target as a random vector. The direction of the vector embodies the temporal dependence, and the length of the vector measures the overall scale across each horizon. Therefore, we respectively apply the von Mises-Fisher (VMF) distribution and the truncated normal distribution to characterize the target vector’s angle and magnitude in our model. Extensive results demonstrate the superiority of our framework over eight state-of-the-art methods.
Yunyi Zhou, Ruohan Gao, Xinping Zheng, Zhixuan Chu
ECAI5
2024 Enhancing Event Sequence Modeling with Contrastive Relational Inference
abstract
Neural temporal point processes(TPPs) have shown promise for modeling continuous-time event sequences. However, capturing the interactions between events is challenging yet critical for performing inference tasks like forecasting on event sequence data. Existing TPP models have focused on parameterizing the conditional distribution of future events but struggle to model event interactions. In this paper, we propose a novel approach that leverages Neural Relational Inference (NRI) to learn a relation graph that infers interactions while simultaneously learning the dynamics patterns from observational data. Our approach, the Contrastive Relational Inference-based Hawkes Process (CRIHP), reasons about event interactions under a variational inference framework. It utilizes intensity-based learning to search for prototype paths to contrast relationship constraints. Extensive experiments on three real-world datasets demonstrate the effectiveness of our model in capturing event interactions for event sequence modeling tasks.
Yan Wang 0002, Zhixuan Chu, Caigao Jiang, Hongyan Hao, Minjie Zhu, Xindong Cai, Qing Cui, James Y. Zhang, Siqiao Xue, Jun Zhou 0011
ICASSP2
2024 Time-LLM: Time Series Forecasting by Reprogramming Large Language Models
abstract
Time series forecasting holds significant importance in many real-world dynamic systems and has been extensively studied. Unlike natural language process (NLP) and computer vision (CV), where a single large model can tackle multiple tasks, models for time series forecasting are often specialized, necessitating distinct designs for different tasks and applications. While pre-trained foundation models have made impressive strides in NLP and CV, their development in time series domains has been constrained by data sparsity. Recent studies have revealed that large language models (LLMs) possess robust pattern recognition and reasoning abilities over complex sequences of tokens. However, the challenge remains in effectively aligning the modalities of time series data and natural language to leverage these capabilities. In this work, we present Time-LLM, a reprogramming framework to repurpose LLMs for general time series forecasting with the backbone language models kept intact. We begin by reprogramming the input time series with text prototypes before feeding it into the frozen LLM to align the two modalities. To augment the LLM's ability to reason with time series data, we propose Prompt-as-Prefix (PaP), which enriches the input context and directs the transformation of reprogrammed input patches. The transformed time series patches from the LLM are finally projected to obtain the forecasts. Our comprehensive evaluations demonstrate that \method is a powerful time series learner that outperforms state-of-the-art, specialized forecasting models. Moreover, Time-LLM excels in both few-shot and zero-shot learning scenarios. The code is made available at https://github.com/KimMeen/Time-LLM.
Ming Jin 0005, Shiyu Wang 0001, Lintao Ma, Zhixuan Chu, James Y. Zhang, Xiaoming Shi 0001, Yuxuan Liang 0002, Yuan-Fang Li, Shirui Pan, Qingsong Wen
ICLR4
2024 EasyTPP: Towards Open Benchmarking Temporal Point Processes
abstract
Continuous-time event sequences play a vital role in real-world domains such as healthcare, finance, online shopping, social networks, and so on. To model such data, temporal point processes (TPPs) have emerged as the most natural and competitive models, making a significant impact in both academic and application communities. Despite the emergence of many powerful models in recent years, there hasn't been a central benchmark for these models and future research endeavors. This lack of standardization impedes researchers and practitioners from comparing methods and reproducing results, potentially slowing down progress in this field. In this paper, we present EasyTPP, the first central repository of research assets (e.g., data, models, evaluation programs, documentations) in the area of event sequence modeling. Our EasyTPP makes several unique contributions to this area: a unified interface of using existing datasets and adding new datasets; a wide range of evaluation programs that are easy to use and extend as well as facilitate reproducible research; implementations of popular neural TPPs, together with a rich library of modules by composing which one could quickly build complex models. We will actively maintain this benchmark and welcome contributions from other researchers and practitioners. Our benchmark will help promote reproducible research in this field, thus accelerating research progress as well as making more significant real-world impacts. The code and data are available at \url{https://github.com/ant-research/EasyTemporalPointProcess}.
Siqiao Xue, Xiaoming Shi 0001, Zhixuan Chu, Yan Wang 0002, Hongyan Hao, Fan Zhou 0012, Caigao Jiang, James Y. Zhang, Qingsong Wen, Jun Zhou 0011, Hongyuan Mei
ICLR3
2024 Intelligent Agents with LLM-based Process Automation
abstract
While intelligent virtual assistants like Siri, Alexa, and Google Assistant have become ubiquitous in modern life, they still face limitations in their ability to follow multi-step instructions and accomplish complex goals articulated in natural language. However, recent breakthroughs in large language models (LLMs) show promise for overcoming existing barriers by enhancing natural language processing and reasoning capabilities. Though promising, applying LLMs to create more advanced virtual assistants still faces challenges like ensuring robust performance and handling variability in real-world user commands. This paper proposes a novel LLM-based virtual assistant that can automatically perform multi-step operations within mobile apps based on high-level user requests. The system represents an advance in assistants by providing an end-to-end solution for parsing instructions, reasoning about goals, and executing actions. LLM-based Process Automation (LLMPA) has modules for decomposing instructions, generating descriptions, detecting interface elements, predicting next actions, and error checking. Experiments demonstrate the system completing complex mobile operation tasks in Alipay based on natural language instructions. This showcases how large language models can enable automated assistants to accomplish real-world tasks. The main contributions are the novel LLMPA architecture optimized for app process automation, the methodology for applying LLMs to mobile apps, and demonstrations of multi-step task completion in a real-world environment. Notably, this work represents the first real-world deployment and extensive evaluation of a large language model-based virtual assistant in a widely used mobile application with an enormous user base numbering in the hundreds of millions.
Yanchu Guan, Dong Wang 0062, Zhixuan Chu, Shiyu Wang 0001, Feiyue Ni, Ruihua Song, Chenyi Zhuang
KDD3
2024 Invariant Graph Learning for Causal Effect Estimation
abstract
Causal effect estimation from networked observational data encounters notable challenges, primarily hidden confounders arising from network structure, or spillover effects that influence unit's outcomes based on neighboring treatment assignments. Existing graph neural network (GNN)-based methods have endeavored to address these challenges, utilizing the GNN's message-passing mechanism to capture hidden confounders or model spillover effects. However, they mainly focus on transductive causal effect learning on a single networked data, limiting their efficacy in inductive settings for real-world applications where networked data often originates from multiple environments influenced by potentially varying time or geographical regions. In light of this, we introduce the principle of invariance to the task of causal effect estimation on networked data, culminating in our Invariant Graph Learning (IGL) framework. Specifically, it first generates multiple networked data to simulate diverse environments from a given observational data. Then it further encourages the model to learn environment-invariant representations for confounders and spillover effects. Such a design enables the model to extrapolate beyond a single observed environment, thereby improving the performance of causal effect estimation in potential new environments. Extensive experiments on two real-world datasets demonstrates the superiority of our approach.
Yongduo Sui, Caizhi Tang, Zhixuan Chu, Junfeng Fang, Yuan Gao 0020, Qing Cui, Jun Zhou 0011, Xiang Wang 0010
WWW3
2024 A Survey of Trustworthy Representation Learning Across Domains
abstract
As AI systems have obtained significant performance to be deployed widely in our daily lives and human society, people both enjoy the benefits brought by these technologies and suffer many social issues induced by these systems. To make AI systems good enough and trustworthy, plenty of researches have been done to build guidelines for trustworthy AI systems. Machine learning is one of the most important parts of AI systems, and representation learning is the fundamental technology in machine learning. How to make representation learning trustworthy in real-world application, e.g., cross domain scenarios, is very valuable and necessary for both machine learning and AI system fields. Inspired by the concepts in trustworthy AI, we proposed the first trustworthy representation learning across domains framework, which includes four concepts, i.e., robustness, privacy, fairness, and explainability, to give a comprehensive literature review on this research direction. Specifically, we first introduce the details of the proposed trustworthy framework for representation learning across domains. Second, we provide basic notions and comprehensively summarize existing methods for the trustworthy framework from four concepts. Finally, we conclude this survey with insights and discussions on future research directions.
Ronghang Zhu, Dongliang Guo 0002, Daiqing Qi, Zhixuan Chu, Xiang Yu 0002, Sheng Li 0001
ACM Trans. Knowl. Discov. Data4
2023 Monotonic Neural Ordinary Differential Equation: Time-series Forecasting for Cumulative Data
abstract
Time-Series Forecasting based on Cumulative Data (TSFCD) is a crucial problem in decision-making across various industrial scenarios. However, existing time-series forecasting methods often overlook two important characteristics of cumulative data, namely monotonicity and irregularity, which limit their practical applicability. To address this limitation, we propose a principled approach called Monotonic neural Ordinary Differential Equation (MODE) within the framework of neural ordinary differential equations. By leveraging MODE, we are able to effectively capture and represent the monotonicity and irregularity in practical cumulative data. Through extensive experiments conducted in a bonus allocation scenario, we demonstrate that MODE outperforms state-of-the-art methods, showcasing its ability to handle both monotonicity and irregularity in cumulative data and delivering superior forecasting performance.
Zhichao Chen 0001, Leilei Ding, Zhixuan Chu, Yucheng Qi, Jianmin Huang, Hao Wang 0049
CIKM3
2023 Unsupervised Anomaly Detection & Diagnosis: A Stein Variational Gradient Descent Approach
abstract
Detecting and diagnosing anomalies in observational data plays a crucial role in various real-world applications, such as e-commerce applet maintenance. Unsupervised machine learning techniques are typically employed for anomaly detection and diagnosis due to their convenience and independence from labeled data. Density estimation (DE), as one of the most widely used unsupervised machine learning techniques for anomaly detection, can be categorized into kernel density estimation (KDE)-based methods and normalizing flow (NF)-based methods. While KDE-based methods offer fast computation speed, they often ignore the complex manifold structure present in observational data. On the other hand, NF-based methods address the manifold issue but suffer from longer computation times. In this study, we propose a novel DE-based anomaly detection & diagnosis method using Stein Variational Gradient Descent (SVGD), aiming to leverage the strengths of KDE and NF approaches. Firstly, we rigorously derive the DE capability of SVGD through mathematical analysis. Subsequently, we demonstrate the ability of the SVGD method to perform anomaly diagnosis based on input feature attribution. Finally, to validate the effectiveness of our approach, we conduct experiments using synthetic, benchmark, and industrial datasets. The results demonstrate the superior performance and practical applicability of our proposed method.
Zhichao Chen 0001, Leilei Ding, Jianmin Huang, Zhixuan Chu, Qingyang Dai, Hao Wang 0049
CIKM4
2023 Continual Learning in Predictive Autoscaling
abstract
Predictive Autoscaling is used to forecast the workloads of servers and prepare the resources in advance to ensure service level objectives (SLOs) in dynamic cloud environments. However, in practice, its prediction task often suffers from performance degradation under abnormal traffics caused by external events (such as sales promotional activities and applications' re-configurations), for which a common solution is to re-train the model with data of a long historical period, but at the expense of high computational and storage costs. To better address this problem, we propose a replay-based continual learning method, i.e., Density-based Memory Selection and Hint-based Network Learning Model (DMSHM), using only a small part of the historical log to achieve accurate predictions. First, we discover the phenomenon of sample overlap when applying replay-based continual learning in prediction tasks. In order to surmount this challenge and effectively integrate new sample distribution, we propose a density-based sample selection strategy that utilizes kernel density estimation to calculate sample density as a reference to compute sample weight, and employs weight sampling to construct a new memory set. Then we implement hint-based network learning based on hint representation to optimize the parameters. Finally, we conduct experiments on public and industrial datasets to demonstrate that our proposed method outperforms state-of-the-art continual learning methods in terms of memory capacity and prediction accuracy. Furthermore, we demonstrate remarkable practicability of DMSHM in real industrial applications.
Hongyan Hao, Zhixuan Chu, Shiyi Zhu, Gangwei Jiang, Yan Wang 0002, Caigao Jiang, James Y. Zhang, Siqiao Xue, Jun Zhou 0011
CIKM2
2023 Continual Causal Inference with Incremental Observational Data
abstract
The era of big data has witnessed an increasing availability of observational data from mobile and social networking, online advertising, web mining, healthcare, education, public policy, marketing campaigns, and so on, which facilitates the development of causal effect estimation. Although significant advances have been made to overcome the challenges in the academic area, such as missing counterfactual outcomes and selection bias, they only focus on source-specific and stationary observational data, which is unrealistic in most industrial applications. In this paper, we investigate a new industrial problem of causal effect estimation from incrementally available observational data and present three new evaluation criteria accordingly, including extensibility, adaptability, and accessibility. We propose a Continual Causal Effect Representation Learning method for estimating causal effects with observational data, which are incrementally available from non-stationary data distributions. Instead of having access to all seen observational data, our method only stores a limited subset of feature representations learned from previous data. Combining selective and balanced representation learning, feature representation distillation, and feature transformation, our method achieves the continual causal effect estimation for new data without compromising the estimation capability for original data. Extensive experiments demonstrate the significance of continual causal effect estimation and the effectiveness of our method.
Zhixuan Chu, Ruopeng Li, Stephen L. Rathbun, Sheng Li 0001
ICDE1
2023 Fair Attribute Completion on Graph with Missing Attributes
Dongliang Guo 0002, Zhixuan Chu, Sheng Li 0001
ICLR2
2023 pTSE: A Multi-model Ensemble Method for Probabilistic Time Series Forecasting
abstract
Various probabilistic time series forecasting models have sprung up and shown remarkably good performance. However, the choice of model highly relies on the characteristics of the input time series and the fixed distribution that model is based on. Due to the fact that the probability distributions cannot be averaged over different models straightforwardly, the current time series model ensemble methods cannot be directly applied to improve the robustness and accuracy of forecasting. To address this issue, we propose pTSE, a multi-model distribution ensemble method for probabilistic forecasting based on Hidden Markov Model (HMM). pTSE only takes off-the-shelf outputs from member models without requiring further information about each model. Besides, we provide a complete theoretical analysis of pTSE to prove that the empirical distribution of time series subject to an HMM will converge to the stationary distribution almost surely. Experiments on benchmarks show the superiority of pTSE over all member models and competitive ensemble methods.
Yunyi Zhou, Zhixuan Chu, Yijia Ruan, Sheng Li 0001
IJCAI2
2023 Prompt-augmented Temporal Point Process for Streaming Event Sequence
abstract
Neural Temporal Point Processes (TPPs) are the prevalent paradigm for modeling continuous-time event sequences, such as user activities on the web and financial transactions. In real world applications, the event data typically comes in a streaming manner, where the distribution of the patterns may shift over time. Under the privacy and memory constraints commonly seen in real scenarios, how to continuously monitor a TPP to learn the streaming event sequence is an important yet under-investigated problem. In this work, we approach this problem by adopting Continual Learning (CL), which aims to enable a model to continuously learn a sequence of tasks without catastrophic forgetting. While CL for event sequence is less well studied, we present a simple yet effective framework, PromptTPP, by integrating the base TPP with a continuous-time retrieval prompt pool. In our proposed framework, prompts are small learnable parameters, maintained in a memory space and jointly optimized with the base TPP so that the model is properly instructed to learn event streams arriving sequentially without buffering past examples or task-specific attributes. We formalize a novel and realistic experimental setup for modeling event streams, where PromptTPP consistently sets state-of-the-art performance across two real user behavior datasets.
Siqiao Xue, Yan Wang 0002, Zhixuan Chu, Xiaoming Shi 0001, Caigao Jiang, Hongyan Hao, Gangwei Jiang, Xiaoyun Feng, James Zhang, Jun Zhou 0011
NeurIPS3
2023 Estimating Propensity Scores with Deep Adaptive Variable Selection
abstract
The proliferation of observational data demands the development of statistical methods for causal inference. Many widely used causal inference methods are based on the propensity score. When estimating the propensity score, one essential question is which covariates should be included in the model. In this paper, we propose a deep adaptive variable selection based propensity score method (DAVSPS) by using representation learning and adaptive group LASSO. The key idea of DAFSPS is to combine the data-driven learning capability of representation learning and variable selection consistency of adaptive group LASSO to improve the estimation of the propensity score by selecting confounders and adjustment variables while removing instrumental and spurious variables. We also provide a detailed theoretical analysis to prove the variable selection consistency of DAVSPS. We evaluate the performance of our method on simulated data to demonstrate its superiority over state-of-the-art methods and apply it to real data.
Zhixuan Chu, Mechelle Claridy, José Cordero, Sheng Li 0001, Stephen L. Rathbun
SDM1
2022 Hierarchical Capsule Prediction Network for Marketing Campaigns Effect
abstract
Marketing campaigns are a set of strategic activities that can promote a business's goal. The effect prediction for marketing campaigns in a real industrial scenario is very complex and challenging due to the fact that prior knowledge is often learned from observation data, without any intervention for the marketing campaign. Furthermore, each subject is always under the interference of several marketing campaigns simultaneously. Therefore, we cannot easily parse and evaluate the effect of a single marketing campaign. To the best of our knowledge, there are currently no effective methodologies to solve such a problem, i.e., modeling an individual-level prediction task based on a hierarchical structure with multiple intertwined events. In this paper, we provide an in-depth analysis of the underlying parse tree-like structure involved in the effect prediction task and we further establish a Hierarchical Capsule Prediction Network (HapNet) for predicting the effects of marketing campaigns. Extensive results based on both the synthetic data and real data demonstrate the superiority of our model over the state-of-the-art methods and show remarkable practicability in real industrial applications.
Zhixuan Chu, Guang Zeng 0001, Tan Yan, Yulin Kang, Sheng Li 0001
CIKM1
2022 Incorporating Casual Analysis into Diversified and Logical Response Generation
Jiayi Liu 0004, Wei Wei 0002, Zhixuan Chu, Ji Zhang 0011, Tan Yan, Yulin Kang
COLING3
2022 Learning Infomax and Domain-Independent Representations for Causal Effect Inference with Real-World Data
abstract
The foremost challenge to causal inference with real-world data is to handle the imbalance in the covariates with respect to different treatment options, caused by treatment selection bias. To address this issue, recent literature has explored domain-invariant representation learning based on different domain divergence metrics (e.g., Wasserstein distance, maximum mean discrepancy, position-dependent metric, and domain overlap). In this paper, we reveal the weaknesses of these strategies, i.e., they lead to the loss of predictive information when enforcing the domain invariance; and the treatment effect estimation performance is unstable, which heavily relies on the characteristics of the domain distributions and the choice of domain divergence metrics. Motivated by information theory, we propose to learn the Infomax and Domain-Independent Representations to solve the above puzzles. Our method utilizes the mutual information between the global feature representations and individual feature representations, and the mutual information between feature representations and treatment assignment predictions, in order to maximally capture the common predictive information for both treatment and control groups. Moreover, our method filters out the influence of instrumental and irrelevant variables, and thus it effectively increases the predictive ability of potential outcomes. Experimental results on both the synthetic and real-world datasets show that our method achieves state-of-the-art performance on causal effect inference. Moreover, our method exhibits reliable prediction performances when facing data with different characteristics of data distributions, complicated variable types, and severe covariate imbalance.
Zhixuan Chu, Stephen L. Rathbun, Sheng Li 0001
SDM1
2022 Knowledge-Guided Article Embedding Refinement for Session-Based News Recommendation
abstract
Personalized news recommendation aims to recommend news articles to customers, by exploiting the personal preferences and short-term reading interest of users. A practical challenge in personalized news recommendations is the lack of logged user interactions. Recently, the session-based news recommendation has attracted increasing attention, which tries to recommend the next news article given previous articles in an active session. Current session-based news recommendation methods mainly extract latent embeddings from news articles and user-item interactions. However, many existing methods could not exploit the semantic-level structural information among news articles. And the feature learning process simply relies on the news articles in training data, which may not be sufficient to learn semantically rich embeddings. This brief presents a context-aware graph embedding (CAGE) approach for session-based news recommendation. It employs external knowledge graphs to improve the semantic-level representations of news articles. Moreover, graph neural networks are incorporated to further enhance the article embeddings. In addition, we consider the similarity among sessions and design attention neural networks to model the short-term user preferences. Extensive results on multiple news recommendation benchmark datasets show that CAGE performs better than some competitive baselines in most cases.
Heng-Shiou Sheu, Zhixuan Chu, Daiqing Qi, Sheng Li 0001
IEEE Trans. Neural Networks Learn. Syst.2
2021 Graph Infomax Adversarial Learning for Treatment Effect Estimation with Networked Observational Data
abstract
Treatment effect estimation from observational data is a critical research topic across many domains. The foremost challenge in treatment effect estimation is how to capture hidden confounders. Recently, the growing availability of networked observational data offers a new opportunity to deal with the issue of hidden confounders. Unlike networked data in traditional graph learning tasks, such as node classification and link detection, the networked data under the causal inference problem has its particularity, i.e., imbalanced network structure. In this paper, we propose a Graph Infomax Adversarial Learning (GIAL) model for treatment effect estimation, which makes full use of the network structure to capture more information by recognizing the imbalance in network structure. We evaluate the performance of our GIAL model on two benchmark datasets, and the results demonstrate superiority over the state-of-the-art methods.
Zhixuan Chu, Stephen L. Rathbun, Sheng Li 0001
KDD1
2021 A Survey on Causal Inference
abstract
Causal inference is a critical research topic across many domains, such as statistics, computer science, education, public policy, and economics, for decades. Nowadays, estimating causal effect from observational data has become an appealing research direction owing to the large amount of available data and low budget requirement, compared with randomized controlled trials. Embraced with the rapidly developed machine learning area, various causal effect estimation methods for observational data have sprung up. In this survey, we provide a comprehensive review of causal inference methods under the potential outcome framework, one of the well-known causal inference frameworks. The methods are divided into two categories depending on whether they require all three assumptions of the potential outcome framework or not. For each category, both the traditional statistical methods and the recent machine learning enhanced methods are discussed and compared. The plausible applications of these methods are also presented, including the applications in advertising, recommendation, medicine, and so on. Moreover, the commonly used benchmark datasets as well as the open-source codes are also summarized, which facilitate researchers and practitioners to explore, evaluate and apply the causal inference methods.
Liuyi Yao, Zhixuan Chu, Sheng Li 0001, Yaliang Li, Jing Gao 0004, Aidong Zhang 0001
ACM Trans. Knowl. Discov. Data2
2020 Matching in Selective and Balanced Representation Space for Treatment Effects Estimation
abstract
The dramatically growing availability of observational data is being witnessed in various domains of science and technology, which facilitates the study of causal inference. However, estimating treatment effects from observational data is faced with two major challenges, missing counterfactual outcomes and treatment selection bias. Matching methods are among the mostly widely used and fundamental approaches to estimating treatment effects, but existing matching methods have poor performance when facing data with high dimensional and complicated variables. We propose a feature selection representation matching (FSRM) method based on deep representation learning and matching, which maps the original covariate space into a selective, nonlinear, and balanced representation space, and then conducts matching in the learned representation space. FSRM adopts deep feature selection to minimize the influence of irrelevant variables for estimating treatment effects and incorporates a regularizer based on the Wasserstein distance to learn balanced representations. We evaluate the performance of our FSRM method on three datasets, and the results demonstrate superiority over the state-of-the-art methods.
Zhixuan Chu, Stephen L. Rathbun, Sheng Li 0001
CIKM1
2020 Causal Inference Meets Machine Learning
abstract
Causal inference has numerous real-world applications in many domains such as health care, marketing, political science and online advertising. Treatment effect estimation, a fundamental problem in causal inference, has been extensively studied in statistics for decades. However, traditional treatment effect estimation methods may not well handle large-scale and high-dimensional heterogeneous data. In recent years, an emerging research direction has attracted increasing attention in the broad artificial intelligence field, which combines the advantages of traditional treatment effect estimation approaches (e.g., matching estimators) and advanced representation learning approaches (e.g., deep neural networks). In this tutorial, we will introduce both traditional and state-of-the-art representation learning algorithms for treatment effect estimation. Background about causal inference, counterfactuals and matching estimators will be covered as well. We will also showcase promising applications of these methods in different application domains.
Peng Cui 0001, Zheyan Shen, Sheng Li 0001, Liuyi Yao, Yaliang Li, Zhixuan Chu, Jing Gao 0004
KDD6