Wonjun Lee

Assistant Professor
Department of Mathematics
The Ohio State University
Office : Math Tower MW548
profile photo

Publications and Preprints

Understanding Contrastive Learning through Variational Analysis and Neural Network Optimization Perspectives

  • Jeff Calder, Wonjun Lee
  • Preprint, 2025.
[ PDF ]

Geometry-Preserving Encoder/Decoder In Latent Generative Models

  • Wonjun Lee, Riley C.W. O'Neill, Dongmian Zou, Jeff Calder, Gilad Lerman
  • Preprint, 2025.
[ PDF ]

Monotone generative modeling via a Gromov-Monge embedding

  • Wonjun Lee, Yifei Yang, Dongmian Zou, Gilad Lerman
  • SIAM Journal of Mathematics of Data Science, 2025.
[ PDF ] [ Code ]

Monotone discretizations of levelset convex geometric PDEs

  • Jeff Calder, Wonjun Lee
  • Numerische Mathematik, 2024.
[ PDF (ArXiv) ] [ PDF (Springer) ] [ Code ]

Improving Hyperbolic Representations via Gromov-Wasserstein Regularization

  • Yifei Yang, Wonjun Lee, Dongmian Zou, Gilad Lerman
  • ECCV, 2024.
[ PDF ]

Deep JKO: time-implicit particle methods for general nonlinear gradient flows

  • Wonjun Lee, Li Wang, Wuchen Li
  • Journal of Computational Physics, 2024.
[ PDF ]

Mean Field Control Problems For Vaccine Distribution

  • Wonjun Lee, Siting Liu, Wuchen Li, Stanley Osher
  • Research in the Mathematical Sciences, 2022.
[ PDF ]

Computational Mean-Field Information Dynamics Associated With Reaction Diffusion Equations

  • Wuchen Li, Wonjun Lee, Stanley Osher
  • Journal of Computational Physics, 2022.
[ PDF ] [ Code ]

Random Features for High-Dimensional Nonlocal Mean-Field Games

  • Sudanshu Agrawal, Wonjun Lee, Samy Wu Fung, Levon Nurbekyan
  • Journal of Computational Physics, Mar, 2022.
[ PDF ]

Weakly-Supervised Convolutional Neural Networks for Vessel Segmentation in Cerebral Angiography

  • Arvind Vepa, Andrew Choi, Noor Nakhaei, Wonjun Lee, Noah Stier, Andrew Vu, Greyson Jenkins, Xiaoyan Yang, Manjot Shergill, Moira Desphy, Kevin Delao, Mia Levy, Cristopher Garduno, Lacy Nelson, Wandi Liu, Fan Hung, Fabien Scalzo
  • Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision (WACV), 2022, pp. 585-594
[ Link ]

Tropical Optimal Transport and Wasserstein Distances in Phylogenetic Tree Space

[ PDF ] [ Code ]

The Back-And-Forth Method For Wasserstein Gradient Flows

[ PDF ] [ Code ]

Controlling Propagation of epidemics via mean-field games

  • Wonjun Lee, Siting Liu, Hamidou Tembine, Wuchen Li, Stanley Osher
  • SIAM Journal on Applied Math, Dec, 2020.
[ PDF ] [ ResearchGate ]

Joint Task Assignment and Trajectory Optimization for a Mobile Robot Swarm by Mean-Field Game

  • Yuhan Kang, Siting Liu, Wonjun Lee, Hongliang Zhang, Wuchen Li, Zhu Han
  • IEEE GLOBECOM, Dec, 2020.
[ PDF ]

Energy-efficient Velocity Control for Massive Rotary-Wing UAVs: A Mean Field Game Approach

  • H. Gao, Wonjun Lee, Wuchen Li, Zhu Han, Stanley Osher, Vincent Poor
  • IEEE GLOBECOM, Dec, 2020.
[ PDF ]

Generalized unnormalized optimal transport and its fast algorithms

[ PDF ]