Publications

You can find a complete list of my publications on my Google Scholar page.

Select Publications

1. Gelphman E, Verma D, Yang N, Osher S, Wu Fung S. End-to-end Training of High-Dimensional Optimal Control with Implicit Hamiltonians via Jacobian-Free Backpropagation, 2025.
2. Wu Fung S, Heaton H, Li Q, McKenzie D, Osher S, Yin W. JFB: Jacobian-free Backpropagation for Implicit Networks, AAAI Conference on Artificial Intelligence, 2022.
3. Ruthotto L, Osher S, Li W, Nurbekyan L, Wu Fung S. A Machine Learning Framework for Solving High-Dimensional Mean Field Game and Mean Field Control Problems, Proceedings of the National Academy of Sciences, 2020.
4. Wu Fung S, Nurbekyan L. Mean-Field Control Barrier Functions: A Framework for Real-Time Swarm Control. American Control Conference, 2025.
5. Onken D, Wu Fung S, Li X, Ruthotto L. OT-Flow: Fast and Accurate Continuous Normalizing Flows via Optimal Transport, AAAI Conference on Artificial Intelligence, 2021.
6. Tibshirani R, Wu Fung S, Heaton H, Osher S. Laplace Meets Moreau: Smooth Approximation to Infimal Convolutions Using Laplace's Method. Journal of Machine Learning Research, 2025.
7. Heaton H, Wu Fung S. Explainable AI via Learning to Optimize, Scientific Reports, 2023.
8. Osher S, Heaton H, Wu Fung S. A Hamilton-Jacobi-based Proximal Operator, Proceedings of the National Academy of Sciences, 2023.
9. McKenzie D, Wu Fung S, Heaton H. Differentiating Through Integer Linear Programs with Quadratic Regularization and Davis-Yin Splitting, Transactions on Machine Learning Research, 2024.
10. Lin AT, Wu Fung S, Li W, Nurbekyan L, Osher S. Alternating the Population and Agent Control via Two Neural Networks to Solve High-Dimensional Stochastic Mean Field Games, Proceedings of the National Academy of Sciences, 2021.