基于不确定性和动态标签相关性的批量选择,提升多标签模型训练效果
Batch Selection for Multi-Label Classification Guided by Uncertainty and Dynamic Label Correlations
- 通过预测差异和当前置信度评估每标签不确定性
- 利用动态演变的标签间不确定性关联,优先选择协同高不确定样本
- 适用于多标签分类场景,尤其在数据标注不全时表现更优
深度神经网络的精度高度依赖于训练过程中的小批量构建效果。在单标签任务中,倾向于选择高不确定性的样本能优于基于难度的方法。尽管已有两种针对多标签数据的批处理选择方法,但均未利用重要的不确定性信息。将不确定性概念拓展至多标签场景面临两大挑战:一是传统方差或熵类不确定性度量忽略了滑动窗口内的预测波动及当前模型状态的重要性;二是现有方法未显式建模标签相关性,尤其是随训练过程演化的不确定性相关性。本文提出一种基于不确定性的多标签批量选择算法,通过分析连续预测间的差异与当前输出置信度来评估各标签的不确定性,并进一步利用动态的不确定性相关性,强调那些在多个标签上协同表现出高不确定性的样本。实验证明,该方法可有效提升多种多标签深度学习模型的性能并加速收敛。
原文摘要 · Abstract (English)
The accuracy of deep neural networks is significantly influenced by the effectiveness of mini-batch construction during training. In single-label scenarios, such as binary and multi-class classification tasks, it has been demonstrated that batch selection algorithms preferring samples with higher uncertainty achieve better performance than difficulty-based methods. Although there are two batch selection methods tailored for multi-label data, none of them leverage important uncertainty information. Adapting the concept of uncertainty to multi-label data is not a trivial task, since there are two issues that should be tackled. First, traditional variance or entropy-based uncertainty measures ignore fluctuations of predictions within sliding windows and the importance of the current model state. Second, existing multi-label methods do not explicitly exploit the label correlations, particularly the uncertainty-based label correlations that evolve during the training process. In this paper, we propose an uncertainty-based multi-label batch selection algorithm. It assesses uncertainty for each label by considering differences between successive predictions and the confidence of current outputs, and further leverages dynamic uncertainty-based label correlations to emphasize instances whose uncertainty is synergistically expressed across multiple labels. Empirical studies demonstrate the effectiveness of our method in improving the performance and accelerating the convergence of various multi-label deep learning models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。