The Heterogeneous Network Community Detection Model Based on Self-Attention

With the advancement of representation learning, graph representation learning has gained significant attention in the field of community detection for heterogeneous networks. A prominent approach in this domain involves the use of meta-paths to capture higher-order relationships between nodes, particularly when bidirectional or reciprocal relationships exist. However, defining effective meta-paths often requires substantial domain expertise. Moreover, these methods typically depend on additional clustering algorithms, which can limit their practical applicability. To address these challenges, context paths have been introduced as an alternative to meta-paths. When combined with a self-attention mechanism, models can dynamically assess the relative importance of different context paths. By leveraging the inherent symmetry within context paths, models enhance their ability to capture balanced relationships between nodes, thereby improving their representation of complex interactions. Building on this idea, we propose BP-GCN, a self-attention-based model for heterogeneous community detection. BP-GCN autonomously identifies node relationships within symmetric context paths, significantly improving community detection accuracy. Furthermore, the model integrates the Bernoulli–Poisson framework to establish an end-to-end detection system that eliminates the need for auxiliary clustering algorithms. Extensive experiments on multiple real-world datasets demonstrate that BP-GCN consistently outperforms existing benchmark methods.