New Framework Improves Biased Positive-Unlabeled Learning with Causal Inference.

Xutao Wang, Hanting Chen, Tianyu Guo, Yunhe Wang· July 16, 2026 View original

Summary

This paper introduces PUe, a framework enhancing Positive-Unlabeled (PU) learning by addressing selection bias in real-world datasets. It uses normalized propensity scores and inverse probability weighting to improve binary classification accuracy with limited positive labels.

Positive-Unlabeled (PU) learning is a machine learning technique used for binary classification when only a small number of positive examples are labeled, alongside a large pool of unlabeled data. A common challenge in these scenarios is selection bias, where the observed positive labels are not randomly distributed, leading to skewed datasets. Researchers have developed a new framework, PUe, which aims to mitigate this bias. Building on existing propensity-weighted methods, PUe incorporates normalized propensity scores and normalized inverse probability weighting to create a more robust risk formulation. The framework also includes theoretical analyses of sample-weight error under biased labeling, regularized deep propensity-score estimation, and integration with current cost-sensitive PU methods. Experimental results on standard datasets like MNIST and CIFAR-10 show that PUe significantly improves performance compared to baseline PU methods when dealing with non-uniform label distributions.

Why it matters

Professionals dealing with imbalanced datasets or limited labeled data can leverage this framework to build more accurate and reliable classification models, especially in domains where labeling is costly or difficult.

How to implement this in your domain

  1. 1Evaluate existing PU learning pipelines for potential selection bias in data labeling.
  2. 2Integrate propensity score estimation techniques to quantify and correct for bias in unlabeled data.
  3. 3Apply normalized inverse probability weighting within your PU learning objective function.
  4. 4Test the PUe framework's effectiveness on your specific datasets, comparing it against current baselines.
  5. 5Consider using deep learning for propensity score estimation to handle complex data distributions.

Who benefits

HealthcareFinanceE-commerceCybersecurityManufacturing

Key takeaways

  • Real-world PU learning often suffers from selection bias due to non-random label distributions.
  • The PUe framework uses causal inference and normalized weighting to correct for this bias.
  • It improves classification accuracy, especially with uneven label distributions and limited positive examples.
  • The framework offers theoretical grounding and practical integration with existing PU methods.

Original post by Xutao Wang, Hanting Chen, Tianyu Guo, Yunhe Wang

"arXiv:2607.13428v1 Announce Type: new Abstract: Positive-Unlabeled (PU) learning aims to achieve high-accuracy binary classification with limited labeled positive examples and numerous unlabeled ones. Existing cost-sensitive-based methods often rely on strong assumptions that exa…"

View on X

Originally posted by Xutao Wang, Hanting Chen, Tianyu Guo, Yunhe Wang on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses

More in AI Research

AI Engineering & DevToolsAI Research

NodeImport Improves Imbalanced Node Classification on Graphs

NodeImport is a new framework addressing class imbalance in graph node classification by assessing node importance to create a balanced meta-set for training. It dynamically filters valuable labeled, unlabeled, and synthetic nodes, outperforming existing baselines across various datasets and GNN architectures.

Nan Chen, Zemin Liu, Bryan Hooi, Bingsheng He, Jun Hu, Jia ChenJul 16, 2026
AI ResearchAI Engineering & DevTools

Neural Spline Flows Aid Dark Matter Search in CMS Data.

This paper reports a search for dark matter produced with a leptonically decaying Z boson using CMS Run 2015D open data and Neural Spline Flows. The method models signal and background densities to set upper limits on signal-strength parameters for various dark matter mediators, though observed limits are weaker than expected due to background modeling discrepancies.

Hitesh Rasineni (VIT-AP University, Amaravati, India), Bhavishya Chebrolu (Mohan Babu University, Tirupati, India)Jul 16, 2026
AI Engineering & DevToolsAI Research

Multiplex Graph Transformer Boosts Power Grid Model Generalization.

Researchers introduce MxGPS, a multiplex graph transformer designed to overcome "topology overfitting" in power grid problems. By jointly training on multiple tasks with a shared encoder, MxGPS achieves superior zero-shot generalization across unseen grid topologies, demonstrating high accuracy and low boundary violation rates with significantly fewer parameters.

Charilaos Papaioannou, Ioannis Tsantilas, Dimitris Giannakakos, Vasilis Michalakopoulos, Sotiris Pelekis, Vangelis Marinakis, Arsam Aryandoust, Antonello Monti, Ricardo J. Bessa, Perdo P. Vergara, Jochen Cremer, Elissaios SarmasJul 16, 2026