Error Discovery by Clustering Influence Embeddings
About
We present a method for identifying groups of test examples -- slices -- on which a model under-performs, a task now known as slice discovery. We formalize coherence -- a requirement that erroneous predictions, within a slice, should be wrong for the same reason -- as a key property that any slice discovery method should satisfy. We then use influence functions to derive a new slice discovery method, InfEmbed, which satisfies coherence by returning slices whose examples are influenced similarly by the training data. InfEmbed is simple, and consists of applying K-Means clustering to a novel representation we deem influence embeddings. We show InfEmbed outperforms current state-of-the-art methods on 2 benchmarks, and is effective for model debugging across several case studies.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Classification | Covertype | Accuracy75.03 | 52 | |
| LLM Alignment | UltraFeedback | Win Rate25.82 | 24 | |
| Tabular Classification | Bank | Accuracy0.8389 | 14 | |
| Recommendation | Yelp | AUC77.44 | 9 | |
| Recommendation | AMAZON | AUC74.03 | 9 | |
| Recommendation | MovieLens | AUC76.25 | 9 | |
| Tabular Data Prediction | Adult | Accuracy78.53 | 9 | |
| Tabular Data Prediction | AIR | MSE0.15 | 9 | |
| Regression | BIKE | -- | 9 | |
| Tabular Classification | Credit | -- | 9 |