vix.ing · top · new · best · stats · spec

Outlier Gradient Analysis: Efficiently Identifying Detrimental Training Samples for Deep Learning Models

2024/05/06 by Anshuman Chhabra, Chhabra, Anshuman, Bo Li +7 · 4 citations
Computer Science · Engineering · Mathematics · #Advanced Statistical Methods and Models #Anomaly Detection Techniques and Applications #Artificial Intelligence (cs.AI) #FOS: Computer and information sciences #Fault Detection and Control Systems #Machine Learning (cs.LG)

paper · pdf · doi:10.48550/arxiv.2405.03869

openalex publication_date 2024/05/06 · openalex created_date 2024/05/11 · openalex updated_date 2026/07/28

Abstract

A core data-centric learning challenge is the identification of training samples that are detrimental to model performance. Influence functions serve as a prominent tool for this task and offer a robust framework for assessing training data influence on model predictions. Despite their widespread use, their high computational cost associated with calculating the inverse of the Hessian matrix pose constraints, particularly when analyzing large-sized deep models. In this paper, we establish a bridge between identifying detrimental training samples via influence functions and outlier gradient detection. This transformation not only presents a straightforward and Hessian-free formulation but also provides insights into the role of the gradient in sample impact. Through systematic empirical evaluations, we first validate the hypothesis of our proposed outlier gradient analysis approach on synthetic datasets. We then demonstrate its effectiveness in detecting mislabeled samples in vision models and selecting data samples for improving performance of natural language processing transformer models. We also extend its use to influential sample identification for fine-tuning Large Language Models.

Cited by

Related