atlas

Research map / Large language models / Instruction and fine-tuning

DataInf: Efficiently Estimating Data Influence in LoRA-tuned LLMs and Diffusion Models

ICLR 2024 · Large language models › Instruction and fine-tuning · approach: Fine-tuning for reasoning data · 5 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes DataInf, a closed-form influence approximation (Sherman-Morrison plus swapping inverse and average) that is fast and memory-efficient for LoRA-tuned LLMs and diffusion models.

Problem

Computing influence functions for training data is computationally and memory expensive for large models such as LLMs and text-to-image diffusion models, especially due to inverse Hessian computation.

Approach

Swap the order of matrix inversion and averaging in the damped Gauss-Newton (gradient second-moment) matrix, then apply Sherman-Morrison to obtain a closed-form inverse for each layer; the resulting estimator costs O(nd) with O(d) memory and is argued to be accurate when LoRA parameter counts are small.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use DataInf instead of Hessian-free for influence estimationLlama-2-13B-chat · Sentence transformationsclass detection AUC: 0.999 ± 0.002 -> 1.000 ± 0.000Helped
Use DataInf instead of Hessian-free for influence estimationLlama-2-13B-chat · Math problems (no reasoning)class detection AUC: 0.770 ± 0.174 -> 1.000 ± 0.000Helped
Use DataInf instead of Hessian-free for influence estimationLlama-2-13B-chat · Math problems (with reasoning)class detection AUC: 0.772 ± 0.173 -> 1.000 ± 0.001Helped
Use DataInf instead of Hessian-free for influence estimationstable-diffusion-v1.5 · Text-to-image style generationclass detection AUC: 0.692 ± 0.007 -> 0.820 ± 0.005Helped
Use DataInf instead of Hessian-free for influence estimationstable-diffusion-v1.5 · Text-to-image subject generationclass detection AUC: 0.820 ± 0.000 -> 0.865 ± 0.000Helped
Use DataInf instead of Hessian-free for influence estimationLlama-2-13B-chat · Sentence transformationsclass detection recall: 0.985 ± 0.033 -> 0.997 ± 0.010Helped
Use DataInf instead of Hessian-free for influence estimationLlama-2-13B-chat · Math problems (no reasoning)class detection recall: 0.258 ± 0.388 -> 0.999 ± 0.006Helped
Use DataInf instead of Hessian-free for influence estimationLlama-2-13B-chat · Math problems (with reasoning)class detection recall: 0.258 ± 0.388 -> 0.996 ± 0.025Helped
Use DataInf instead of Hessian-free for influence estimationstable-diffusion-v1.5 · Text-to-image style generationclass detection recall: 0.533 ± 0.008 -> 0.687 ± 0.006Helped
Use DataInf instead of Hessian-free for influence estimationstable-diffusion-v1.5 · Text-to-image subject generationclass detection recall: 0.210 ± 0.003 -> 0.315 ± 0.003Helped
LoRA fine-tuningLlama-2-13B-chat · Sentence transformationsaccuracy: 0.01 -> 0.35Helped
Use DataInf instead of LiSSA for influence computationRoBERTa-large · GLUE-QQP (4500 influence values)runtime: 70 seconds -> 13 secondsHelped

Surprising findings

Limitations

Experimental setup

Datasets: GLUE-MRPC GLUE-SST2 GLUE-WNLI GLUE-QQP Sentence transformations (synthetic) Math word problems (synthetic) cartoon image-text pairs pixel-art image-text pairs line sketch image-text pairs DreamBooth

Benchmarks: GLUE-MRPC GLUE-SST2 GLUE-WNLI GLUE-QQP DreamBooth Sentence transformations Math word problems

Metrics: Pearson correlation with exact influence AUC for mislabeled detection runtime class detection AUC class detection recall FID

Baselines: Exact influence LiSSA Hessian-free EK-FAC

Builds on: influence functions LiSSA EK-FAC LoRA

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy