arXiv:2411.15292cs.LGcs.AI2024-11

用可快速计算的正则性切向量,实现回归模型高效主动学习。

Influence functions and regularity tangents for efficient active learning

  • 通过训练时同步计算正则性切向量,衡量数据点对模型复杂度的影响。
  • 仅需额外一倍存储,评估新数据点的“好奇心”速度与损失梯度相当。
  • 该方法基于影响函数,适合需要高效标注选择的半监督回归任务。

本文提出一种高效方法,使回归模型具备对数据的‘好奇心’。在机器学习中,这一机制称为主动学习,旨在半监督设置下自动选择需标注的数据点。所提方法通过在训练过程中同步计算一个‘正则性切向量’(regularity tangent),其维度与模型参数向量相同。该切向量与某数据点损失梯度的内积,构成一个影响函数,用于度量该点被加权后引起的模型复杂度预期平方变化。在最简实例中,仅需一个切向量,且其计算仅带来恒定的延迟。训练完成后,评估任意候选数据点的‘好奇心’,速度等同于计算损失梯度。该向量仅使模型存储量翻倍。我们展示了该量作为影响函数的有效性,并提出了多种利用此量及其他相关量来选择新训练数据的方法。

原文摘要 · Abstract (English)

In this paper we describe an efficient method for providing a regression model with a sense of curiosity about its data. In the field of machine learning, our framework for representing curiosity is called Active Learning, which concerns the problem of automatically choosing data points for which to query labels in the semi-supervised setting. The methods we propose are based on computing a "regularity tangent" vector that can be calculated (with only a constant slow-down) together with the model's parameter vector during training. We then take the inner product of this tangent vector with the gradient vector of the model's loss at a given data point to obtain a measure of the influence of that point on the complexity of the model. In the simplest instantiation, there is only a single regularity tangent vector, of the same dimension as the parameter vector. Thus, in the proposed technique, once training is complete, evaluating our "curiosity" about a potential query data point can be done as quickly as calculating the model's loss gradient at that point. The new vector only doubles the amount of storage required by the model. We show that the quantity computed by our technique is an example of an "influence function", and that it measures the expected squared change in model complexity incurred by up-weighting a given data point. We propose a number of ways for using this and other related quantities to choose new training data points for a regression model.

主动学习影响函数回归模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。