Machine learning models are brittle, and small changes in the training data can result in different predictions. We study the problem of proving that a prediction is robust to \emph{data poisoning}, where an attacker can inject a number of malicious elements into the training set to influence the learned model. We target decision-tree models, a popular and simple class of machine learning models that underlies many complex learning techniques. We present a sound verification technique based on \emph{abstract interpretation} and implement it in a tool called Antidote. Antidote abstractly trains decision trees for an intractably large space of possible poisoned datasets. Due to the soundness of our abstraction, Antidote can produce proofs that, for a given input, the corresponding prediction would not have changed had the training set been tampered with or not. We demonstrate the effectiveness of Antidote on a number of popular datasets.
Thu 18 JunDisplayed time zone: Pacific Time (US & Canada) change
10:40 - 11:40 | Machine Learning IIPLDI Research Papers at PLDI Research Papers live stream Chair(s): Ke Wang Visa Research | ||
10:40 20mTalk | Proving Data-Poisoning Robustness in Decision Trees PLDI Research Papers Samuel Drews University of Wisconsin-Madison, USA, Aws Albarghouthi University of Wisconsin-Madison, USA, Loris D'Antoni University of Wisconsin-Madison, USA | ||
11:00 20mTalk | A Study of the Learnability of Relational Properties: Model Counting Meets Machine Learning (MCML) PLDI Research Papers Muhammad Usman University of Texas at Austin, USA, Wenxi Wang University of Texas at Austin, USA, Marko Vasic University of Texas at Austin, USA, Kaiyuan Wang Google, USA, Haris Vikalo University of Texas at Austin, USA, Sarfraz Khurshid University of Texas at Austin, USA | ||
11:20 20mTalk | Learning Fast and Precise Numerical Analysis PLDI Research Papers Jingxuan He ETH Zurich, Switzerland, Gagandeep Singh ETH Zurich, Switzerland, Markus PĆ¼schel ETH Zurich, Switzerland, Martin Vechev ETH Zurich, Switzerland |