arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1906.00512cs.LGstat.ML
收藏

随机广义对抗标签学习

Stochastic Generalized Adversarial Label Learning

Virginia Tech(弗吉尼亚理工学院)

Chidubem Arachie, Bert Huang

摘要:

机器学习模型的使用已经大幅增长,并且正在扩展到几个应用领域。使用机器学习模型的一个常见需求是收集训练这些模型所需的数据。在某些情况下,标记大量数据集可能是一个严重的瓶颈,因此需要开发在难以获得大量数据的训练标签时工作的模型。一个可能的解决方案是弱监督,它使用容易从多个来源获得的噪声标签。挑战在于如何最好地结合这些嘈杂的标签,并训练一个模型,使其在给定的任务中表现良好。在本文中,我们提出了随机广义对抗标签学习(Stoch-GALL),这是一个用于训练机器学习模型的框架,当提供噪声和可能相关的标签时,该模型表现良好。我们的框架允许用户提供不同的弱标签和对这些标签的多个约束。然后,我们的模型试图通过求解非零和博弈优化来学习数据的参数。这个游戏是在为数据选择标签的对手和最小化对手标签造成的错误的模型之间进行的。我们通过训练卷积神经网络模型在三个数据集上测试了我们的方法,这些模型学习对训练标签访问受限的图像对象进行分类。我们的方法即使在弱监督混淆了最先进的弱监督学习方法的情况下也能够学习。我们的实验结果表明,这种方法的适用性一般的分类任务。

英文摘要:

The usage of machine learning models has grown substantially and is spreading into several application domains. A common need in using machine learning models is collecting the data required to train these models. In some cases, labeling a massive dataset can be a crippling bottleneck, so there is need to develop models that work when training labels for large amounts of data are not easily obtained. A possible solution is weak supervision, which uses noisy labels that are easily obtained from multiple sources. The challenge is how best to combine these noisy labels and train a model to perform well given a task. In this paper, we propose stochastic generalized adversarial label learning (Stoch-GALL), a framework for training machine learning models that perform well when noisy and possibly correlated labels are provided. Our framework allows users to provide different weak labels and multiple constraints on these labels. Our model then attempts to learn parameters for the data by solving a non-zero sum game optimization. The game is between an adversary that chooses labels for the data and a model that minimizes the error made by the adversarial labels. We test our method on three datasets by training convolutional neural network models that learn to classify image objects with limited access to training labels. Our approach is able to learn even in settings where the weak supervision confounds state-of-the-art weakly supervised learning methods. The results of our experiments demonstrate the applicability of this approach to general classification tasks.

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

↑