arXiv:2503.07227cs.LGcs.DS2025-03ICLR被引 1

用核心集加速谱聚类,提升大规模稀疏图聚类效率与精度。

Coreset Spectral Clustering

  • 基于核k均值与归一化割的关联,构建核心集图进行聚类。
  • 核心集上α近似解可保证原图上O(α)近似,理论保证强。
  • 适用于大规模稀疏图,尤其在多簇场景下比现有方法快得多。

核心集已成为在大数据集上解决小簇数k均值和核k均值聚类问题的重要工具。而谱聚类在稀疏图上表现良好,并已扩展至高效处理大量簇。本文利用核k均值与归一化割问题之间的联系,结合两者优势,提出一种针对图的核核心集谱聚类算法:通过聚类核心集图来推断原始图的良好标签。证明了核心集图上的α近似解,在原图上为O(α)近似。同时,将现有核k均值核心集算法在稀疏核下的运行时间从˜O(nk)改进为˜O(n·min{k, d_avg}),其中d_avg为n×n核矩阵每行非零元素的平均数。实验表明,该算法在大型真实世界图上具有渐进更快的运行速度,并克服了核核心集k均值在稀疏核下易陷入局部最优的主要挑战。

原文摘要 · Abstract (English)

Coresets have become an invaluable tool for solving $k$-means and kernel $k$-means clustering problems on large datasets with small numbers of clusters. On the other hand, spectral clustering works well on sparse graphs and has recently been extended to scale efficiently to large numbers of clusters. We exploit the connection between kernel $k$-means and the normalised cut problem to combine the benefits of both. Our main result is a coreset spectral clustering algorithm for graphs that clusters a coreset graph to infer a good labelling of the original graph. We prove that an $α$-approximation for the normalised cut problem on the coreset graph is an $O(α)$-approximation on the original. We also improve the running time of the state-of-the-art coreset algorithm for kernel $k$-means on sparse kernels, from $\tilde{O}(nk)$ to $\tilde{O}(n\cdot \min \{k, d_{avg}\})$, where $d_{avg}$ is the average number of non-zero entries in each row of the $n\times n$ kernel matrix. Our experiments confirm our coreset algorithm is asymptotically faster on large real-world graphs with many clusters, and show that our clustering algorithm overcomes the main challenge faced by coreset kernel $k$-means on sparse kernels which is getting stuck in local optima.

谱聚类核心集稀疏图聚类优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。