1 citations · 2 across the 4 of their papers we have counts for
Showing cs.LGShow all
3 papers · 1 filter
cs.LG2026
In-Context Operator Learning on the Space of Probability Measures
Frank Cole, Dixi Wang, Yineng Chen +2
We introduce \emph{in-context operator learning on probability measure spaces} for optimal transport (OT). The goal is to learn a single solution operator that maps a pair of distr…
cs.LG2025★ 1 cited
In-Context Learning of Linear Dynamical Systems with Transformers: Approximation Bounds and Depth-Separation
Frank Cole, Yuxuan Zhao, Yulong Lu +1
This paper investigates approximation-theoretic aspects of the in-context learning capability of the transformers in representing a family of noisy linear dynamical systems. Our fi…
cs.LG2024
In-Context Learning of Linear Systems: Generalization Theory and Applications to Operator Learning
Frank Cole, Yulong Lu, Wuzhe Xu +1
We study theoretical guarantees for solving linear systems in-context using a linear transformer architecture. For in-domain generalization, we provide neural scaling laws that bou…