REVIEW 3 major objections 4 minor 10 references
A physics-informed neural network solves the Monge-Ampère beam-shaping equation for flat optics, producing phase masks that outperform conventional phase retrieval on a structured far-field target.
Reviewed by Pith at T0; open to challenge. T0 means a machine referee read the full paper against a public rubric. the ladder, T0–T4 →
2026-08-01 16:21 UTC pith:QXMP5WRO
load-bearing objection Legitimate new PINN application with real diffraction validation, but the Monge–Ampère solver omits the boundary condition that makes the transport map onto, and the GS comparison is not convergence-checked. the 3 major comments →
Physics-Informed Neural Networks for Optimal Beam Shaping in Flat Optics
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
Core claim
The central claim is that a PINN can directly solve the Monge-Ampère equation that governs energy-conserving ray mappings for a phase-only optical element, for both finite-distance and far-field targets. The network represents the phase potential and is trained by minimizing the residual of the PDE, using automatic differentiation to compute the required second derivatives. For a Gaussian beam to a 'DT' logo target in the far field, the learned phase produces an energy efficiency of about 99.98% inside the target support, versus 95.90% for a conventional projection-based algorithm, and suppresses the high-frequency speckle caused by irregular phase retrieval. The ray-based design is confirme
What carries the argument
The key machinery is the Monge-Ampère equation derived from the generalized Snell law and power conservation. For far-field targets, the equation reduces to a standard Hessian-determinant form in which the phase gradient directly gives the direction cosines of the target; for finite distances, additional terms involving the ray propagation length and the longitudinal component K appear. The field is represented by a fully connected neural network with smooth tanh activations, and automatic differentiation provides all derivatives needed for the residual. The target intensity is sampled at the phase-dependent mapped coordinates using differentiable bilinear resampling, so the gradients flow t
Load-bearing premise
The method assumes that satisfying the interior Monge-Ampère residual alone is enough to produce a globally energy-conserving ray map, without an explicit boundary condition linking source aperture to target boundary; if this fails, parts of the target may be missing or under-illuminated.
What would settle it
A direct test would be to design a target distribution concentrated near the edge of the aperture or with a sharp cutoff, and measure via wave propagation whether the PINN's phase produces the correct boundary coverage and total energy. A failure would show missing or extra energy at the edges, or an efficiency below the reported 99.98%.
If this is right
- If the method holds, designing phase masks for complicated two-dimensional beam shapes becomes a PDE-solving task rather than an iterative Fourier projection, potentially yielding smoother and more accurate optic surfaces.
- The approach can be extended to non-cylindrically symmetric targets, opening up uses in structured illumination, holography, and display backlighting.
- The ray-based PINN solution remains valid under scalar wave propagation for the tested cases, suggesting that the method bridges geometric and wave descriptions of beam shaping.
- Because the training signal is the physics residual itself, the method requires no labeled phase data and can be adapted to new source-target pairs by retraining the network for that specific design.
- The reported efficiency and error metrics imply that phase-only elements can channel nearly all incident power into a desired pattern, which is valuable for high-power laser applications.
Where Pith is reading between the lines
- The absence of an explicit boundary condition enforcing that the source aperture maps onto the entire target domain is a likely weakness. Without such a condition, the interior residual may leave portions of the target unilluminated or violate global energy conservation, even if post-hoc admissibility checks catch gross errors.
- A natural testable extension is to add a boundary loss or use a conservative formulation; if successful, this would make the method robust for targets whose support is strongly clipped by the aperture.
- The same residual approach could be combined with a parameterized target family to build a generative model for beam-shaping optics, though such a model would require many training runs and may struggle with very sharp or discontinuous target features.
- The method's reliance on smooth tanh activations and a scalar phase potential may limit its ability to represent wavefronts with vortices or other phase singularities, which are outside the Monge-Ampère transport-map description.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The manuscript proposes a physics-informed neural network (PINN) approach for designing phase-only flat-optics elements that reshape an input beam into prescribed intensity distributions. The phase profile is represented by a fully connected network and trained by minimizing residuals of Monge–Ampère equations derived from generalized-Snell-law ray mappings, for both finite-distance targets (Eq. 29) and far-field angular targets (Eq. 37). The learned phases are validated with independent scalar diffraction simulations for a Gaussian-to-star finite-distance example and a Gaussian-to-logo far-field example, and the far-field result is compared against 60 iterations of the Gerchberg–Saxton algorithm. The central claims are that this is the first PINN application to flat-optics beam shaping and that it achieves lower error and higher energy efficiency than GS for the tested target.
Significance. If fully supported, the method would be a useful addition to flat-optics design: it replaces iterative phase retrieval with a PDE-constrained neural optimization, handles non-cylindrically symmetric targets, and the authors provide open code and independent diffraction validation. The ray-mapping derivation appears self-consistent, and the normalization of the equations is nontrivial and largely convincing. However, the central claim is currently weakened by a structural gap in the PDE formulation: no boundary condition is imposed on the Monge–Ampère equation. Because the method is demonstrated on only one target per geometry and the GS baseline is not shown to be converged, the generality and superiority claims are not yet established at the level required for publication.
major comments (3)
- [§3.2, Eqs. (29), (37)] The Monge–Ampère equation is solved without any boundary condition. The manuscript explicitly states that no Dirichlet condition is imposed and that admissibility is only checked after training by verifying that mapped points remain in the target domain and that K is real. This is not sufficient: in optimal transport, the correct boundary condition—typically that the source aperture boundary maps onto the target boundary—is required for the solution to be a genuine transport map that covers the full target and conserves energy globally. A network can minimize the interior residual in Eq. (41) while producing a ray map that is not onto, leaving target regions unilluminated or delivering source power multiple times to the same region. The post-hoc checks are local admissibility checks, not global flux-balance checks. The reported 99.98% energy efficiency comes from wave-optics simulations
- [§4.3] The comparison with Gerchberg–Saxton is weakened by the use of only 60 iterations with no reported convergence criterion. GS is known to converge slowly for structured targets such as a two-letter logo, so an underconverged baseline is not a fair reference for the claimed superiority (relative RMS 2.97% vs 22.6%, energy efficiency 99.98% vs 95.90%). Please report the error as a function of iteration count and either run GS until a convergence metric stagnates or state the stopping rule. Without this, the quantitative superiority claim is not decisive.
- [§4.1–§4.2, §5] The paper's general claim—that the PINN approach solves beam-shaping problems in flat optics—rests on only two numerical examples, one per formulation. While the finite-distance star example is smooth and the far-field logo example has internal strokes, a single realization per geometry is not enough to establish that the method reliably produces admissible, boundary-satisfying transport maps, especially given the missing boundary condition. I recommend adding at least one additional target per formulation, or tempering the conclusions to a proof-of-concept demonstration. This is load-bearing for the stated scope of the method.
minor comments (4)
- [§3.2] The notation D_near in Eq. (39) is introduced without a formal definition. It is presumably the left-hand side of Eq. (29) before multiplying by K^4; please define it explicitly to avoid ambiguity.
- [§4.3] The MSE values (3.59e6 and 2.08e8) are reported without normalization or units. Please report per-pixel or normalized MSE and specify the integration domain, so the comparison is reproducible.
- [Fig. 5] The panels in Fig. 5 are not individually labeled or captioned. Adding labels (a)–(d) or similar would greatly improve readability, especially since the figure is central to the validation claim.
- [References] The derivation of the Monge–Ampère beam-shaping equation cites the author's own project report [2] and a PhD thesis [3]. For a journal submission, please also cite peer-reviewed sources for the ray-mapping/Monge–Ampère formulation, e.g., standard references on nonimaging optics or phase retrieval.
Circularity Check
No significant circularity: the PINN phase design is trained on a physics-based Monge–Ampère residual and validated by independent scalar diffraction, so the central claim is not equivalent to its inputs.
full rationale
The paper’s derivation chain is self-contained. It starts from the thin-element phase ansatz in Eq. (1), the generalized Snell law in Eqs. (3)–(4) (an external result from Yu et al., [8]), and the geometric ray map in Eqs. (8)–(9). Power conservation in Eqs. (10)–(11) yields the Jacobian condition, which is expanded into the finite-distance Monge–Ampère equation Eq. (13) and the far-field Monge–Ampère equation Eq. (24). The target intensity E or I_e,Ωcosθ is a prescribed input, not an output of the network; the free function is the phase Φ, initialized with no phase labels or precomputed ray correspondences. The loss Eq. (41) penalizes the residual of the PDE only, with automatic differentiation providing derivatives. The resulting phase is then checked by scalar wave propagation using diffractsim, an independent forward model. No step reduces the claimed prediction—the phase mask—to the target input by construction. Several references are to the author’s own prior work ([2], [6], [7], [9]), but these are contextual or supporting citations: the Monge–Ampère equation is re-derived explicitly rather than imported as an unverified black box, and the prior metasurface demonstrations are cited only as motivation for extending to non-cylindrical targets. The absence of an explicit transport boundary condition is a genuine mathematical well-posedness concern and could affect the generality of the method, but it does not make the derivation circular: it is a structural gap in the PDE formulation, not an identity between inputs and outputs. The paper therefore exhibits no significant circularity.
Axiom & Free-Parameter Ledger
free parameters (1)
- sigma_Phi =
not specified (chosen via Bayesian optimization)
axioms (4)
- domain assumption Thin-phase element: the output field is input amplitude times phase factor (amplitude unchanged immediately after surface).
- domain assumption Energy-conserving, single-valued ray map: no caustics or folds.
- domain assumption Scalar diffraction theory (angular-spectrum) is valid for validation.
- standard math The PINN can approximate the PDE solution and training converges.
Cite this review
Pith. "Pith review of Physics-Informed Neural Networks for Optimal Beam Shaping in Flat Optics." pith.science (2026). https://pith.science/paper/QXMP5WRO
@misc{pith2026260718012,
author = {Pith},
title = {Pith review of: Physics-Informed Neural Networks for Optimal Beam Shaping in Flat Optics},
year = {2026},
howpublished = {\url{https://pith.science/paper/QXMP5WRO}},
note = {Machine review of arXiv:2607.18012}
}
read the original abstract
We introduce a physics-informed neural network (PINN) approach for designing phase profiles in flat optics that reshape an incident beam into a prescribed target intensity distribution. The method solves Monge--Amp\`ere beam-shaping equations associated with energy-conserving ray mappings generated by a phase-only optical element. We treat both finite-distance and far-field targets using a generalized-Snell-law formulation. The learned phase profiles are validated by scalar diffraction simulations and compared with conventional phase-retrieval methods such as Gerchberg--Saxton. To our knowledge, this is the first time a PINN has been used for beam shaping problems in flat optics.
Figures
Reference graph
Works this paper leans on
-
[1]
A practical algorithm for the determination of phase from image and diffraction plane pic- tures,
R. W. Gerchberg and W. O. Saxton, “A practical algorithm for the determination of phase from image and diffraction plane pic- tures,”Optik35, 237–246 (1972)
1972
-
[2]
Design and simulation of structured beams using phase profiles,
Rafael de la Fuente Herrezuelo, “Design and simulation of structured beams using phase profiles,” project report, Univer- sidad Nacional de Educaci´on a Distancia (2025)
2025
-
[3]
Nielsen, K. E. S. (2024). Optical metasurfaces for lighting ap- plications. Department of Physics, Technical University of Den- mark
2024
-
[4]
M. Raissi, P. Perdikaris, and G. E. Karniadakis, “Physics- informed neural networks: A deep learning framework for solv- ing forward and inverse problems involving nonlinear partial differential equations,”Journal of Computational Physics378, 686–707 (2019). doi:10.1016/j.jcp.2018.10.045
-
[5]
Physics-informed neural networks for inverse problems in nano-optics and metamaterials,
Y . Chen, L. Lu, G. E. Karniadakis, and L. Dal Negro, “Physics-informed neural networks for inverse problems in nano-optics and metamaterials,”Optics Express28(8), 11618– 11633 (2020). doi:10.1364/OE.384875
-
[6]
Cylindrically-symmetric collimated beam shaping using non- imaging metasurfaces,
Kirstine E. S. Nielsen, Xavier Zambrana-Puyalto, Rafael de la Fuente Herrezuelo, Villads Egede Johansen, and Søren Raza, “Cylindrically-symmetric collimated beam shaping using non- imaging metasurfaces,”Proc. SPIE13132, 1313208 (2024). doi:10.1117/12.3027738
-
[7]
Experimental demonstration of a beam shaping non- imaging metasurface,
Kirstine E. S. Poulsen, Xavier Zambrana-Puyalto, Rafael de la Fuente Herrezuelo, Villads Egede Johansen, and Søren Raza, “Experimental demonstration of a beam shaping non- imaging metasurface,”Optics Express33(9), 19119–19129 (2025). doi:10.1364/OE.559542
-
[8]
Light propagation with phase discon- tinuities: generalized laws of reflection and refraction,
N. Yu, P. Genevet, M. A. Kats, F. Aieta, J.-P. Tetienne, F. Ca- passo, and Z. Gaburro, “Light propagation with phase discon- tinuities: generalized laws of reflection and refraction,”Science 334(6054), 333–337 (2011). doi:10.1126/science.1210713
-
[9]
diffractsim: A flex- ible Python diffraction simulator,
R. de la Fuente Herrezuelo, “diffractsim: A flex- ible Python diffraction simulator,” GitHub (2022). doi:10.5281/zenodo.6147771
-
[10]
M. Jaderberget al., “Spatial Transformer Networks,”Ad- vances in Neural Information Processing Systems28, 2017– 2025 (2015). doi:10.48550/arXiv.1506.02025 5
This paper was first reviewed by deepseek-v4-flash on August 1, 2026.
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.