Conformal Load Prediction with Transductive Graph Autoencoders

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Luo, Rui, Colombo, Nicolo
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866909222445252608
author Luo, Rui
Colombo, Nicolo
author_facet Luo, Rui
Colombo, Nicolo
contents Predicting edge weights on graphs has various applications, from transportation systems to social networks. This paper describes a Graph Neural Network (GNN) approach for edge weight prediction with guaranteed coverage. We leverage conformal prediction to calibrate the GNN outputs and produce valid prediction intervals. We handle data heteroscedasticity through error reweighting and Conformalized Quantile Regression (CQR). We compare the performance of our method against baseline techniques on real-world transportation datasets. Our approach has better coverage and efficiency than all baselines and showcases robustness and adaptability.
format Preprint
id arxiv_https___arxiv_org_abs_2406_08281
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Conformal Load Prediction with Transductive Graph Autoencoders
Luo, Rui
Colombo, Nicolo
Machine Learning
Predicting edge weights on graphs has various applications, from transportation systems to social networks. This paper describes a Graph Neural Network (GNN) approach for edge weight prediction with guaranteed coverage. We leverage conformal prediction to calibrate the GNN outputs and produce valid prediction intervals. We handle data heteroscedasticity through error reweighting and Conformalized Quantile Regression (CQR). We compare the performance of our method against baseline techniques on real-world transportation datasets. Our approach has better coverage and efficiency than all baselines and showcases robustness and adaptability.
title Conformal Load Prediction with Transductive Graph Autoencoders
topic Machine Learning
url https://arxiv.org/abs/2406.08281