Progressive Training for Learning From Label Proportions.

Journal: IEEE transactions on neural networks and learning systems
Published Date:

Abstract

Learning from label proportions (LLPs), which aims to learn an instance-level classifier using proportion-based grouped training data, has garnered increasing attention in the field of machine learning. Existing deep learning-based LLP methods employ end-to-end pipelines to derive proportional loss functions via the Kullback-Leibler (KL) divergence between bag-level prior and posterior class distributions. However, the optimal solutions of these methods often struggle to conform to the given proportions, inevitably leading to degradation in the final classification performance. In this article, we address this issue by proposing a novel progressive training method for LLP, termed PT-LLP, which strives to meet the proportion constraints from the bag level to the instance level. Specifically, we first train a model by using the existing KL-divergence-based LLP methods that are consistent with bag-level proportion information. Then, we impose additional constraints on strict proportion consistency to the classifier to further move toward a more ideal solution by reformulating it as a constrained optimization problem, which can be efficiently solved using optimal transport (OT) algorithms. In particular, the knowledge distillation is employed as a transition stage to transfer the bag-level information to the instance level using a teacher-student framework. Finally, our framework is model-agnostic and demonstrates significant performance improvements through extensive experiments on different datasets when incorporated into other deep LLP methods as the first training stage.

Authors

  • Jiabin Liu
    School of Computer and Control Engineering, University of Chinese Academy Sciences, Beijing 100190, China.
  • Bo Wang
    Department of Clinical Laboratory Medicine Center, Inner Mongolia Autonomous Region People's Hospital, Hohhot, Inner Mongolia, China.
  • Yuping Zhang
    School of Automation Engineering, University of Electronic Science and Technology of China, Chengdu, Sichuan 611731, PR China.
  • Huadong Wang
    School of Computer Science and Technology, Anhui University, China.
  • Biao Li
    Key Laboratory of Renewable Energy, Guangdong Key Laboratory of New and Renewable Energy Research and Development, Guangzhou Institute of Energy Conversion, Chinese Academy of Sciences, Guangzhou 510640, China.
  • Xin Shen
  • Gang Kou
    School of Business Administration, Southwestern University of Finance and Economics, Chengdu 611130, China. Electronic address: kougang@swufe.edu.cn.

Keywords

No keywords available for this article.