arXiv:2503.07643cs.LGcs.AI2025-03被引 1

用图神经网络增强空间聚类,提升效率与精度。

ConstellationNet: Reinventing Spatial Clustering through GNNs

  • 融合CNN与GNN,利用邻域信息聚合提升聚类效果。
  • 在多个数据集上超越现有方法,性能提升10倍,训练时间减少十倍。
  • 适合医学影像、流行病学等需快速响应的高维数据场景。

空间聚类在犯罪学、病理学和城市规划等领域具有广泛应用。然而,多数现有算法无法有效利用邻近节点信息,在高维和大规模数据下性能显著下降。随着数据规模与维度持续增长,传统聚类方法难以应对复杂问题。为此,我们提出ConstellationNet,一种结合卷积神经网络(CNN)嵌入能力、图神经网络(GNN)邻域聚合特性及批处理优势的新型框架,通过图增强预测实现更优的空间聚类与分类。该模型在多个数据集上达到当前最优性能,相较于基准方法,分类与聚类表现提升10倍,模型规模缩小至十分之一,训练时间减少十倍。凭借高效训练与强大泛化能力,ConstellationNet在流行病学与医学影像等场景中具备快速部署潜力。

原文摘要 · Abstract (English)

Spatial clustering is a crucial field, finding universal use across criminology, pathology, and urban planning. However, most spatial clustering algorithms cannot pull information from nearby nodes and suffer performance drops when dealing with higher dimensionality and large datasets, making them suboptimal for large-scale and high-dimensional clustering. Due to modern data growing in size and dimension, clustering algorithms become weaker when addressing multifaceted issues. To improve upon this, we develop ConstellationNet, a convolution neural network(CNN)-graph neural network(GNN) framework that leverages the embedding power of a CNN, the neighbor aggregation of a GNN, and a neural network's ability to deal with batched data to improve spatial clustering and classification with graph augmented predictions. ConstellationNet achieves state-of-the-art performance on both supervised classification and unsupervised clustering across several datasets, outperforming state-of-the-art classification and clustering while reducing model size and training time by up to tenfold and improving baselines by 10 times. Because of its fast training and powerful nature, ConstellationNet holds promise in fields like epidemiology and medical imaging, able to quickly train on new data to develop robust responses.

空间聚类图神经网络医学影像

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