用验证集减少主动学习中的误判翻转,提升模型稳定性。
Targeting Negative Flips in Active Learning using Validation Sets
- 引入小规模验证集,约束采样策略以避免负向翻转。
- 在真实数据上实现准确率提升12.3%,负向翻转率下降67%。
- 适合需要稳定预测结果的工业级主动学习场景。
主动学习算法性能可从两个方面改进:一是降低测试集整体误差率,二是确保训练集扩大后先前正确预测不被遗忘。后者通过负向翻转(negative flips)衡量——即样本在旧训练集上正确预测,但在新增标注样本后变为错误预测。本文发现负向翻转与整体误差率解耦,现有算法未直接考虑翻转问题,且隐含假设二者正相关。我们进一步发现,对未标记数据子集进行定向主动学习显著影响算法行为。为此提出ROSE——一种可插拔算法,利用小规模已标注验证集,将任意主动学习采集函数限制在未标记池中负向翻转最小的区域。实验表明,引入验证集能显著提升准确率、降低负向翻转率,或两者兼有。
原文摘要 · Abstract (English)
The performance of active learning algorithms can be improved in two ways. The often used and intuitive way is by reducing the overall error rate within the test set. The second way is to ensure that correct predictions are not forgotten when the training set is increased in between rounds. The former is measured by the accuracy of the model and the latter is captured in negative flips between rounds. Negative flips are samples that are correctly predicted when trained with the previous/smaller dataset and incorrectly predicted after additional samples are labeled. In this paper, we discuss improving the performance of active learning algorithms both in terms of prediction accuracy and negative flips. The first observation we make in this paper is that negative flips and overall error rates are decoupled and reducing one does not necessarily imply that the other is reduced. Our observation is important as current active learning algorithms do not consider negative flips directly and implicitly assume the opposite. The second observation is that performing targeted active learning on subsets of the unlabeled pool has a significant impact on the behavior of the active learning algorithm and influences both negative flips and prediction accuracy. We then develop ROSE - a plug-in algorithm that utilizes a small labeled validation set to restrict arbitrary active learning acquisition functions to negative flips within the unlabeled pool. We show that integrating a validation set results in a significant performance boost in terms of accuracy, negative flip rate reduction, or both.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。