用神经网络求解带传输边界条件的蒙日-安培方程,助力光学设计。
A neural network approach for solving the Monge-Ampère equation with transport boundary condition
- 用多层感知机最小化残差、边界与凸性约束来逼近解
- 在圆到圆、方到圆等映射中性能优于传统有限差分法
- 适合需要快速原型设计的光学系统开发者
本文提出一种基于神经网络的新方法,用于求解带有传输边界条件的蒙日-安培方程,重点面向光学设计应用。通过多层感知机网络,以L-BFGS优化损失函数(包含方程残差、边界条件和凸性约束),学习近似解。在对称与非对称的圆到圆、方到圆、圆到花形反射器映射问题中,该方法表现优异。与传统的最小二乘有限差分求解器相比,神经网络方法在测试案例中展现出竞争力甚至更优性能。全面的超参数研究揭示了采样密度、网络结构和优化算法的影响。尽管尚需进一步验证其在复杂问题中的鲁棒性和收敛一致性,但该方法的简洁性与灵活性使其成为专用偏微分方程求解器的有力替代方案。
原文摘要 · Abstract (English)
This paper introduces a novel neural network-based approach to solving the Monge-Ampère equation with the transport boundary condition, specifically targeted towards optical design applications. We leverage multilayer perceptron networks to learn approximate solutions by minimizing a loss function that encompasses the equation's residual, boundary conditions, and convexity constraints. Our main results demonstrate the efficacy of this method, optimized using L-BFGS, through a series of test cases encompassing symmetric and asymmetric circle-to-circle, square-to-circle, and circle-to-flower reflector mapping problems. Comparative analysis with a conventional least-squares finite-difference solver reveals the competitive, and often superior, performance of our neural network approach on the test cases examined here. A comprehensive hyperparameter study further illuminates the impact of factors such as sampling density, network architecture, and optimization algorithm. While promising, further investigation is needed to verify the method's robustness for more complicated problems and to ensure consistent convergence. Nonetheless, the simplicity and adaptability of this neural network-based approach position it as a compelling alternative to specialized partial differential equation solvers.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。