A4 Vertaisarvioitu artikkeli konferenssijulkaisussa

Ensemble of Deep Variational Mixture Models for Unsupervised Clustering




TekijätTan, Xu; Chen, Junqi; Yang, Jiawei; Rahardja, Sylwan; Wang, Mou; Rahardja, Susanto

ToimittajaN/A

Konferenssin vakiintunut nimiIEEE International Conference on Image Processing

Julkaisuvuosi2024

JournalIEEE International Conference on Image Processing

Kokoomateoksen nimi2024 IEEE International Conference on Image Processing (ICIP)

Aloitussivu807

Lopetussivu813

ISBN979-8-3503-4940-5

eISBN979-8-3503-4939-9

ISSN1522-4880

eISSN2381-8549

DOIhttps://doi.org/10.1109/ICIP51287.2024.10647931

Verkko-osoitehttps://ieeexplore.ieee.org/document/10647931


Tiivistelmä

Deep variational mixture models (DVMMs) have demonstrated promising performance in unsupervised clustering for complicated high-dimensional data such as images. However, their prediction accuracy is often unstable and significantly influenced by randomness, particularly during the initialization of parameters. To reduce this uncertainty, we propose an ensemble approach that combines the predictions of multiple base models. Specifically, we introduce two individual ensemble strategies: voting and merging. In the voting strategy, the final label is determined by selecting the predicted class label with the most votes and lowest Shannon entropy. In the merging strategy, the class probability vectors (scaled by the temperature parameter) from different models are combined to predict the final class label. Experimental results on two image datasets demonstrate that these proposed methods yield reliable and superior clustering performance.



Last updated on 2025-13-02 at 10:10