自动搜索最优主动学习策略,提升模型标注效率。
AutoAL: Automated Active Learning with Differentiable Query Strategy Search
- 用可微分双层优化框架联合训练搜索与拟合网络
- 在多个数据集上表现优于所有候选算法
- 适合需要高效标注的深度学习任务
随着深度学习发展,数据效率愈发重要。由于大规模数据标注耗时且昂贵,主动学习(AL)通过迭代选择最具信息量的样本进行标注,有效降低标注成本。然而,不同AL算法在各类数据场景中表现差异大,难以确定最优策略。本文提出首个可微分的主动学习策略搜索方法AutoAL,基于现有采样策略构建。AutoAL包含SearchNet和FitNet两个神经网络,在可微分双层优化框架下协同优化。针对具体任务,二者利用已标注数据共同学习各候选AL算法的表现。最终由SearchNet从无标注池中选出最优子集进行标注,实现高效模型训练。实验表明,AutoAL在多个任务上均显著优于所有候选算法及其他选择性主动学习方法,展现了其在跨任务、跨领域整合多种现有AL方法的巨大潜力。代码已开源:https://github.com/haizailache999/AutoAL。
原文摘要 · Abstract (English)
As deep learning continues to evolve, the need for data efficiency becomes increasingly important. Considering labeling large datasets is both time-consuming and expensive, active learning (AL) provides a promising solution to this challenge by iteratively selecting the most informative subsets of examples to train deep neural networks, thereby reducing the labeling cost. However, the effectiveness of different AL algorithms can vary significantly across data scenarios, and determining which AL algorithm best fits a given task remains a challenging problem. This work presents the first differentiable AL strategy search method, named AutoAL, which is designed on top of existing AL sampling strategies. AutoAL consists of two neural nets, named SearchNet and FitNet, which are optimized concurrently under a differentiable bi-level optimization framework. For any given task, SearchNet and FitNet are iteratively co-optimized using the labeled data, learning how well a set of candidate AL algorithms perform on that task. With the optimal AL strategies identified, SearchNet selects a small subset from the unlabeled pool for querying their annotations, enabling efficient training of the task model. Experimental results demonstrate that AutoAL consistently achieves superior accuracy compared to all candidate AL algorithms and other selective AL approaches, showcasing its potential for adapting and integrating multiple existing AL methods across diverse tasks and domains. Code is available at: https://github.com/haizailache999/AutoAL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。