arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.02110cs.LGcs.AIcs.CVcs.NE
收藏

复合主动学习:面向多领域主动学习的理论保证

Composite Active Learning: Towards Multi-Domain Active Learning with Theoretical Guarantees

Rutgers(罗格斯大学)

Guang-Yuan Hao, Hengguan Huang, Haotian Wang, Jie Gao, Hao Wang

AI总结:

本文提出复合主动学习(CAL),首个通用多领域主动学习方法,通过优化上界误差分配领域预算并结合实例级查询策略,在理论和实证上优于现有方法。

摘要:

主动学习(AL)旨在通过选择最具信息量的数据点进行标注,在固定的标注预算内提高模型性能。现有的主动学习主要关注单领域场景,即所有数据来自同一领域(例如同一数据集)。然而,许多现实世界任务往往涉及多个领域。例如,在视觉识别中,通常需要训练一个能在不同环境(如不同背景)下工作的图像分类器,其中每个环境的图像构成一个领域。这种多领域主动学习设置对先前方法具有挑战性,因为它们(1)在分配标注预算时忽略了不同领域之间的相似性,(2)未能处理跨领域数据的分布偏移。在本文中,我们提出了第一个通用方法,称为复合主动学习(CAL),用于多领域主动学习。我们的方法明确考虑了问题中的领域级和实例级信息;CAL首先根据领域级重要性分配领域级预算,该重要性通过优化我们开发的上界误差来估计;在领域级预算下,CAL随后利用某种实例级查询策略从每个领域中选择样本进行标注。我们的理论分析表明,与当前的主动学习方法相比,我们的方法实现了更优的误差界。我们的实证结果表明,在合成和真实世界的多领域数据集上,我们的方法显著优于最先进的主动学习方法。代码可在 https://github.com/Wang-ML-Lab/multi-domain-active-learning 获取。

英文摘要:

Active learning (AL) aims to improve model performance within a fixed labeling budget by choosing the most informative data points to label. Existing AL focuses on the single-domain setting, where all data come from the same domain (e.g., the same dataset). However, many real-world tasks often involve multiple domains. For example, in visual recognition, it is often desirable to train an image classifier that works across different environments (e.g., different backgrounds), where images from each environment constitute one domain. Such a multi-domain AL setting is challenging for prior methods because they (1) ignore the similarity among different domains when assigning labeling budget and (2) fail to handle distribution shift of data across different domains. In this paper, we propose the first general method, dubbed composite active learning (CAL), for multi-domain AL. Our approach explicitly considers the domain-level and instance-level information in the problem; CAL first assigns domain-level budgets according to domain-level importance, which is estimated by optimizing an upper error bound that we develop; with the domain-level budgets, CAL then leverages a certain instance-level query strategy to select samples to label from each domain. Our theoretical analysis shows that our method achieves a better error bound compared to current AL methods. Our empirical results demonstrate that our approach significantly outperforms the state-of-the-art AL methods on both synthetic and real-world multi-domain datasets. Code is available at https://github.com/Wang-ML-Lab/multi-domain-active-learning.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑