← Search

Xiaojin Zhu

27 accepted papers

2024

Exact Policy Recovery in Offline RL with Both Heavy-Tailed Rewards and Data Corruption

AAAI 2024technical

We study offline reinforcement learning (RL) with heavy-tailed reward distribution and data corruption: (i) Moving beyond subGaussian reward distribution, we allow the rewards to have infinite variances; (ii) We allow corruptions where an attacker can arbitrarily modify a small fraction of the rewar…

Cited by 2SourcePDFScholar
2024

Optimal Attack and Defense for Reinforcement Learning

AAAI 2024technical

To ensure the usefulness of Reinforcement Learning (RL) in real systems, it is crucial to ensure they are robust to noise and adversarial attacks. In adversarial RL, an external attacker has the power to manipulate the victim agent's interaction with the environment. We study the full class of onlin…

2023

Byzantine-Robust Online and Offline Distributed Reinforcement Learning

AISTATS 2023poster

We consider a distributed reinforcement learning setting where multiple agents separately explore the environment and communicate their experiences through a central server. However, $\alpha$-fraction of agents are adversarial and can report arbitrary fake information. Critically, these adversarial…

Cited by 23SourcePDFScholar
2023

Reward Poisoning Attacks on Offline Multi-Agent Reinforcement Learning

AAAI 2023technical

In offline multi-agent reinforcement learning (MARL), agents estimate policies from a given dataset. We study reward-poisoning attacks in this setting where an exogenous attacker modifies the rewards in the dataset before the agents see the dataset. The attacker wants to guide each agent into a nefa…

Cited by 29SourcePDFScholar
2022

Out-of-Distribution Detection with Deep Nearest Neighbors

ICML 2022spotlight

Out-of-distribution (OOD) detection is a critical task for deploying machine learning models in the open world. Distance-based methods have demonstrated promise, where testing samples are detected as OOD if they are relatively far away from in-distribution (ID) data. However, prior methods impose a…

2021

Robust Policy Gradient against Strong Data Corruption

ICML 2021spotlight

We study the problem of robust reinforcement learning under adversarial corruption on both rewards and transitions. Our attack model assumes an \textit{adaptive} adversary who can arbitrarily corrupt the reward and transition at every step within an episode, for at most $\epsilon$-fraction of the le…

2021

Sequential Attacks on Kalman Filter-based Forward Collision Warning Systems

AAAI 2021technical

Kalman Filter (KF) is widely used in various domains to perform sequential learning or variable estimation. In the context of autonomous vehicles, KF constitutes the core component of many Advanced Driver Assistance Systems (ADAS), such as Forward Collision Warning (FCW). It tracks the states (dista…

Cited by 17SourcePDFScholar
2021

The Sample Complexity of Teaching by Reinforcement on Q-Learning

AAAI 2021technical

We study the sample complexity of teaching, termed as ``teaching dimension" (TDim) in the literature, for the teaching-by-reinforcement paradigm, where the teacher guides the student through rewards. This is distinct from the teaching-by-demonstration paradigm motivated by robotics applications, whe…

Cited by 11SourcePDFScholar
2020

Adaptive Reward-Poisoning Attacks against Reinforcement Learning

ICML 2020poster

In reward-poisoning attacks against reinforcement learning (RL), an attacker can perturb the environment reward $r_t$ into $r_t+\delta_t$ at each step, with the goal of forcing the RL agent to learn a nefarious policy. We categorize such attacks by the infinity-norm constraint on $\delta_t$: We prov…

Cited by 160SourcePDFScholar
2020

Policy Teaching via Environment Poisoning: Training-time Adversarial Attacks against Reinforcement Learning

ICML 2020poster

We study a security threat to reinforcement learning where an attacker poisons the learning environment to force the agent into executing a target policy chosen by the attacker. As a victim, we consider RL agents whose objective is to find a policy that maximizes average reward in undiscounted infin…

2019

A Unified Framework for Data Poisoning Attack to Graph-based Semi-supervised Learning

NeurIPS 2019poster

In this paper, we proposed a general framework for data poisoning attacks to graph-based semi-supervised learning (G-SSL). In this framework, we first unify different tasks, goals and constraints into a single formula for data poisoning attack in G-SSL, then we propose two specialized algorithms to…

Cited by 112SourcePDFScholar
2019

An Optimal Control Approach to Sequential Machine Teaching

AISTATS 2019poster

Given a sequential learning algorithm and a target model, sequential machine teaching aims to find the shortest training sequence to drive the learning algorithm to the target model. We present the first principled way to find such shortest training sequences. Our key insight is to formulate sequent…

2019

Policy Poisoning in Batch Reinforcement Learning and Control

NeurIPS 2019poster

We study a security threat to batch reinforcement learning and control where the attacker aims to poison the learned policy. The victim is a reinforcement learner / controller which first estimates the dynamics and the rewards from a batch data set, and then solves for the optimal policy with respec…

2019

Preference-Based Batch and Sequential Teaching: Towards a Unified View of Models

NeurIPS 2019poster

Algorithmic machine teaching studies the interaction between a teacher and a learner where the teacher selects labeled examples aiming at teaching a target hypothesis. In a quest to lower teaching complexity and to achieve more natural teacher-learner interactions, several teaching models and comple…

Cited by 39SourcePDFScholar
2018

Teacher Improves Learning by Selecting a Training Subset

AISTATS 2018poster

We call a learner super-teachable if a teacher can trim down an iid training set while making the learner learn even better. We provide sharp super-teaching guarantees on two learners: the maximum likelihood estimator for the mean of a Gaussian, and the large margin classifier in 1D. For general lea…

Cited by 0SourcePDFScholar
2016

Top Arm Identification in Multi-Armed Bandits with Batch Arm Pulls

AISTATS 2016poster

We introduce a new multi-armed bandit (MAB) problem in which arms must be sampled in batches, rather than one at a time. This is motivated by applications in social media monitoring and biological experimentation where such batch constraints naturally arise. This paper develops and analyzes algorith…

Cited by 99SourcePDFScholar
2015

Human Memory Search as Initial-Visit Emitting Random Walk

NeurIPS 2015poster

Imagine a random walk that outputs a state only when visiting it for the first time. The observed output is therefore a repeat-censored version of the underlying walk, and consists of a permutation of the states or a prefix of it. We call this model initial-visit emitting random walk (INVITE). Prior…

Cited by 17SourcePDFScholar