Loading [a11y]/accessibility-menu.js
SDCG: Silhouette-based Deep Clustering with GNN for Improved Graph Node Clustering | IEEE Conference Publication | IEEE Xplore

SDCG: Silhouette-based Deep Clustering with GNN for Improved Graph Node Clustering


Abstract:

Graph Neural Networks (GNNs) are powerful tools for analyzing graph-structured data in various fields because of their great expressive power for graph data. They use a m...Show More

Abstract:

Graph Neural Networks (GNNs) are powerful tools for analyzing graph-structured data in various fields because of their great expressive power for graph data. They use a message-passing mechanism to update node embeddings, which are then used for tasks such as node classification and link prediction. Recently, node embeddings have also been used in research on graph node clustering, which aims to group similar nodes based on their features and graph topology. However, traditional methods for node clustering have a limitation in that GNNs only focus on generating node embeddings without considering the ultimate objective of clustering. To address this issue, a novel technique called "Deep Clustering" has been proposed, which integrates both node embedding and clustering stages. This requires defining a new loss function by simultaneously minimizing the GNN loss and the clustering loss. Our proposed loss function incorporates not only the distance within clusters but also the distance between clusters by applying the Silhouette coefficient, which enables us to achieve better clustering results. In this paper, we propose a Silhouette-based Deep Clustering with GNN (SDCG) to more effectively cluster nodes in a graph by iteratively training the embedding model to produce embedding vectors with improved clustering results. Through extensive experiments, we demonstrate that SDCG outperforms the conventional approach of performing embedding and clustering independently.
Date of Conference: 23-25 May 2023
Date Added to IEEE Xplore: 03 August 2023
ISBN Information:

ISSN Information:

Conference Location: Orlando, FL, USA

Funding Agency:


Contact IEEE to Subscribe

References

References is not available for this document.