Image Details

Choose export citation format:

diffhydro: Inverse Multiphysics Modeling and Embedded Machine Learning in Astrophysical Flows

  • Authors: Benjamin Horowitz, Zarija Lukić, Kentaro Nagamine, Yuri Oku

Benjamin Horowitz et al 2026 The Astrophysical Journal 1012 .

  • Provider: AAS Journals

Caption: Figure 8.

Comparison of gradient computation strategies for a differentiable Poisson solver. (a) Unrolled backpropagation: the Poisson solve Aϕ = F is implemented as an explicit sequence of multigrid (MG) iterations, and gradients of the loss ﹩{ \mathcal L }﹩ are obtained by backpropagating through the solver steps. This corresponds to differentiating the algorithm and requires storing or recomputing intermediate iterates, leading to memory and compute costs that scale with the number of solver iterations. (b) Adjoint method: the solver is treated as an implicit differentiable operator, and gradients are computed by solving a single adjoint system ﹩{A}^{T}{g}_{F}=\partial { \mathcal L }/\partial \phi ﹩. For symmetric Poisson operators AT = A, allowing for reuse of the same MG solver. This yields gradients of the converged solution with constant memory overhead, independent of the forward iteration count.

Other Images in This Article

Show More

Copyright and Terms & Conditions

Additional terms of reuse