Jakin S. Ng

Jakin S. Ng

Scholar Title

MIT EECS | Lincoln Laboratory Undergraduate Research and Innovation Scholar

Research Title

Learned Initializers for Iterative PDE Solvers

Cohort

2023–2024

Department

Mathematics, Electrical Engineering and Computer Science

Research Areas
  • Artificial Intelligence and Machine Learning
Supervisor

Justin Solomon

Abstract

Partial differential equations (PDEs) are ubiquitous throughout science, describing fundamental physical phenomena. Designing efficient and accurate numerical solvers is crucial but highly non-trivial. We propose using Fourier neural operators (FNOs), which learn the solution mapping for a family of PDEs, as learned initializers for iterative PDE solvers. By initializing close to the actual solution, the number of iterations required for the solver to converge is reduced, while achieving at least equivalent convergence and accuracy guarantees. Advantages include reduced computational cost for an entire family of PDEs, as well as equivalent accuracy to the original numerical solver.

Quote

During the SuperUROP program, I am looking forward to experiencing the full research process, as well as learning to communicate and present my results more effectively. I’m most excited to work on integrating ML and numerical methods to solve PDEs quickly and accurately.

Back to Scholars