Publications & Proceedings

Local, mean, and barycenter policies
David Millard, Ali Baheri
Can Optimal Transport Improve Federated Inverse Reinforcement Learning?
Proceedings of the 8th Annual Learning for Dynamics and Control Conference, PMLR 331:1939–1953, 2026.
Proceedings arXiv
Split conformal prediction in function space
David Millard, Lars Lindemann, Ali Baheri
Split Conformal Prediction in the Function Space via Neural Operator Learning
ICLR 2026 Workshop on AI and Partial Differential Equations.
OpenReview arXiv
PEARL: Preconditioner Enhancement
David Millard, Arielle Carr, Stéphane Gaudreault, Ali Baheri
PEARL: Preconditioner Enhancement through Actor-critic Reinforcement Learning
2026 Joint Mathematics Meetings, SIAM Minisymposium on Recent Advances in Numerical Linear Algebra.
Conference arXiv
Deep Ritz
Anton Selitskiy, David Millard
Deep Ritz Method for Elliptic Differential-Difference Equations
Proceedings of the AMS Spring Eastern Sectional Meeting, 2025.
Conference
Koopman Operator
David Millard, Arielle Carr, Stéphane Gaudreault
Deep Learning for Koopman Operator Estimation in Idealized Atmospheric Dynamics
Proceedings of the 2024 IEEE International Conference on Big Data.
arXiv IEEE Xplore
Initial Guess Selection
David Millard, Arielle Carr, Stéphane Gaudreault
Data-Driven Initial Guess Selection for Numerical Weather Prediction Solvers
Proceedings of the 2024 IEEE International Conference on Big Data.
IEEE Xplore

Articles

Validation potential error across learning-rate ratios
Anton Selitskiy, David Millard
Stability of the Monge Map in Semi-Dual Optimal Transport
Preprint.
arXiv
Comparison of CNN and neural semi-Lagrangian weather advection
Carlos A. Pereira, Stéphane Gaudreault, Valentin Dallerit, Christopher Subich, Shoyon Panday, Siqi Wei, Sasa Zhang, Siddharth Rout, Eldad Haber, Raymond J. Spiteri, David Millard, Emilia Diaconescu
Learning to Advect: A Neural Semi-Lagrangian Architecture for Weather Forecasting
Preprint.
arXiv
FedAvg and risk-weighted barycenter distributions
David Millard, Cecilia Alm, Rashid Ali, Pengcheng Shi, Ali Baheri
Federated Distributional Reinforcement Learning with Distributional Critic Regularization
Preprint.
arXiv
DEF: Diffusion-augmented
David Millard, Arielle Carr, Stéphane Gaudreault, Ali Baheri
DEF: Diffusion-augmented Ensemble Forecasting
Preprint.
arXiv