Deep learning from label proportions with labeled samples

Yong Shi, Jiabin Liu, Bo Wang, Zhiquan Qi*, Ying Jie Tian

*此作品的通讯作者

科研成果: 期刊稿件文章同行评审

12 引用 (Scopus)

摘要

Learning from label proportions (LLP), where the training data is in form of bags, and only the proportions of classes in each bag are available, has attracted wide interest in machine learning community. In general, most LLP algorithms adopt random sampling to obtain the proportional information of different categories, which correspondingly obtains some labeled samples in each bag. However, LLP training process always fails to leverage these labeled samples, which may contain essential data distribution information. To address this issue, in this paper, we propose end-to-end LLP solver based on convolutional neural networks (ConvNets), called LLP with labeled samples (LLP-LS). First, we reshape the cross entropy loss in ConvNets, so that it can combine the proportional information and labeled samples in each bag. Second, in order to comply with the training data in a bag manner, ADAM based on batch is employed to train LLP-LS. Hence, the batch size in training process is in accordance with the bag size. Compared with up-to-date methods on multi-class problem, our algorithm can obtain the state-of-the-art on several image datasets.

源语言英语
页(从-至)73-81
页数9
期刊Neural Networks
128
DOI
出版状态已出版 - 8月 2020
已对外发布

指纹

探究 'Deep learning from label proportions with labeled samples' 的科研主题。它们共同构成独一无二的指纹。

引用此