CalibrateMix: Guided-Mixup Calibration of Image Semi-Supervised Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Rahman, Mehrab Mustafy, Mohan, Jayanth, Sosea, Tiberiu, Caragea, Cornelia
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915621760925696
author Rahman, Mehrab Mustafy
Mohan, Jayanth
Sosea, Tiberiu
Caragea, Cornelia
author_facet Rahman, Mehrab Mustafy
Mohan, Jayanth
Sosea, Tiberiu
Caragea, Cornelia
contents Semi-supervised learning (SSL) has demonstrated high performance in image classification tasks by effectively utilizing both labeled and unlabeled data. However, existing SSL methods often suffer from poor calibration, with models yielding overconfident predictions that misrepresent actual prediction likelihoods. Recently, neural networks trained with {\tt mixup} that linearly interpolates random examples from the training set have shown better calibration in supervised settings. However, calibration of neural models remains under-explored in semi-supervised settings. Although effective in supervised model calibration, random mixup of pseudolabels in SSL presents challenges due to the overconfidence and unreliability of pseudolabels. In this work, we introduce CalibrateMix, a targeted mixup-based approach that aims to improve the calibration of SSL models while maintaining or even improving their classification accuracy. Our method leverages training dynamics of labeled and unlabeled samples to identify ``easy-to-learn'' and ``hard-to-learn'' samples, which in turn are utilized in a targeted mixup of easy and hard samples. Experimental results across several benchmark image datasets show that our method achieves lower expected calibration error (ECE) and superior accuracy compared to existing SSL approaches.
format Preprint
id arxiv_https___arxiv_org_abs_2511_12964
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle CalibrateMix: Guided-Mixup Calibration of Image Semi-Supervised Models
Rahman, Mehrab Mustafy
Mohan, Jayanth
Sosea, Tiberiu
Caragea, Cornelia
Computer Vision and Pattern Recognition
Artificial Intelligence
Semi-supervised learning (SSL) has demonstrated high performance in image classification tasks by effectively utilizing both labeled and unlabeled data. However, existing SSL methods often suffer from poor calibration, with models yielding overconfident predictions that misrepresent actual prediction likelihoods. Recently, neural networks trained with {\tt mixup} that linearly interpolates random examples from the training set have shown better calibration in supervised settings. However, calibration of neural models remains under-explored in semi-supervised settings. Although effective in supervised model calibration, random mixup of pseudolabels in SSL presents challenges due to the overconfidence and unreliability of pseudolabels. In this work, we introduce CalibrateMix, a targeted mixup-based approach that aims to improve the calibration of SSL models while maintaining or even improving their classification accuracy. Our method leverages training dynamics of labeled and unlabeled samples to identify ``easy-to-learn'' and ``hard-to-learn'' samples, which in turn are utilized in a targeted mixup of easy and hard samples. Experimental results across several benchmark image datasets show that our method achieves lower expected calibration error (ECE) and superior accuracy compared to existing SSL approaches.
title CalibrateMix: Guided-Mixup Calibration of Image Semi-Supervised Models
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2511.12964