Adaptive Consensus Clustering for Multiple K-Means Via Base Results Refining

Peng Zhou,Liang Du,Xuejun Li
DOI: https://doi.org/10.1109/tkde.2023.3264970
IF: 9.235
2023-01-01
IEEE Transactions on Knowledge and Data Engineering
Abstract:Consensus clustering, which learns a consensus clustering result from multiple weak base results, has been widely studied. However, conventional consensus clustering methods only focus on the ensemble process while ignoring the quality improvement of the base results, and thus they just use the fixed base results for consensus learning. In this paper, we provide an alternative idea to improve the final consensus clustering performance by considering the base results refining. In our framework, we adaptively refine the base results in the process of the ensemble. In more detail, on one hand, we ensemble multiple K-means results to learn the consensus one by considering the consensus and diversity; on the other hand, we apply the consensus result to design a graph filter to learn a more cluster-friendly embedding for refining the base K-means results. In our framework, the consensus learning and base results refining are integrated into one unified objective function so that these two tasks can be boosted by each other. Then we design an effective iterative algorithm to optimize the carefully designed objective function. The extensive experiments on benchmark data sets demonstrate that the proposed method can outperform both the single clustering and the state-of-the-art consensus clustering methods. The codes of this paper are released in http://Doctor-Nobody.github.io/codes/ACMK.zip .
What problem does this paper attempt to address?