English

Enhanced Transformer architecture for in-context learning of dynamical systems

Machine Learning 2024-10-07 v1 Artificial Intelligence Systems and Control Systems and Control

Abstract

Recently introduced by some of the authors, the in-context identification paradigm aims at estimating, offline and based on synthetic data, a meta-model that describes the behavior of a whole class of systems. Once trained, this meta-model is fed with an observed input/output sequence (context) generated by a real system to predict its behavior in a zero-shot learning fashion. In this paper, we enhance the original meta-modeling framework through three key innovations: by formulating the learning task within a probabilistic framework; by managing non-contiguous context and query windows; and by adopting recurrent patching to effectively handle long context sequences. The efficacy of these modifications is demonstrated through a numerical example focusing on the Wiener-Hammerstein system class, highlighting the model's enhanced performance and scalability.

Keywords

Cite

@article{arxiv.2410.03291,
  title  = {Enhanced Transformer architecture for in-context learning of dynamical systems},
  author = {Matteo Rufolo and Dario Piga and Gabriele Maroni and Marco Forgione},
  journal= {arXiv preprint arXiv:2410.03291},
  year   = {2024}
}
R2 v1 2026-06-28T19:08:20.950Z