arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2501.12946cs.SI
收藏

少即是多:基于图卷积网络的简单而有效的启发式社区检测

Less is More: Simple yet Effective Heuristic Community Detection with Graph Convolution Network

College of Physics and Information Engineering, Minnan Normal University(闽南师范大学物理与信息工程学院)

Hong Wang, Yinglong Zhang, Zhangqi Zhao, Zhicong Cai, Xuewen Xia, Xing Xu

AI总结:

本文提出一种无需数据增强和对比优化的简单社区检测算法,通过预检测提取全局结构、用GCN融合局部结构与属性特征,并结合模块度最大化实现自适应、高效准确的社区发现。

摘要:

社区检测在数据挖掘中至关重要。传统方法主要关注图结构,往往忽略属性特征的重要性。相比之下,基于深度学习的方法通过对比学习融合属性特征和局部结构信息,从而提升检测性能。然而,现有算法设计复杂且需要联合优化,导致训练困难并降低检测效率。此外,这些方法要求预先定义社区数量,使结果容易受到人为干扰。为解决上述问题,我们提出一种简单而有效的社区检测算法,该算法能够自适应地检测社区,无需依赖数据增强和对比优化。所提算法首先进行社区预检测,以自适应地提取全局结构信息;然后利用图卷积网络(GCN)融合局部结构和属性特征;随后在特征空间中结合全局结构、局部结构和属性特征来发现社区归属;最后采用模块度最大化方法,基于这三类信息对社区进行优化,从而揭示每个节点的社区归属。我们在多个图数据集上进行了实验比较,将所提算法与传统方法及最先进的社区检测算法进行对比评估。实验结果表明,我们的算法在检测速度和效果方面均取得了更高的效率和准确性。代码可在 https://github.com/wuanghoong/Less-is-More.git 获取。

英文摘要:

Community detection is crucial in data mining. Traditional methods primarily focus on graph structure, often neglecting the significance of attribute features. In contrast, deep learning-based approaches incorporate attribute features and local structural information through contrastive learning, improving detection performance. However, existing algorithms' complex design and joint optimization make them difficult to train and reduce detection efficiency. Additionally, these methods require the number of communities to be predefined, making the results susceptible to artificial interference. To address these challenges, we propose a simple yet effective community detection algorithm that can adaptively detect communities without relying on data augmentation and contrastive optimization. The proposed algorithm first performs community pre-detection to extract global structural information adaptively. It then utilizes GCN to integrate local structures and attribute features. Subsequently, it combines global, local structures and attribute features in the feature space to discover community affiliations. Finally, a modularity maximization method is employed to optimize the communities based on these three types of information, thereby uncovering the community affiliation of each node. We conduct experimental comparisons across various graph datasets, evaluating the proposed algorithm against traditional methods and state-of-the-art community detection algorithms. The experimental results demonstrate that our algorithm achieves greater efficiency and accuracy in terms of both detection speed and effectiveness. The code is available at https://github.com/wuanghoong/Less-is-More.git.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑