arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2202.05613cs.LG

CMW-Net: 面向鲁棒深度学习的类别感知样本加权映射学习

CMW-Net: Learning a Class-Aware Sample Weighting Mapping for Robust Deep Learning

Peng Cheng Laboratory(鹏城实验室);Macau University of Science and Technology(澳门科技大学)

Jun Shu, Xiang Yuan, Deyu Meng, Zongben Xu

AI总结:

针对现有样本重加权方法需手动指定方案和超参数、通用性差的问题,提出CMW-Net元模型,自适应学习类别感知的显式加权映射,在多种数据偏差场景下性能优于现有方法且具备可迁移性和通用性。

摘要:

现代深度神经网络很容易在包含损坏标签或类别不平衡的有偏训练数据上过拟合。样本重加权方法被广泛用于缓解这一数据偏差问题。然而,当前大多数方法都需要根据所研究问题和训练数据的特征,手动预先指定加权方案及其额外超参数。由于数据偏差情况的高度复杂性和类别间差异,这些方法很难在实际场景中通用。为解决这一问题,我们提出一种元模型,能够直接从数据中自适应学习显式加权方案。具体而言,我们将每个训练类别视为一个独立的学习任务,旨在提取一个以样本损失和任务/类别特征为输入、样本权重为输出的显式加权函数,期望根据不同样本类别的固有偏差特征,为其施加自适应变化的加权方案。合成数据和真实数据实验证实,我们的方法能够在多种数据偏差场景下生成合理的加权方案,包括类别不平衡、特征无关和特征相关的标签噪声场景,以及超出常规情况的更复杂偏差场景。此外,通过将在规模较小的CIFAR-10数据集上学到的加权函数直接部署到规模大得多的完整WebVision数据集上,也证实了所学加权方案的任务可迁移性。与此前的最优(SOAT)方法相比,无需额外的超参数调优和元梯度下降步骤即可获得性能提升。我们的方法在部分标签学习、半监督学习和选择性分类等多个鲁棒深度学习问题上的通用性也得到了验证。

英文摘要:

Modern deep neural networks can easily overfit to biased training data containing corrupted labels or class imbalance. Sample re-weighting methods are popularly used to alleviate this data bias issue. Most current methods, however, require to manually pre-specify the weighting schemes as well as their additional hyper-parameters relying on the characteristics of the investigated problem and training data. This makes them fairly hard to be generally applied in practical scenarios, due to their significant complexities and inter-class variations of data bias situations. To address this issue, we propose a meta-model capable of adaptively learning an explicit weighting scheme directly from data. Specifically, by seeing each training class as a separate learning task, our method aims to extract an explicit weighting function with sample loss and task/class feature as input, and sample weight as output, expecting to impose adaptively varying weighting schemes to different sample classes based on their own intrinsic bias characteristics. Synthetic and real data experiments substantiate the capability of our method on achieving proper weighting schemes in various data bias cases, like the class imbalance, feature-independent and dependent label noise scenarios, and more complicated bias scenarios beyond conventional cases. Besides, the task-transferability of the learned weighting scheme is also substantiated, by readily deploying the weighting function learned on relatively smaller-scale CIFAR-10 dataset on much larger-scale full WebVision dataset. A performance gain can be readily achieved compared with previous SOAT ones without additional hyper-parameter tuning and meta gradient descent step. The general availability of our method for multiple robust deep learning issues, including partial-label learning, semi-supervised learning and selective classification, has also been validated.

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

↑