Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

HyperINF: Unleashing the HyperPower of the Schulz's Method for Data Influence Estimation

About

Influence functions provide a principled method to assess the contribution of individual training samples to a specific target. Yet, their high computational costs limit their applications on large-scale models and datasets. Existing methods proposed for influence function approximation have significantly reduced the computational overheads. However, they mostly suffer from inaccurate estimation due to the lack of strong convergence guarantees from the algorithm. The family of hyperpower methods are well-known for their rigorous convergence guarantees on matrix inverse approximation, while the matrix multiplication operation can involve intractable memory and computation costs on large-scale models. We propose HyperINF, an efficient and accurate influence function approximation method which leverages the hyperpower method, specifically Schulz's iterative algorithm. To deal with the computation-intensive matrix multiplication, we incorporate the generalized fisher information (GFIM) as a low-rank approximation of the Hessian matrix, which reduces the memory and computation overheads to constant costs independent of ranks on LoRA-tuned models. We first demonstrate the superior accuracy and stability of HyperINF compared to other baselines through a synthetic convergence simulation for matrix inversion. We further validate the efficacy of HyperINF through extensive real-world data attribution tasks, including mislabeled data detection and data selection for LLM and VLM fine-tuning. On LoRA-tuned models, HyperINF achieves superior downstream performance with minimal memory and computational overhead, while other baselines suffer from significant degradation. Our codebase is available at https://github.com/Blackzxy/HyperINF.

Xinyu Zhou, Simin Fan, Martin Jaggi• 2024

Related benchmarks

TaskDatasetResultRank
Medical Visual Question AnsweringPMC-Reasoning
MMMU56.47
10
Influential data identificationSentence transformations
AUC100
10
Mislabeled Data DetectionMislabeled Data Detection VLM
AUC100
10
Image-to-text subject generationImage-to-text subject generation (VLM)
AUC99.3
10
Influential data identificationMath Problem w/o reasoning
AUC99.5
10
Influential data identificationMath Problem w/ reasoning
AUC99.4
10
Image-to-text style generationImage-to-text style generation (VLM)
AUC51.6
10
Medical Question AnsweringMedQA, MedMCQA, PubMedQA, MMLU-Pro-med, GPQA-med held-out (test)
Accuracy (MedQA)56.94
9
Mathematical ReasoningGSM8K
Accuracy42.8
9
High-quality Data DetectionNoise-Huatuo-Complex-CoT
Detection Accuracy15.1
4
Showing 10 of 10 rows

Other info

Follow for update