Graph-structured data is essential in numerous domains. The quality of node representations plays a crucial role in graph mining tasks. Random-walk-based techniques, such as DeepWalk (DW) [1], are effective for this task. However, the embeddings they generate often exhibit noise and redundancy. As a result, clustering efforts may face potential obstacles. To tackle this issue, we introduce the DWACE (DeepWalk Autoencoder Contrastive Enhancement) framework. This proposal enhances these embeddings, thus increasing clustering efficacy. DWACE employs a nonlinear autoencoder to denoise and compress the raw embeddings while preserving their structural integrity. It additionally integrates a contrastive learning objective to preserve the consistency of embeddings and enhance intra-community coherence. Our investigations on five benchmark datasets demonstrate that DWACE surpasses baseline approaches, attaining state-of-the-art performance in clustering.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

DWACE: Enhancing Graph Clustering via Autoencoder and Contrastive Learning Refinement of DeepWalk Embeddings

  • Nguyen Xuan Thao Mai,
  • Cong Phap Huynh,
  • Supeeti Kunchan,
  • Dai Tho Dang

摘要

Graph-structured data is essential in numerous domains. The quality of node representations plays a crucial role in graph mining tasks. Random-walk-based techniques, such as DeepWalk (DW) [1], are effective for this task. However, the embeddings they generate often exhibit noise and redundancy. As a result, clustering efforts may face potential obstacles. To tackle this issue, we introduce the DWACE (DeepWalk Autoencoder Contrastive Enhancement) framework. This proposal enhances these embeddings, thus increasing clustering efficacy. DWACE employs a nonlinear autoencoder to denoise and compress the raw embeddings while preserving their structural integrity. It additionally integrates a contrastive learning objective to preserve the consistency of embeddings and enhance intra-community coherence. Our investigations on five benchmark datasets demonstrate that DWACE surpasses baseline approaches, attaining state-of-the-art performance in clustering.