Leveraging Label Proportion Prior for Class-Imbalanced Semi-Supervised Learning

Kohki Akiba, Shinnosuke Matsuo, Shota Harada, Ryoma Bise
3/3/2026
cs.LGcs.CV

No confident code implementation found for this paper. 2 possible matches can be shown below.

Abstract

Semi-supervised learning (SSL) often suffers under class imbalance, where pseudo-labeling amplifies majority bias and suppresses minority performance. We address this issue with a lightweight framework that, to our knowledge, is the first to introduce Proportion Loss from learning from label proportions (LLP) into SSL as a regularization term. Proportion Loss aligns model predictions with the global class distribution, mitigating bias across both majority and minority classes. To further stabilize training, we formulate a stochastic variant that accounts for fluctuations in mini-batch composition. Experiments on the Long-tailed CIFAR-10 benchmark show that integrating Proportion Loss into FixMatch and ReMixMatch consistently improves performance over the baselines across imbalance severities and label ratios, and achieves competitive or superior results compared to existing CISSL methods, particularly under scarce-label conditions.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. 2 weaker matches are hidden by default — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{akiba2026leveraging,
  title  = {Leveraging Label Proportion Prior for Class-Imbalanced Semi-Supervised Learning},
  author = {Kohki Akiba and Shinnosuke Matsuo and Shota Harada and Ryoma Bise},
  year   = {2026},
  eprint = {2603.02957},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2603.02957v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2603.02957.svg)](https://codeofpaper.com/paper/2603.02957)

Discussion