A Theory-Based Evaluation of Nearest Neighbor Models Put Into Practice

TL;DR

This paper introduces a property testing algorithm for $k$-nearest neighbor graphs that efficiently determines their correctness and can identify inaccurate models faster than traditional methods.

Contribution

The paper develops the first randomized property tester for $k$-NN graphs with proven complexity bounds and empirical validation.

Findings

The tester has complexity $O(\sqrt{n} k^2 / \epsilon^2)$.

It can distinguish $k$-NN graphs from $\epsilon$-far graphs.

Empirical results show it detects inaccurate $k$-NN models faster than building the models.

Abstract

In the $k$-nearest neighborhood model ($k$-NN), we are given a set of points $P$, and we shall answer queries $q$ by returning the $k$ nearest neighbors of $q$ in $P$ according to some metric. This concept is crucial in many areas of data analysis and data processing, e.g., computer vision, document retrieval and machine learning. Many $k$-NN algorithms have been published and implemented, but often the relation between parameters and accuracy of the computed $k$-NN is not explicit. We study property testing of $k$-NN graphs in theory and evaluate it empirically: given a point set $P \subset \mathbb{R}^\delta$ and a directed graph $G=(P,E)$, is $G$ a $k$-NN graph, i.e., every point $p \in P$ has outgoing edges to its $k$ nearest neighbors, or is it $\epsilon$-far from being a $k$-NN graph? Here, $\epsilon$-far means that one has to change more than an $\epsilon$-fraction of the edges in order to make $G$ a $k$-NN graph. We develop a randomized algorithm with one-sided error that decides this question, i.e., a property tester for the $k$-NN property, with complexity $O(\sqrt{n} k^2 / \epsilon^2)$ measured in terms of the number of vertices and edges it inspects, and we prove a lower bound of $\Omega(\sqrt{n / \epsilon k})$. We evaluate our tester empirically on the $k$-NN models computed by various algorithms and show that it can be used to detect $k$-NN models with bad accuracy in significantly less time than the building time of the $k$-NN model.