An Attention-based Spatio-Temporal Neural Operator for Evolving Physics

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Karkaria, Vispi, Lee, Doksoo, Chen, Yi-Ping, Yu, Yue, Chen, Wei
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915340759334912
author Karkaria, Vispi
Lee, Doksoo
Chen, Yi-Ping
Yu, Yue
Chen, Wei
author_facet Karkaria, Vispi
Lee, Doksoo
Chen, Yi-Ping
Yu, Yue
Chen, Wei
contents In scientific machine learning (SciML), a key challenge is learning unknown, evolving physical processes and making predictions across spatio-temporal scales. For example, in real-world manufacturing problems like additive manufacturing, users adjust known machine settings while unknown environmental parameters simultaneously fluctuate. To make reliable predictions, it is desired for a model to not only capture long-range spatio-temporal interactions from data but also adapt to new and unknown environments; traditional machine learning models excel at the first task but often lack physical interpretability and struggle to generalize under varying environmental conditions. To tackle these challenges, we propose the Attention-based Spatio-Temporal Neural Operator (ASNO), a novel architecture that combines separable attention mechanisms for spatial and temporal interactions and adapts to unseen physical parameters. Inspired by the backward differentiation formula (BDF), ASNO learns a transformer for temporal prediction and extrapolation and an attention-based neural operator for handling varying external loads, enhancing interpretability by isolating historical state contributions and external forces, enabling the discovery of underlying physical laws and generalizability to unseen physical environments. Empirical results on SciML benchmarks demonstrate that ASNO outperforms over existing models, establishing its potential for engineering applications, physics discovery, and interpretable machine learning.
format Preprint
id arxiv_https___arxiv_org_abs_2506_11328
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle An Attention-based Spatio-Temporal Neural Operator for Evolving Physics
Karkaria, Vispi
Lee, Doksoo
Chen, Yi-Ping
Yu, Yue
Chen, Wei
Machine Learning
Computational Engineering, Finance, and Science
In scientific machine learning (SciML), a key challenge is learning unknown, evolving physical processes and making predictions across spatio-temporal scales. For example, in real-world manufacturing problems like additive manufacturing, users adjust known machine settings while unknown environmental parameters simultaneously fluctuate. To make reliable predictions, it is desired for a model to not only capture long-range spatio-temporal interactions from data but also adapt to new and unknown environments; traditional machine learning models excel at the first task but often lack physical interpretability and struggle to generalize under varying environmental conditions. To tackle these challenges, we propose the Attention-based Spatio-Temporal Neural Operator (ASNO), a novel architecture that combines separable attention mechanisms for spatial and temporal interactions and adapts to unseen physical parameters. Inspired by the backward differentiation formula (BDF), ASNO learns a transformer for temporal prediction and extrapolation and an attention-based neural operator for handling varying external loads, enhancing interpretability by isolating historical state contributions and external forces, enabling the discovery of underlying physical laws and generalizability to unseen physical environments. Empirical results on SciML benchmarks demonstrate that ASNO outperforms over existing models, establishing its potential for engineering applications, physics discovery, and interpretable machine learning.
title An Attention-based Spatio-Temporal Neural Operator for Evolving Physics
topic Machine Learning
Computational Engineering, Finance, and Science
url https://arxiv.org/abs/2506.11328