arXiv:2512.00524cs.LGstat.ML2025-12AAAI被引 3

用双曲空间优化层次聚类,提升结构感知能力

Hyperbolic Continuous Structural Entropy for Hierarchical Clustering

  • 将数据映射到双曲空间,通过连续结构熵优化聚类
  • 在7个数据集上表现优于现有方法,最高提升6.2%
  • 适合需要精细层次结构建模的任务

层次聚类是将数据点分组为树状结构的基础机器学习技术。现有方法主要面临两大挑战:一是大多方法缺乏全局目标来定义树状结构;二是基于图的方法常忽略图结构的重要性,在完整或预定义的静态图上优化目标。本文提出双曲连续结构熵神经网络(HypCSE),实现增强结构的连续层次聚类。核心思想是在双曲空间中映射数据点,并在结构增强图上最小化松弛后的连续结构熵(CSE)。具体地,使用双曲图神经网络编码图顶点,最小化基于图嵌入的近似结构熵。为使目标可微,我们通过树结构中的最近公共祖先(LCA)重构该目标,并借助双曲嵌入与划分树的类比将其松弛为连续结构熵。为确保图结构有效捕捉数据点的层次关系,采用图结构学习(GSL)策略,在训练中动态更新图结构。在7个数据集上的大量实验表明,HypCSE表现出色,显著优于现有方法。

原文摘要 · Abstract (English)

Hierarchical clustering is a fundamental machine-learning technique for grouping data points into dendrograms. However, existing hierarchical clustering methods encounter two primary challenges: 1) Most methods specify dendrograms without a global objective. 2) Graph-based methods often neglect the significance of graph structure, optimizing objectives on complete or static predefined graphs. In this work, we propose Hyperbolic Continuous Structural Entropy neural networks, namely HypCSE, for structure-enhanced continuous hierarchical clustering. Our key idea is to map data points in the hyperbolic space and minimize the relaxed continuous structural entropy (SE) on structure-enhanced graphs. Specifically, we encode graph vertices in hyperbolic space using hyperbolic graph neural networks and minimize approximate SE defined on graph embeddings. To make the SE objective differentiable for optimization, we reformulate it into a function using the lowest common ancestor (LCA) on trees and then relax it into continuous SE (CSE) by the analogy of hyperbolic graph embeddings and partitioning trees. To ensure a graph structure that effectively captures the hierarchy of data points for CSE calculation, we employ a graph structure learning (GSL) strategy that updates the graph structure during training. Extensive experiments on seven datasets demonstrate the superior performance of HypCSE.

层次聚类双曲空间结构熵

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