提出新方法精准计算数据对模型的影响,无需假设模型凸性。
Understanding Data Influence with Differential Approximation
- 通过累积训练步间影响差异,实现高精度数据影响估计。
- 在多个任务中优于现有方法,大规模视觉语言预训练可处理百万级数据。
- 计算效率接近一阶方法,适合实际应用与大规模数据场景。
数据在人工智能的突破性进展中起关键作用,量化分析数据能显著提升模型训练效率与质量。然而,现有分析工具常因准确性不足而受限,部分方法甚至假设神经网络损失函数为凸函数,难以有效应用。本文提出一种新方法 Diff-In,通过累加连续训练步骤间样本影响的变化来近似其影响。该方法采用二阶逼近,高精度估算差异项,且无需依赖模型凸性假设。尽管为二阶方法,其计算复杂度仍与一阶方法相当,得益于对海森矩阵与梯度乘积的高效有限差分近似。理论与实证分析均表明,Diff-In 的近似误差显著低于现有估计器。在数据清洗、数据删除和核心集选择等三类数据驱动任务中,多基准数据集实验验证了其优越性能。尤其在大规模视觉语言预训练的数据剪枝任务中,可处理数百万数据点,超越强基线方法。
原文摘要 · Abstract (English)
Data plays a pivotal role in the groundbreaking advancements in artificial intelligence. The quantitative analysis of data significantly contributes to model training, enhancing both the efficiency and quality of data utilization. However, existing data analysis tools often lag in accuracy. For instance, many of these tools even assume that the loss function of neural networks is convex. These limitations make it challenging to implement current methods effectively. In this paper, we introduce a new formulation to approximate a sample's influence by accumulating the differences in influence between consecutive learning steps, which we term Diff-In. Specifically, we formulate the sample-wise influence as the cumulative sum of its changes/differences across successive training iterations. By employing second-order approximations, we approximate these difference terms with high accuracy while eliminating the need for model convexity required by existing methods. Despite being a second-order method, Diff-In maintains computational complexity comparable to that of first-order methods and remains scalable. This efficiency is achieved by computing the product of the Hessian and gradient, which can be efficiently approximated using finite differences of first-order gradients. We assess the approximation accuracy of Diff-In both theoretically and empirically. Our theoretical analysis demonstrates that Diff-In achieves significantly lower approximation error compared to existing influence estimators. Extensive experiments further confirm its superior performance across multiple benchmark datasets in three data-centric tasks: data cleaning, data deletion, and coreset selection. Notably, our experiments on data pruning for large-scale vision-language pre-training show that Diff-In can scale to millions of data points and outperforms strong baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。