Skip to main navigation Skip to search Skip to main content

Prompt-Guided Community Search Under Extreme Few-Shot Supervision

  • Beijing Institute of Technology

Research output: Chapter in Book/Report/Conference proceeding › Conference contribution › peer-review

Abstract

Community search, which aims to identify cohesive subgraphs containing given query nodes, is a fundamental task in graph analysis. Existing learning-based methods depend heavily on abundant labeled communities, making them ineffective in realistic settings where labels are extreme scarce. Although several methods have been proposed to mitigate label scarcity, they still fail to perform robustly across both sparsely and densely labeled datasets due to three key limitations: (i) a query-nodecentric design that always treats the query node as the community center, causing errors when the query lies near the community boundary; (ii) lack of connectivity guarantee, as similarity-based expansions may yield disconnected results; and (iii) dependence on labeled data, where unsupervised models lack supervision while semi-supervised ones easily overfit when labels are rare. To address these issues, we propose Prompt-Guided Community Search (PGCS), a novel 'pre-train, prompt, and search' framework. We design novel pretraining tasks that capture structural characteristics such as density and cohesiveness. We introduce a prompt function that measures the goodness of subgraph expansion, effectively bridging pretrained knowledge with community search. To overcome label scarcity, we propose a dual-prompt mix-training strategy, which utilize both ground truth communities and soft labels generated from unlabeled nodes. Finally, we develop an online search algorithm that expands from the query node into a coherent and connected community. Extensive experiments on nine real-world datasets show that PGCS consistently outperforms traditional and recent learning-based baselines in both accuracy and efficiency. On average, our proposed method outperforms SOTAs by up to 35.16% across all datasets.

Original languageEnglish
Title of host publicationProceedings - 2026 IEEE 42nd International Conference on Data Engineering, ICDE 2026
EditorsKhuzaima Daudjee, Qiong Luo, Alkis Simitsis
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages3053-3066
Number of pages14
ISBN (Electronic)9798331583651
DOIs
Publication statusPublished - 2026
Externally publishedYes
Event42nd International Conference on Data Engineering, ICDE 2026 - Montreal, Canada
Duration: 4 May 2026 → …

Publication series

NameProceedings - 2026 IEEE 42nd International Conference on Data Engineering, ICDE 2026

Conference

Conference42nd International Conference on Data Engineering, ICDE 2026
Country/TerritoryCanada
CityMontreal
Period4/05/26 → …

Keywords

  • Community Search
  • Data Mining
  • Graph Prompt Learning
  • Social Network

Fingerprint

Dive into the research topics of 'Prompt-Guided Community Search Under Extreme Few-Shot Supervision'. Together they form a unique fingerprint.

Cite this