Abstract
Semi-supervised learning has achieved extraordinary success in prevalent image-classification benchmarks. However, a class-balanced distribution that differs notably from real-world data distribution is required. In general, models trained under class-imbalanced semi-supervised learning conditions are severely biased towards the majority classes. To address this issue, we propose a novel framework called ABAE by implanting an Auxiliary Balanced AutoEncoder branch into existing semi-supervised learning algorithms. Considering that adaptive feature augmentation for different classes can alleviate confirmation bias, we devise a class-aware reconstruction loss to train the AutoEncoder module. To further smooth the output, we adopt a graph-based label propagation scheme at the end of the AutoEncoder. Extensive experiments on CIFAR-10/100-LT, SVHN-LT and Small ImageNet-127 demonstrate the effectiveness of ABAE.
Sign-in/Register to access full text options 
Talk to us
Join us for a 30 min session where you can share your feedback and ask us any queries you have
Schedule a callDisclaimer: All third-party content on this website/platform is and will remain the property of their respective owners and is provided on "as is" basis without any warranties, express or implied. Use of third-party content does not indicate any affiliation, sponsorship with or endorsement by them. Any references to third-party content is to identify the corresponding services and shall be considered fair use under The CopyrightLaw.
