Semi-Supervised Mixture Models under the Concept of Missing at Radom with Margin Confidence and Aranda Ordaz Function
Jinyang Liao, Ziyang Lyu
TL;DR
The paper tackles semi-supervised learning with missing labels under Missing at Random (MAR) within Gaussian mixture models. It introduces margin confidence to quantify classification uncertainty and the Aranda--Ordaz (AO) link to flexibly model how missingness depends on this uncertainty, enabling a MAR-aware mixture modeling approach. An Expectation-Conditional Maximization (ECM) algorithm jointly estimates the GMM parameters and MAR parameters, yielding a Bayesian classifier to impute missing labels. Across controlled simulations and a real-world MAGIC Gamma Telescope dataset, the proposed ECM--AO framework improves calibration and maintains robust discrimination under moderate label sparsity, outperforming a logistic baseline that ignores missingness. The AO link’s flexibility provides stability in uncertain settings, while limitations remain for extreme missingness and multi-component mixtures, pointing to future work on broader uncertainty measures and higher-order mixtures.
Abstract
This paper presents a semi-supervised learning framework for Gaussian mixture modelling under a Missing at Random (MAR) mechanism. The method explicitly parameterizes the missingness mechanism by modelling the probability of missingness as a function of classification uncertainty. To quantify classification uncertainty, we introduce margin confidence and incorporate the Aranda Ordaz (AO) link function to flexibly capture the asymmetric relationships between uncertainty and missing probability. Based on this formulation, we develop an efficient Expectation Conditional Maximization (ECM) algorithm that jointly estimates all parameters appearing in both the Gaussian mixture model (GMM) and the missingness mechanism, and subsequently imputes the missing labels by a Bayesian classifier derived from the fitted mixture model. This method effectively alleviates the bias induced by ignoring the missingness mechanism while enhancing the robustness of semi-supervised learning. The resulting uncertainty-aware framework delivers reliable classification performance in realistic MAR scenarios with substantial proportions of missing labels.
