构建工业级图聚类评测基准,解决学术与应用脱节问题
Bridging Academia and Industry: A Comprehensive Benchmark for Attributed Graph Clustering
- 统一编码-聚类-优化框架,支持大规模小批量训练
- 涵盖2.7K至1.11亿节点的12个真实工业数据集
- 强调无监督指标与效率评估,适合工业部署研究
属性图聚类(AGC)是融合图结构与节点属性以发现隐含模式的基础无监督任务,在欺诈检测和用户分群等工业场景中意义重大。然而,当前学术研究与实际部署之间存在显著鸿沟:评估体系受限于小规模、高同质性引用数据集,采用不可扩展的全批训练,且依赖有监督指标,无法反映标签稀缺环境下的真实性能。为此,我们提出PyAGC——一个面向生产环境的综合性评测基准与工具库,可对各类AGC方法在多样尺度与结构特性下进行压力测试。该框架统一现有方法为模块化编码-聚类-优化流程,并首次提供多种主流AGC算法的内存高效小批量实现。基准包含12个多样化数据集,节点数从2.7K到1.11M不等,特别纳入具有复杂表格特征和低同质性的工业图数据。同时倡导全面评估协议,要求结合无监督结构指标与效率分析,超越传统有监督评价。该基准已在蚂蚁集团高风险工业流程中实战验证,为社区提供可复现、可扩展的研究平台,推动AGC向真实部署演进。代码与资源已通过GitHub、PyPI及文档网站公开。
原文摘要 · Abstract (English)
Attributed Graph Clustering (AGC) is a fundamental unsupervised task that integrates structural topology and node attributes to uncover latent patterns in graph-structured data. Despite its significance in industrial applications such as fraud detection and user segmentation, a significant chasm persists between academic research and real-world deployment. Current evaluation protocols suffer from the small-scale, high-homophily citation datasets, non-scalable full-batch training paradigms, and a reliance on supervised metrics that fail to reflect performance in label-scarce environments. To bridge these gaps, we present PyAGC, a comprehensive, production-ready benchmark and library designed to stress-test AGC methods across diverse scales and structural properties. We unify existing methodologies into a modular Encode-Cluster-Optimize framework and, for the first time, provide memory-efficient, mini-batch implementations for a wide array of state-of-the-art AGC algorithms. Our benchmark curates 12 diverse datasets, ranging from 2.7K to 111M nodes, specifically incorporating industrial graphs with complex tabular features and low homophily. Furthermore, we advocate for a holistic evaluation protocol that mandates unsupervised structural metrics and efficiency profiling alongside traditional supervised metrics. Battle-tested in high-stakes industrial workflows at Ant Group, this benchmark offers the community a robust, reproducible, and scalable platform to advance AGC research towards realistic deployment. The code and resources are publicly available via GitHub (https://github.com/Cloudy1225/PyAGC), PyPI (https://pypi.org/project/pyagc), and Documentation (https://pyagc.readthedocs.io).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。