Abstract | ||
---|---|---|
Finding valuable training data points for deep neural networks has been a core research challenge with many applications. In recent years, various techniques for calculating the "value" of individual training datapoints have been proposed for explaining trained models. However, the value of a training datapoint also depends on other selected training datapoints - a notion which is not explicitly captured by existing methods. In this paper, we study the problem of selecting high-value subsets of training data. The key idea is to design a learnable framework for online subset selection, which can be learned using mini-batches of training data, thus making our method scalable. This results in a parameterised convex subset selection problem that is amenable to a differentiable convex programming paradigm, thus allowing us to learn the parameters of the selection model in an end-to-end training. Using this framework, we design an online alternating minimization based algorithm for jointly learning the parameters of the selection model and ML model. Extensive evaluation on a synthetic dataset, and three standard datasets, show that our algorithm finds consistently higher value subsets of training data, compared to the recent state of the art methods, sometimes similar to 20% higher value than existing methods. The subsets are also useful in finding mislabelled training data. Our algorithm takes running time comparable to the existing valuation functions. |
Year | DOI | Venue |
---|---|---|
2021 | 10.1007/978-3-030-86520-7_41 | MACHINE LEARNING AND KNOWLEDGE DISCOVERY IN DATABASES, ECML PKDD 2021: RESEARCH TRACK, PT II |
Keywords | DocType | Volume |
Data valuation, Subset selection, Convex optimisation, Explainability | Conference | 12976 |
ISSN | Citations | PageRank |
0302-9743 | 0 | 0.34 |
References | Authors | |
0 | 5 |
Name | Order | Citations | PageRank |
---|---|---|---|
Soumi Das | 1 | 0 | 1.01 |
Arshdeep Singh | 2 | 0 | 2.03 |
Saptarshi Chatterjee | 3 | 0 | 0.34 |
Suparna Bhattacharya | 4 | 92 | 7.89 |
Sourangshu Bhattacharya | 5 | 94 | 14.00 |