Training-Time Attacks against K-nearest Neighbors

Authors

  • Ara Vartanian University Wisconsin--Madison
  • Will Rosenbaum Amherst College
  • Scott Alfeld Amherst College

DOI:

https://doi.org/10.1609/aaai.v37i8.26198

Keywords:

ML: Adversarial Learning & Robustness

Abstract

Nearest neighbor-based methods are commonly used for classification tasks and as subroutines of other data-analysis methods. An attacker with the capability of inserting their own data points into the training set can manipulate the inferred nearest neighbor structure. We distill this goal to the task of performing a training-set data insertion attack against k-Nearest Neighbor classification (kNN). We prove that computing an optimal training-time (a.k.a. poisoning) attack against kNN classification is NP-Hard, even when k = 1 and the attacker can insert only a single data point. We provide an anytime algorithm to perform such an attack, and a greedy algorithm for general k and attacker budget. We provide theoretical bounds and empirically demonstrate the effectiveness and practicality of our methods on synthetic and real-world datasets. Empirically, we find that kNN is vulnerable in practice and that dimensionality reduction is an effective defense. We conclude with a discussion of open problems illuminated by our analysis.

Downloads

Published

2023-06-26

How to Cite

Vartanian, A., Rosenbaum, W., & Alfeld, S. (2023). Training-Time Attacks against K-nearest Neighbors. Proceedings of the AAAI Conference on Artificial Intelligence, 37(8), 10053-10060. https://doi.org/10.1609/aaai.v37i8.26198

Issue

Section

AAAI Technical Track on Machine Learning III