{"about":{"site":"https://codewithpapers.app","non_affiliation":"Code with Papers and Syntology are not affiliated with, endorsed by, or sponsored by Papers with Code, Meta, or the pwc-archive mirror.","licence":"CC BY-SA 4.0","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","attribution":"https://codewithpapers.app/attribution","modified":"archive material modified by Syntology; see the attribution page"},"url":"/paper/mitigating-embedding-and-class-assignment","title":"Mitigating Embedding and Class Assignment Mismatch in Unsupervised Image Classification","arxiv_id":null,"date":"2020-08-01","proceeding":"ECCV 2020 8","authors":["Sungwon Han","Sungwon Park","Sungkyu Park","Sundong Kim","Meeyoung Cha"],"abstract":"Unsupervised image classification is a challenging computer vision task. Deep learning-based algorithms have achieved superb results, where the latest approach adopts unified losses from embedding and class assignment processes. Since these processes inherently have different goals, jointly optimizing them may lead to a suboptimal solution. To address this limitation, we propose a novel two-stage algorithm in which an embedding module for pretraining precedes a refining module that concurrently performs embedding and class assignment. Our model outperforms SOTA when tested with multiple datasets, by substantially high accuracy of 81.0% for the CIFAR-10 dataset (i.e., increased by 19.3 percent points), 35.3% accuracy for CIFAR-100-20 (9.6 pp) and 66.5% accuracy for STL-10 (6.9 pp) in unsupervised tasks.","url_abs":"https://www.ecva.net/papers/eccv_2020/papers_ECCV/html/4802_ECCV_2020_paper.php","url_pdf":"https://www.ecva.net/papers/eccv_2020/papers_ECCV/papers/123690749.pdf","source":{"archive":"pwc-archive (Hugging Face), CC BY-SA 4.0","snapshot":"2025-07-28","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","row_kind":"abstracts"},"code_links":[{"paper_slug":"mitigating-embedding-and-class-assignment","repo_url":"https://github.com/dscig/TwoStageUC","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":0,"framework":"pytorch","reach":null}],"tasks":[{"task_slug":"classification-1","task_name":"Classification"},{"task_slug":"classification","task_name":"General Classification"},{"task_slug":"image-classification","task_name":"Image Classification"},{"task_slug":"image-clustering","task_name":"Image Clustering"},{"task_slug":"unsupervised-image-classification","task_name":"Unsupervised Image Classification"},{"task_slug":"image-classification","task_name":"image-classification"}],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[{"leaderboard":"/sota/image-clustering-on-cifar-10","task":"Image Clustering","dataset":"CIFAR-10","model":"TSUC","rank_in_archive_order":28,"of":40,"metrics":{"ARI":"-","Accuracy":"0.81","Backbone":"ResNet-18","NMI":"-","Train set":"Train"},"uses_additional_data":true},{"leaderboard":"/sota/image-clustering-on-cifar-100","task":"Image Clustering","dataset":"CIFAR-100","model":"TSUC","rank_in_archive_order":22,"of":30,"metrics":{"Accuracy":"0.353"},"uses_additional_data":true},{"leaderboard":"/sota/image-clustering-on-stl-10","task":"Image Clustering","dataset":"STL-10","model":"TSUC","rank_in_archive_order":21,"of":29,"metrics":{"Accuracy":"0.665","Backbone":"ResNet-18"},"uses_additional_data":true},{"leaderboard":"/sota/unsupervised-image-classification-on-cifar-10","task":"Unsupervised Image Classification","dataset":"CIFAR-10","model":"TSUC","rank_in_archive_order":7,"of":9,"metrics":{"Accuracy":"81.0"},"uses_additional_data":true},{"leaderboard":"/sota/unsupervised-image-classification-on-cifar-20","task":"Unsupervised Image Classification","dataset":"CIFAR-20","model":"TSUC","rank_in_archive_order":13,"of":14,"metrics":{"Accuracy":"35.3"},"uses_additional_data":false},{"leaderboard":"/sota/unsupervised-image-classification-on-stl-10","task":"Unsupervised Image Classification","dataset":"STL-10","model":"TSUC","rank_in_archive_order":7,"of":9,"metrics":{"Accuracy":"66.50"},"uses_additional_data":false}],"syntology":{"atlas_url":null,"mcp":null,"developers":"https://syntology.ai/developers"},"arxiv_metadata":null,"syntology_extracted_results":null}