Leveraging Ensemble Diversity for Robust Self-Training in the Presence of Sample Selection Bias

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Odonnat, Ambroise, Feofanov, Vasilii, Redko, Ievgen
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911825080090624
author Odonnat, Ambroise
Feofanov, Vasilii
Redko, Ievgen
author_facet Odonnat, Ambroise
Feofanov, Vasilii
Redko, Ievgen
contents Self-training is a well-known approach for semi-supervised learning. It consists of iteratively assigning pseudo-labels to unlabeled data for which the model is confident and treating them as labeled examples. For neural networks, softmax prediction probabilities are often used as a confidence measure, although they are known to be overconfident, even for wrong predictions. This phenomenon is particularly intensified in the presence of sample selection bias, i.e., when data labeling is subject to some constraint. To address this issue, we propose a novel confidence measure, called $\mathcal{T}$-similarity, built upon the prediction diversity of an ensemble of linear classifiers. We provide the theoretical analysis of our approach by studying stationary points and describing the relationship between the diversity of the individual members and their performance. We empirically demonstrate the benefit of our confidence measure for three different pseudo-labeling policies on classification datasets of various data modalities. The code is available at https://github.com/ambroiseodt/tsim.
format Preprint
id arxiv_https___arxiv_org_abs_2310_14814
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Leveraging Ensemble Diversity for Robust Self-Training in the Presence of Sample Selection Bias
Odonnat, Ambroise
Feofanov, Vasilii
Redko, Ievgen
Machine Learning
Artificial Intelligence
Self-training is a well-known approach for semi-supervised learning. It consists of iteratively assigning pseudo-labels to unlabeled data for which the model is confident and treating them as labeled examples. For neural networks, softmax prediction probabilities are often used as a confidence measure, although they are known to be overconfident, even for wrong predictions. This phenomenon is particularly intensified in the presence of sample selection bias, i.e., when data labeling is subject to some constraint. To address this issue, we propose a novel confidence measure, called $\mathcal{T}$-similarity, built upon the prediction diversity of an ensemble of linear classifiers. We provide the theoretical analysis of our approach by studying stationary points and describing the relationship between the diversity of the individual members and their performance. We empirically demonstrate the benefit of our confidence measure for three different pseudo-labeling policies on classification datasets of various data modalities. The code is available at https://github.com/ambroiseodt/tsim.
title Leveraging Ensemble Diversity for Robust Self-Training in the Presence of Sample Selection Bias
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2310.14814