arXiv · 2412.04158
LossVal: Efficient Data Valuation for Neural Networks
Abstract
Assessing the importance of individual training samples is a key challenge in machine learning. Traditional approaches retrain models with and without specific samples, which is computationally expensive and ignores dependencies between data points. We introduce LossVal, an efficient data valuation method that computes importance scores during neural network training by embedding a self-weighting mechanism into loss functions like cross-entropy and mean squared error. LossVal reduces computational costs, making it suitable for large datasets and practical applications. Experiments on classification and regression tasks across multiple datasets show that LossVal effectively identifies noisy samples and is able to distinguish helpful from harmful samples. We examine the gradient calculation of LossVal to highlight its advantages. The source code is available at: https://github.com/twibiral/LossVal
Explore related subjects
Keep this discovery
Tim Wibiral, Mohamed Karim Belaid, Maximilian Rabus, Ansgar Scherp. 2024-12-05. LossVal: Efficient Data Valuation for Neural Networks. https://arxiv.org/abs/2412.04158
Cite the original work for its findings. Save a collection to share your selection of sources.