TY - GEN
T1 - Prompt-Guided Community Search Under Extreme Few-Shot Supervision
AU - Yang, Wenxin
AU - Feng, Kaiyu
AU - Fang, Lanting
AU - Zhao, Kangfei
AU - Wu, Xia
N1 - Publisher Copyright:
© 2026 IEEE.
PY - 2026
Y1 - 2026
N2 - 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.
AB - 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.
KW - Community Search
KW - Data Mining
KW - Graph Prompt Learning
KW - Social Network
UR - https://www.scopus.com/pages/publications/105048958447
U2 - 10.1109/ICDE65706.2026.00227
DO - 10.1109/ICDE65706.2026.00227
M3 - Conference contribution
AN - SCOPUS:105048958447
T3 - Proceedings - 2026 IEEE 42nd International Conference on Data Engineering, ICDE 2026
SP - 3053
EP - 3066
BT - Proceedings - 2026 IEEE 42nd International Conference on Data Engineering, ICDE 2026
A2 - Daudjee, Khuzaima
A2 - Luo, Qiong
A2 - Simitsis, Alkis
PB - Institute of Electrical and Electronics Engineers Inc.
T2 - 42nd International Conference on Data Engineering, ICDE 2026
Y2 - 4 May 2026
ER -