Leveraging Data Symmetries to Select an Optimal Subset of Training Data under Label Noise

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Shubham, Kumar, Karjol, Pavan, K, Kiran M, AP, Prathosh
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866909011446595584
author Shubham, Kumar
Karjol, Pavan
K, Kiran M
AP, Prathosh
author_facet Shubham, Kumar
Karjol, Pavan
K, Kiran M
AP, Prathosh
contents The performance of machine learning models often relies on large labeled datasets; however, data collected from diverse sources can contain label noise. Recent work has shown that, in noisy settings, there may exist a subset of the training data on which models can achieve performance comparable to training on a noise-free dataset. A widely used method for identifying such subsets is cutstats, which employs k-nearest neighbors (k-NN) to detect low-noise samples. However, its performance on high-dimensional data remains largely unexplored. In this work, we formally establish that the performance of a classifier trained on a subset of a noisy dataset selected via cutstats is influenced by the accuracy of k-NN. We further demonstrate that, in noisy environments, exploiting data invariance and knowledge of underlying symmetries can significantly enhance the performance of k-NN, bringing it closer to the Bayes optimal classifier even in high-dimensional regimes. Finally, we show that for real-world scenarios, where information about the underlying invariance is only partially known, learnt invariant representations can still facilitate the identification of near-optimal subsets.
format Preprint
id arxiv_https___arxiv_org_abs_2605_01874
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Leveraging Data Symmetries to Select an Optimal Subset of Training Data under Label Noise
Shubham, Kumar
Karjol, Pavan
K, Kiran M
AP, Prathosh
Machine Learning
Artificial Intelligence
The performance of machine learning models often relies on large labeled datasets; however, data collected from diverse sources can contain label noise. Recent work has shown that, in noisy settings, there may exist a subset of the training data on which models can achieve performance comparable to training on a noise-free dataset. A widely used method for identifying such subsets is cutstats, which employs k-nearest neighbors (k-NN) to detect low-noise samples. However, its performance on high-dimensional data remains largely unexplored. In this work, we formally establish that the performance of a classifier trained on a subset of a noisy dataset selected via cutstats is influenced by the accuracy of k-NN. We further demonstrate that, in noisy environments, exploiting data invariance and knowledge of underlying symmetries can significantly enhance the performance of k-NN, bringing it closer to the Bayes optimal classifier even in high-dimensional regimes. Finally, we show that for real-world scenarios, where information about the underlying invariance is only partially known, learnt invariant representations can still facilitate the identification of near-optimal subsets.
title Leveraging Data Symmetries to Select an Optimal Subset of Training Data under Label Noise
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2605.01874