Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

GRAPHLCP: Structure-Aware Localized Conformal Prediction on Graphs

About

Conformal prediction (CP) provides a distribution-free approach to uncertainty quantification with finite-sample guarantees. However, applying CP to graph neural networks (GNNs) remains challenging as the combinatorial nature of graphs often leads to insufficiently certain predictions and indiscriminative embeddings. Existing methods primarily rely on embedding-space proximity for localization, which can be unreliable for graphs and yield inefficient prediction sets. We propose GRAPHLCP, a proximity-based localized CP framework that explicitly incorporates graph topology and inter-node dependencies into localization and weighting. Our approach introduces a feature-aware densification step to mitigate locality bias in sparse graphs, followed by a Personalized PageRank-based kernel computation to model structural proximity. This enables topology-dependent anchor sampling and calibration weighting that captures both local and long-range dependencies. Extensive experiments on several regression and classification datasets demonstrate that GRAPHLCP guarantees marginal coverage with finite samples while efficiently attaining favorable test conditional coverage across various conditioning scenarios.

Peyman Baghershahi, Fangxin Wang, Debmalya Mandal, Sourav Medya• 2026

Related benchmarks

TaskDatasetResultRank
ClassificationGOOD-CBAS
WSC Prediction Length2.91
7
ClassificationPubMed (PMD)
WSC Prediction Length1.84
7
Marginal Prediction LengthPubMed (PMD)
Marginal Prediction Length1.83
7
Worst-slab coverage (WSC)CBAS
WSC Coverage64
7
ClassificationGOOD-WebKB
WSC Prediction Length3.55
7
Node ClassificationCBAS (test)
Marginal Coverage91.7
7
Node ClassificationWKB (test)
Marginal Coverage90.3
7
Node ClassificationPMD (test)
Marginal Coverage90.4
7
Worst-slab coverage (WSC)WKB
WSC Coverage68
7
Worst-slab coverage (WSC)PMD
WSC Coverage75
7
Showing 10 of 54 rows

Other info

Follow for update