Btda: Basis Transformation Based Distribution Alignment for Imbalanced Semi-Supervised Learning

Jinhuang Ye,Xiaozhi Gao,Zuoyong Li,Jiawei Wu,Xiaofeng Xu,Xianghan Zheng
DOI: https://doi.org/10.1007/s13042-024-02122-6
2024-01-01
International Journal of Machine Learning and Cybernetics
Abstract:Semi-supervised learning (SSL) employs unlabeled data with limited labeled samples to enhance deep networks, but imbalance degrades performance due to biased pseudo-labels skewing decision boundaries. To address this challenge, we propose two optimization conditions inspired by our theoretical analysis. These conditions focus on aligning class distributions and representations. Additionally, we introduce a plug-and-play method called Basis Transformation based distribution alignment (BTDA) that efficiently aligns class distributions while considering inter-class relationships. BTDA mitigates the negative impact of biased pseudo-labels through basis transformation, which involves a learnable transition matrix. Extensive experiments demonstrate the effectiveness of integrating existing SSL methods with BTDA in image classification tasks with class imbalance. For example, BTDA achieves accuracy improvements ranging from 2.47 to 6.66
What problem does this paper attempt to address?