Enhancing the Multiclass Image Classification Accuracy using Binary Classifiers for Semi-Supervised Learning
Küçük Resim Yok
Tarih
2023
Yazarlar
Dergi Başlığı
Dergi ISSN
Cilt Başlığı
Yayıncı
Institute of Electrical and Electronics Engineers Inc.
Erişim Hakkı
info:eu-repo/semantics/closedAccess
Özet
Image classification poses a fundamental challenge in deep learning, especially in scenarios where labeled data is scarce but unlabeled data is abundant. Precise pseudo-labels are crucial to facilitate classification in such situations. One common approach involves the use of binary classifiers with a one-vs-all strategy to assign pseudo-labels to unlabeled data, offering the advantage of tailored predictions for each class. However, this method faces challenges, including class imbalance, often requiring oversampling for resolution, and extended training times due to multiple binary classifiers. Our proposed approach addresses the inherent class imbalance in the one-vs-all method, eliminating the need for oversampling. We achieve this by training a single multi-class classifier through a combination of binary classifiers, transfer learning, and fine-tuning while enforcing a stringent prediction threshold for pseudo-labels. This transition to a single multi-class classifier significantly reduces both training duration and storage demands. Our model's effectiveness is rigorously evaluated on two diverse datasets, MNIST, and Fashion MNIST, achieving impressive test accuracies of 95.59% and 84.84%, respectively, for a pseudo-label generation. © 2023 IEEE.
Açıklama
Central Michigan University (CMU);IEEE
2023 IEEE International Conference on Artificial Intelligence, Blockchain, and Internet of Things, AIBThings 2023 -- 16 September 2023 through 17 September 2023 -- -- 194014
2023 IEEE International Conference on Artificial Intelligence, Blockchain, and Internet of Things, AIBThings 2023 -- 16 September 2023 through 17 September 2023 -- -- 194014
Anahtar Kelimeler
Binary Classifiers, Class Imbalance, Image Classification, Pseudo-Label Generation, Semi-Supervised Learning
Kaynak
2023 IEEE International Conference on Artificial Intelligence, Blockchain, and Internet of Things, AIBThings 2023 - Proceedings
WoS Q Değeri
Scopus Q Değeri
N/A
