Data Pruning Can Do More: A Comprehensive Data Pruning Approach for Object Re-identification

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Yang, Zi, Yang, Haojin, Majumder, Soumajit, Cardoso, Jorge, Gallego, Guillermo
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915062924443648
author Yang, Zi
Yang, Haojin
Majumder, Soumajit
Cardoso, Jorge
Gallego, Guillermo
author_facet Yang, Zi
Yang, Haojin
Majumder, Soumajit
Cardoso, Jorge
Gallego, Guillermo
contents Previous studies have demonstrated that not each sample in a dataset is of equal importance during training. Data pruning aims to remove less important or informative samples while still achieving comparable results as training on the original (untruncated) dataset, thereby reducing storage and training costs. However, the majority of data pruning methods are applied to image classification tasks. To our knowledge, this work is the first to explore the feasibility of these pruning methods applied to object re-identification (ReID) tasks, while also presenting a more comprehensive data pruning approach. By fully leveraging the logit history during training, our approach offers a more accurate and comprehensive metric for quantifying sample importance, as well as correcting mislabeled samples and recognizing outliers. Furthermore, our approach is highly efficient, reducing the cost of importance score estimation by 10 times compared to existing methods. Our approach is a plug-and-play, architecture-agnostic framework that can eliminate/reduce 35%, 30%, and 5% of samples/training time on the VeRi, MSMT17 and Market1501 datasets, respectively, with negligible loss in accuracy (< 0.1%). The lists of important, mislabeled, and outlier samples from these ReID datasets are available at https://github.com/Zi-Y/data-pruning-reid.
format Preprint
id arxiv_https___arxiv_org_abs_2412_10091
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Data Pruning Can Do More: A Comprehensive Data Pruning Approach for Object Re-identification
Yang, Zi
Yang, Haojin
Majumder, Soumajit
Cardoso, Jorge
Gallego, Guillermo
Computer Vision and Pattern Recognition
Artificial Intelligence
Previous studies have demonstrated that not each sample in a dataset is of equal importance during training. Data pruning aims to remove less important or informative samples while still achieving comparable results as training on the original (untruncated) dataset, thereby reducing storage and training costs. However, the majority of data pruning methods are applied to image classification tasks. To our knowledge, this work is the first to explore the feasibility of these pruning methods applied to object re-identification (ReID) tasks, while also presenting a more comprehensive data pruning approach. By fully leveraging the logit history during training, our approach offers a more accurate and comprehensive metric for quantifying sample importance, as well as correcting mislabeled samples and recognizing outliers. Furthermore, our approach is highly efficient, reducing the cost of importance score estimation by 10 times compared to existing methods. Our approach is a plug-and-play, architecture-agnostic framework that can eliminate/reduce 35%, 30%, and 5% of samples/training time on the VeRi, MSMT17 and Market1501 datasets, respectively, with negligible loss in accuracy (< 0.1%). The lists of important, mislabeled, and outlier samples from these ReID datasets are available at https://github.com/Zi-Y/data-pruning-reid.
title Data Pruning Can Do More: A Comprehensive Data Pruning Approach for Object Re-identification
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2412.10091