Clustered Influence Functions
Abstract
Lay Summary
Training modern AI systems requires vast amounts of data, and a natural question is: which training examples actually matter for a given prediction? Influence functions answer this by estimating how much each training point shifts the model's output, but computing them is expensive, especially when you need answers for thousands of different subsets of the data at once, as in cross-validation or data debugging workflows. We introduce Clustered Influence Functions (CiF), which turns this repeated computation into a one-time investment. Instead of solving a costly linear system for every new query, CiF groups the training data by gradient similarity, solves the system once per group, and stores the results in a cache. Any future query is then answered in milliseconds by combining cached responses, no new solves required. This makes influence-based data analysis such as auditing training sets, detecting poisoned examples, and running large-scale cross-validation more practical at larger scales.