arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2302.02450cs.LG
收藏

基于模型的聚类中的正则化与优化

Regularization and Optimization in Model-Based Clustering

PSR;Pontifícia Universidade Católica do Rio de Janeiro (PUC-Rio)(里约热内卢天主教大学(PUC-Rio));LAMPS;CIRRELT;SCALE-AI Chair in Data-Driven Supply Chains(SCALE-AI数据驱动供应链讲席);Polytechnique Montréal(蒙特利尔理工学院)

Raphael Araujo Sampaio, Joaquim Dias Garcia, Marcus Poggi, Thibaut Vidal

AI总结:

针对k-means变体拟合能力有限及一般GMM优化难、易过拟合的问题,提出结合正则化与优化策略,显著提升聚类恢复性能,并开源代码与Julia包。

摘要:

由于其概念上的简单性,k-means算法变体已被广泛用于无监督聚类分析。然而,这些算法的一个主要缺点是,它们本质上将相同球形高斯的混合分布拟合到与这种分布大相径庭的数据上。相比之下,一般的高斯混合模型(GMMs)可以拟合更丰富的结构,但需要估计每个簇的二次数量的参数来表示协方差矩阵。这带来了两个主要问题:(i)由于局部最小值数量较多,底层优化问题具有挑战性;(ii)它们的解可能过拟合数据。在这项工作中,我们设计了规避这两个问题的搜索策略。我们为一般GMMs开发了更有效的优化算法,并将这些算法与避免过拟合的正则化策略相结合。通过广泛的计算分析,我们观察到单独的优化或正则化并不能显著改善簇恢复。然而,结合这些技术可以实现k-means算法变体先前无法达到的全新性能水平,揭示出截然不同的簇结构。这些结果对GMM和k-means方法之间的现状提供了新的见解,并建议更频繁地使用一般GMMs进行数据探索。为促进此类应用,我们提供了开源代码以及实现所提出技术的Julia包(UnsupervisedClustering.jl和RegularizedCovarianceMatrices.jl)。

英文摘要:

Due to their conceptual simplicity, k-means algorithm variants have been extensively used for unsupervised cluster analysis. However, one main shortcoming of these algorithms is that they essentially fit a mixture of identical spherical Gaussians to data that vastly deviates from such a distribution. In comparison, general Gaussian Mixture Models (GMMs) can fit richer structures but require estimating a quadratic number of parameters per cluster to represent the covariance matrices. This poses two main issues: (i) the underlying optimization problems are challenging due to their larger number of local minima, and (ii) their solutions can overfit the data. In this work, we design search strategies that circumvent both issues. We develop more effective optimization algorithms for general GMMs, and we combine these algorithms with regularization strategies that avoid overfitting. Through extensive computational analyses, we observe that optimization or regularization in isolation does not substantially improve cluster recovery. However, combining these techniques permits a completely new level of performance previously unachieved by k-means algorithm variants, unraveling vastly different cluster structures. These results shed new light on the current status quo between GMM and k-means methods and suggest the more frequent use of general GMMs for data exploration. To facilitate such applications, we provide open-source code as well as Julia packages (UnsupervisedClustering.jl and RegularizedCovarianceMatrices.jl) implementing the proposed techniques.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑