arXiv:2506.05495cs.DScs.LG2025-06ICML被引 10

用外接查询增强聚类,突破传统算法的性能瓶颈。

Learning-Augmented Hierarchical Clustering

  • 引入分裂预言机,指导聚类树构建过程。
  • 在多项式时间内实现对Dasgupta目标的常数倍近似。
  • 适合需要高精度聚类且可获取辅助信息的场景。

层次聚类(HC)是一种重要数据分析技术,旨在递归划分数据集形成树状结构,使每一层的相似数据点被分组。然而,许多现有HC目标存在逼近算法的理论障碍。本文考虑在自然预言机提供的辅助信息下进行层次聚类,聚焦于一种‘分裂预言机’:当给定三个顶点 (u,v,w) 时,可回答在最优树中包含三者的最低公共祖先所涉及的顶点对,即识别哪个顶点‘分离出去’。利用该预言机,我们获得以下结果:一个多项式时间算法,输出对Dasgupta目标的O(1)近似解;一个近线性时间算法,输出对Moseley-Wang目标的(1−o(1))近似解。在合理的小集合扩张假设下,任何多项式时间算法都无法对Dasgupta目标达到常数近似,也无法对Moseley-Wang目标达到(1−C)近似(C>0为常数)。因此,我们的结果表明,分裂预言机使算法能突破标准方法的性能极限。此外,我们的方法可扩展至子线性设置,首次提出具有更优保证的流式与PRAM层次聚类算法。

原文摘要 · Abstract (English)

Hierarchical clustering (HC) is an important data analysis technique in which the goal is to recursively partition a dataset into a tree-like structure while grouping together similar data points at each level of granularity. Unfortunately, for many of the proposed HC objectives, there exist strong barriers to approximation algorithms with the hardness of approximation. Thus, we consider the problem of hierarchical clustering given auxiliary information from natural oracles. Specifically, we focus on a *splitting oracle* which, when provided with a triplet of vertices $(u,v,w)$, answers (possibly erroneously) the pairs of vertices whose lowest common ancestor includes all three vertices in an optimal tree, i.e., identifying which vertex ``splits away'' from the others. Using such an oracle, we obtain the following results: - A polynomial-time algorithm that outputs a hierarchical clustering tree with $O(1)$-approximation to the Dasgupta objective (Dasgupta [STOC'16]). - A near-linear time algorithm that outputs a hierarchical clustering tree with $(1-o(1))$-approximation to the Moseley-Wang objective (Moseley and Wang [NeurIPS'17]). Under the plausible Small Set Expansion Hypothesis, no polynomial-time algorithm can achieve any constant approximation for Dasgupta's objective or $(1-C)$-approximation for the Moseley-Wang objective for some constant $C>0$. As such, our results demonstrate that the splitting oracle enables algorithms to outperform standard HC approaches and overcome hardness constraints. Furthermore, our approaches extend to sublinear settings, in which we show new streaming and PRAM algorithms for HC with improved guarantees.

层次聚类近似算法预言机子线性

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