Contrastive Deep Nonnegative Matrix Factorization for Community Detection

Yuecheng Li,Jialong Chen,Chuan Chen,Lei Yang,Zibin Zheng

2024-02-14

Abstract:Recently, nonnegative matrix factorization (NMF) has been widely adopted for community detection, because of its better interpretability. However, the existing NMF-based methods have the following three problems: 1) they directly transform the original network into community membership space, so it is difficult for them to capture the hierarchical information; 2) they often only pay attention to the topology of the network and ignore its node attributes; 3) it is hard for them to learn the global structure information necessary for community detection. Therefore, we propose a new community detection algorithm, named Contrastive Deep Nonnegative Matrix Factorization (CDNMF). Firstly, we deepen NMF to strengthen its capacity for information extraction. Subsequently, inspired by contrastive learning, our algorithm creatively constructs network topology and node attributes as two contrasting views. Furthermore, we utilize a debiased negative sampling layer and learn node similarity at the community level, thereby enhancing the suitability of our model for community detection. We conduct experiments on three public real graph datasets and the proposed model has achieved better results than state-of-the-art methods. Code available at <a class="link-external link-https" href="https://github.com/6lyc/CDNMF.git" rel="external noopener nofollow">this https URL</a>.

Machine Learning,Artificial Intelligence,Social and Information Networks

What problem does this paper attempt to address?

### Problems the paper attempts to solve This paper attempts to solve several key problems in community detection (CD). Specifically, the existing methods based on nonnegative matrix factorization (NMF) have the following three main problems: 1. **Difficulty in extracting hierarchical information**: - Existing NMF methods usually perform single - or double - layer mapping only, directly transforming from the original network to the community member space. This makes it difficult for them to capture complex hierarchical information because networks in the real world have complex organizational principles. 2. **Ignoring node attributes**: - Existing NMF methods tend to focus only on the topological structure of the network and ignore the attributes of nodes. In social sciences, it has been proven that node attributes can reflect and influence their community structure. 3. **Difficulty in capturing global structural information**: - Existing NMF methods mainly focus on the similarity between neighboring nodes and are difficult to extract the global structural information that is crucial for community detection. To solve these problems, the author proposes a new community detection algorithm, called contrastive deep nonnegative matrix factorization (CDNMF). The main contributions of this algorithm are as follows: 1. **Using deep NMF to enhance the representation learning ability**: - By increasing the number of layers of NMF, the information extraction ability of the model is enhanced. 2. **Constructing a contrastive learning framework**: - Unifying the learning of graph topology and node attributes based on deep NMF. By filtering out false - negative samples through the de - biasing negative sampling layer, the model can better learn community - level information. 3. **Extensive experimental verification**: - Through experiments on three publicly available real - graph datasets, the superiority, effectiveness, and efficiency of CDNMF are verified. The experimental results show that this algorithm is superior to other state - of - the - art community detection methods. ### Summary The paper aims to solve the limitations of existing NMF methods in community detection by introducing deep NMF and a contrastive learning framework, thereby improving the performance of community detection.

Contrastive Deep Nonnegative Matrix Factorization for Community Detection

Structural Deep Nonnegative Matrix Factorization for Community Detection

Deep graph convolution neural network with non-negative matrix factorization for community discovery

Link Community Detection by Non-Negative Matrix Factorization with Multi-Step Similarities

Nonnegative Matrix Factorization Based on Node Centrality for Community Detection

A Survey of Community Detection in Complex Networks Using Nonnegative Matrix Factorization

Differentiated Graph Regularized Non-Negative Matrix Factorization for Semi-Supervised Community Detection

An Algorithm Based on Non-Negative Matrix Factorization for Detecting Communities in Networks

Community Detection Method Based on Robust Semi-Supervised Nonnegative Matrix Factorization

Community Detection Via Multihop Nonnegative Matrix Factorization.

Community Detection Approach Via Graph Regularized Non-Negative Matrix Factorization

Community Inference With Bayesian Non-Negative Matrix Factorization

Overlapping Community Detection Using Non-Negative Matrix Factorization With Orthogonal and Sparseness Constraints

Constrained Symmetric Non-Negative Matrix Factorization with Deep Autoencoders for Community Detection

High-order Order Proximity-Incorporated, Symmetry and Graph-Regularized Nonnegative Matrix Factorization for Community Detection

A Symmetry and Graph Regularized Nonnegative Matrix Factorization Model for Community Detection

Adaptive Community Detection Incorporating Topology and Content in Social Networks

Constraint-Induced Symmetric Nonnegative Matrix Factorization for Accurate Community Detection

A Network Embedding-Enhanced NMF Method for Finding Communities in Attributed Networks.

Multi-constraint non-negative matrix factorization for community detection: orthogonal regular sparse constraint non-negative matrix factorization

Symmetry and Graph Bi-Regularized Non-Negative Matrix Factorization for Precise Community Detection