7 papers
A Greedy PDE Router for Blending Neural Operators and Classical Methods
Sahana Rayan, Yash Patel, Ambuj Tewari
When solving PDEs, classical numerical solvers are often computationally expensive, while machine learning methods can suffer from spectral bias, failing to capture high-frequency…
Operator Learning for Schrödinger Equation: Unitarity, Error Bounds, and Time Generalization
Yash Patel, Unique Subedi, Ambuj Tewari
We consider the problem of learning the evolution operator for the time-dependent Schrödinger equation, where the Hamiltonian may vary with time. Existing neural network-based sur…
Distribution-Free Robust Predict-Then-Optimize in Function Spaces
Yash Patel, Ambuj Tewari
The need to rapidly solve PDEs in engineering design workflows has spurred the rise of neural surrogate models. In particular, neural operator models provide a discretization-invar…
Continuum Transformers Perform In-Context Learning by Operator Gradient Descent
Abhiti Mishra, Yash Patel, Ambuj Tewari
Transformers robustly exhibit the ability to perform in-context learning, whereby their predictive accuracy on a task can increase not by parameter updates but merely with the plac…
Conformal Robust Control of Linear Systems
Yash Patel, Sahana Rayan, Ambuj Tewari
End-to-end engineering design pipelines, in which designs are evaluated using concurrently defined optimal controllers, are becoming increasingly common in practice. To discover de…
Conformal Prediction for Ensembles: Improving Efficiency via Score-Based Aggregation
Eduardo Ochoa Rivera, Yash Patel, Ambuj Tewari
Distribution-free uncertainty estimation for ensemble methods is increasingly desirable due to the widening deployment of multi-modal black-box predictive models. Conformal predict…