On hyperparameter tuning in general clustering problemsm
Xinjie Fan, Yuguang Yue, Purnamrita Sarkar, Y. X. Rachel Wang
Abstract
Tuning hyperparameters for unsupervised learning problems is difficult in general due to the lack of ground truth for validation. However, the success of most clustering methods depends heavily on the correct choice of the involved hyperparameters. Take for example the Lagrange multipliers of penalty terms in semidefinite programming (SDP) relaxations of community detection in networks, or the bandwidth parameter needed in the Gaussian kernel used to construct similarity matrices for spectral clustering. Despite the popularity of these clustering algorithms, there are not many provable methods for tuning these hyperparameters. In this paper, we provide an overarching framework with provable guarantees for tuning hyperparameters in the above class of problems under two different models. Our framework can be augmented with a cross validation procedure to do model selection as well. In a variety of simulation and real data experiments, we show that our framework outperforms other widely used tuning procedures in a broad range of parameter settings.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Related papers
- Hypergraph Modeling via Spectral Embedding Connection: Hypergraph Cut, Weighted Kernel k-Means, and Heat KernelShota SaitoAAAI 2022 · 6 citations
- Likelihood Adjusted Semidefinite Programs for Clustering Heterogeneous DataYubo Zhuang, Xiaohui Chen, Yun YangICML 2023 · 2 citations
- Random Matrix Analysis to Balance between Supervised and Unsupervised Learning under the Low Density Separation AssumptionVasilii Feofanov, Malik Tiomoko, Aladin VirmauxICML 2023 · 8 citations
- A Simple Approach to Automated Spectral ClusteringJicong Fan, Yiheng Tu, Zhao Zhang, Mingbo Zhao et al.NeurIPS 2022 · 35 citations
- On the Robustness of Spectral Algorithms for Semirandom Stochastic Block ModelsAditya Bhaskara, Agastya Vibhuti Jha, Michael Kapralov, Naren Manoj et al.NeurIPS 2024 · 3 citations
