Traditional Physics-Informed Distributed Federated Learning (PIDFL) for solving the 2D Poisson equation faces challenges of slow convergence and high communication costs. We propose a Krylov-accelerated federated PINN framework that integrates matrix-free Kronecker Factored Approximate Curvature (K-FAC) with Conjugate Gradient (CG), exploiting the symmetric positive definite structure of discretized Poisson problems to enable scalable second-order optimization in distributed settings. Compared to first-order methods like Adam, our approach reduces communication rounds and achieves improved accuracy. Results across multiple grid sizes demonstrate faster convergence and superior physics residual minimization, supporting its applicability to PDE-constrained learning with communication constraints.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Krylov-Accelerated Federated PINNs for 2D PDEs with K-FAC+CG

  • Amit Kumar Upadhyay,
  • Dharavath Ramesh

摘要

Traditional Physics-Informed Distributed Federated Learning (PIDFL) for solving the 2D Poisson equation faces challenges of slow convergence and high communication costs. We propose a Krylov-accelerated federated PINN framework that integrates matrix-free Kronecker Factored Approximate Curvature (K-FAC) with Conjugate Gradient (CG), exploiting the symmetric positive definite structure of discretized Poisson problems to enable scalable second-order optimization in distributed settings. Compared to first-order methods like Adam, our approach reduces communication rounds and achieves improved accuracy. Results across multiple grid sizes demonstrate faster convergence and superior physics residual minimization, supporting its applicability to PDE-constrained learning with communication constraints.