Graph In-Context Operator Networks for Generalizable Spatiotemporal Prediction

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Wu, Chenghan, Yu, Zongmin, Sun, Boai, Yang, Liu
Format: Preprint
Veröffentlicht: 2026
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866910129846222848
author Wu, Chenghan
Yu, Zongmin
Sun, Boai
Yang, Liu
author_facet Wu, Chenghan
Yu, Zongmin
Sun, Boai
Yang, Liu
contents In-context operator learning enables neural networks to infer solution operators from contextual examples without weight updates. While prior work has demonstrated the effectiveness of this paradigm in leveraging vast datasets, a systematic comparison against single-operator learning using identical training data has been absent. We address this gap through controlled experiments comparing in-context operator learning against classical operator learning (single-operator models trained without contextual examples), under the same training steps and dataset. To enable this investigation on real-world spatiotemporal systems, we propose GICON (Graph In-Context Operator Network), combining graph message passing for geometric generalization with example-aware positional encoding for cardinality generalization. Experiments on air quality prediction across two Chinese regions show that in-context operator learning outperforms classical operator learning on complex tasks, generalizing across spatial domains and scaling robustly from few training examples to 100 at inference.
format Preprint
id arxiv_https___arxiv_org_abs_2603_12725
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Graph In-Context Operator Networks for Generalizable Spatiotemporal Prediction
Wu, Chenghan
Yu, Zongmin
Sun, Boai
Yang, Liu
Machine Learning
Artificial Intelligence
In-context operator learning enables neural networks to infer solution operators from contextual examples without weight updates. While prior work has demonstrated the effectiveness of this paradigm in leveraging vast datasets, a systematic comparison against single-operator learning using identical training data has been absent. We address this gap through controlled experiments comparing in-context operator learning against classical operator learning (single-operator models trained without contextual examples), under the same training steps and dataset. To enable this investigation on real-world spatiotemporal systems, we propose GICON (Graph In-Context Operator Network), combining graph message passing for geometric generalization with example-aware positional encoding for cardinality generalization. Experiments on air quality prediction across two Chinese regions show that in-context operator learning outperforms classical operator learning on complex tasks, generalizing across spatial domains and scaling robustly from few training examples to 100 at inference.
title Graph In-Context Operator Networks for Generalizable Spatiotemporal Prediction
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2603.12725