arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2407.14720cs.LG
收藏

下游-预文本领域知识回溯用于主动学习

Downstream-Pretext Domain Knowledge Traceback for Active Learning

University of Chinese Academy of Sciences(中国科学院大学);Institute of Computing Technology, Chinese Academy of Sciences(中国科学院计算技术研究所);University of Science and Technology of China(中国科学技术大学);University of Rochester(罗切斯特大学)

Beichen Zhang, Liang Li, Zheng-Jun Zha, Jiebo Luo, Qingming Huang

AI总结:

针对主动学习中预训练表示不足的问题,提出DOKT方法,通过回溯下游知识与预训练指导的交互,结合多样性指标和不确定性估计选择多样且重要的样本,在十个数据集上优于现有方法。

摘要:

主动学习(AL)旨在通过迭代选择最具信息量的样本来构建高质量的标注数据集。这种采样在很大程度上依赖于数据表示,而近期预训练在鲁棒特征学习方面颇为流行。然而,由于预训练利用的是缺乏标注的低级预文本任务,在主动学习中直接使用预训练表示不足以确定采样分数。为解决此问题,我们提出了一种下游-预文本领域知识回溯(DOKT)方法,该方法追踪下游知识与预训练指导的数据交互,以选择决策边界附近多样且具有指导性的样本。DOKT由回溯多样性指标和基于领域的确定性估计器组成。多样性指标基于预训练预文本模型和来自标注的下游知识构建两个特征空间,通过该指标在预文本空间中定位来自下游空间的未标注数据的邻居,以探索样本的交互。借助此机制,DOKT统一了低级和高级表示的数据关系以估计回溯多样性。接下来,在确定性估计器中,设计了领域混合以对预文本空间中具有相似视觉补丁的未标注样本强制进行感知扰动。然后测量扰动样本的差异以估计领域不确定性。因此,DOKT基于这两个模块选择最多样且重要的样本。在十个数据集上进行的实验表明,我们的模型优于其他最先进的方法,并能很好地泛化到各种应用场景,如语义分割和图像字幕生成。

英文摘要:

Active learning (AL) is designed to construct a high-quality labeled dataset by iteratively selecting the most informative samples. Such sampling heavily relies on data representation, while recently pre-training is popular for robust feature learning. However, as pre-training utilizes low-level pretext tasks that lack annotation, directly using pre-trained representation in AL is inadequate for determining the sampling score. To address this problem, we propose a downstream-pretext domain knowledge traceback (DOKT) method that traces the data interactions of downstream knowledge and pre-training guidance for selecting diverse and instructive samples near the decision boundary. DOKT consists of a traceback diversity indicator and a domain-based uncertainty estimator. The diversity indicator constructs two feature spaces based on the pre-training pretext model and the downstream knowledge from annotation, by which it locates the neighbors of unlabeled data from the downstream space in the pretext space to explore the interaction of samples. With this mechanism, DOKT unifies the data relations of low-level and high-level representations to estimate traceback diversity. Next, in the uncertainty estimator, domain mixing is designed to enforce perceptual perturbing to unlabeled samples with similar visual patches in the pretext space. Then the divergence of perturbed samples is measured to estimate the domain uncertainty. As a result, DOKT selects the most diverse and important samples based on these two modules. The experiments conducted on ten datasets show that our model outperforms other state-of-the-art methods and generalizes well to various application scenarios such as semantic segmentation and image captioning.

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

↑