Pre-trained Prompt-driven Semi-supervised Local Community Detection

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Ni, Li, Xu, Hengkai, Mu, Lin, Zhang, Yiwen, Luo, Wenjian
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910974745772032
author Ni, Li
Xu, Hengkai
Mu, Lin
Zhang, Yiwen
Luo, Wenjian
author_facet Ni, Li
Xu, Hengkai
Mu, Lin
Zhang, Yiwen
Luo, Wenjian
contents Semi-supervised local community detection aims to leverage known communities to detect the community containing a given node. Although existing semi-supervised local community detection studies yield promising results, they suffer from time-consuming issues, highlighting the need for more efficient algorithms. Therefore, we apply the "pre-train, prompt" paradigm to semi-supervised local community detection and propose the Pre-trained Prompt-driven Semi-supervised Local community detection method (PPSL). PPSL consists of three main components: node encoding, sample generation, and prompt-driven fine-tuning. Specifically, the node encoding component employs graph neural networks to learn the representations of nodes and communities. Based on representations of nodes and communities, the sample generation component selects known communities that are structurally similar to the local structure of the given node as training samples. Finally, the prompt-driven fine-tuning component leverages these training samples as prompts to guide the final community prediction. Experimental results on five real-world datasets demonstrate that PPSL outperforms baselines in both community quality and efficiency.
format Preprint
id arxiv_https___arxiv_org_abs_2505_12304
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Pre-trained Prompt-driven Semi-supervised Local Community Detection
Ni, Li
Xu, Hengkai
Mu, Lin
Zhang, Yiwen
Luo, Wenjian
Social and Information Networks
Artificial Intelligence
Semi-supervised local community detection aims to leverage known communities to detect the community containing a given node. Although existing semi-supervised local community detection studies yield promising results, they suffer from time-consuming issues, highlighting the need for more efficient algorithms. Therefore, we apply the "pre-train, prompt" paradigm to semi-supervised local community detection and propose the Pre-trained Prompt-driven Semi-supervised Local community detection method (PPSL). PPSL consists of three main components: node encoding, sample generation, and prompt-driven fine-tuning. Specifically, the node encoding component employs graph neural networks to learn the representations of nodes and communities. Based on representations of nodes and communities, the sample generation component selects known communities that are structurally similar to the local structure of the given node as training samples. Finally, the prompt-driven fine-tuning component leverages these training samples as prompts to guide the final community prediction. Experimental results on five real-world datasets demonstrate that PPSL outperforms baselines in both community quality and efficiency.
title Pre-trained Prompt-driven Semi-supervised Local Community Detection
topic Social and Information Networks
Artificial Intelligence
url https://arxiv.org/abs/2505.12304