Explaining k-Nearest Neighbors: Abductive and Counterfactual Explanations
Pablo Barceló, Alexander Kozachinskiy, Miguel Romero Orth, Bernardo Subercaseaux, José Verschae
TL;DR
The paper investigates how to explain $k$-NN classifications from a feature-centric perspective using abductive (minimum sufficient reasons) and counterfactual explanations. It conducts a thorough complexity analysis across continuous $(\mathbb{R},D_p)$ and discrete $(\{0,1\}^n,D_H)$ settings, showing NP-hardness for minimum sufficient reasons in all settings and revealing a separation: tractable $\ell_2$-based tasks versus harder $\ell_1$-based and discrete $k\ge3$ tasks. It provides polynomial-time algorithms for several $\ell_2$-distance problems and proves NP-hardness for $\ell_1$-based counterfactuals, with a precise map of when explanations are tractable. The authors also demonstrate practical computation via Integer Quadratic Programming and SAT encodings, and validate the approach on MNIST and synthetic data, illustrating feasibility for hundreds of features. Overall, the work clarifies when feature-based explanations for $k$-NN are tractable and offers concrete algorithms for real-world explainability tasks.
Abstract
Despite the wide use of $k$-Nearest Neighbors as classification models, their explainability properties remain poorly understood from a theoretical perspective. While nearest neighbors classifiers offer interpretability from a "data perspective", in which the classification of an input vector $\bar{x}$ is explained by identifying the vectors $\bar{v}_1, \ldots, \bar{v}_k$ in the training set that determine the classification of $\bar{x}$, we argue that such explanations can be impractical in high-dimensional applications, where each vector has hundreds or thousands of features and it is not clear what their relative importance is. Hence, we focus on understanding nearest neighbor classifications through a "feature perspective", in which the goal is to identify how the values of the features in $\bar{x}$ affect its classification. Concretely, we study abductive explanations such as "minimum sufficient reasons", which correspond to sets of features in $\bar{x}$ that are enough to guarantee its classification, and "counterfactual explanations" based on the minimum distance feature changes one would have to perform in $\bar{x}$ to change its classification. We present a detailed landscape of positive and negative complexity results for counterfactual and abductive explanations, distinguishing between discrete and continuous feature spaces, and considering the impact of the choice of distance function involved. Finally, we show that despite some negative complexity results, Integer Quadratic Programming and SAT solving allow for computing explanations in practice.
