Consensus Clustering With Unsupervised Representation Learning
Autor: | Jayanth Regatti, Aniket Anand Deshmukh, Eren Manavoglu, Urun Dogan |
---|---|
Rok vydání: | 2020 |
Předmět: |
FOS: Computer and information sciences
Computer Science - Machine Learning Artificial neural network Contextual image classification Computer science business.industry Computer Vision and Pattern Recognition (cs.CV) Computer Science - Computer Vision and Pattern Recognition Machine learning computer.software_genre Ensemble learning Machine Learning (cs.LG) Statistical classification ComputingMethodologies_PATTERNRECOGNITION Consensus clustering Artificial intelligence Representation (mathematics) business Cluster analysis Feature learning computer |
Zdroj: | IJCNN |
DOI: | 10.48550/arxiv.2010.01245 |
Popis: | Recent advances in deep clustering and unsupervised representation learning are based on the idea that different views of an input image (generated through data augmentation techniques) must either be closer in the representation space, or have a similar cluster assignment. Bootstrap Your Own Latent (BYOL) is one such representation learning algorithm that has achieved state-of-the-art results in self-supervised image classification on ImageNet under the linear evaluation protocol. However, the utility of the learnt features of BYOL to perform clustering is not explored. In this work, we study the clustering ability of BYOL and observe that features learnt using BYOL may not be optimal for clustering. We propose a novel consensus clustering based loss function, and train BYOL with the proposed loss in an end-to-end way that improves the clustering ability and outperforms similar clustering based methods on some popular computer vision datasets. Comment: Accepted by the 2021 International Joint Conference on Neural Networks (IJCNN 2021) |
Databáze: | OpenAIRE |
Externí odkaz: |