Optimal control of stochastic Schrödinger equation via physics informed neural networks: applications in quantum optics

In this paper, authors present a physics-informed neural networks (PINN) approach for solving the optimal control problem (OCP) of the stochastic Schrödinger equation (SSE), with an emphasis on applications in quantum optics. The SSE models quantum systems under uncertainty, a scenario often encountered in open quantum systems where environmental interactions introduce randomness into the system dynamics. These stochastic dynamics are particularly relevant in quantum optics, where precise control over quantum states is crucial for applications such as quantum computing, secure communication, and measurement protocols. Traditional numerical methods for solving such problems face limitations due to high computational costs and the complexity associated with stochastic partial differential equations. The proposed approach utilises PINN, a type of deep learning model that integrates physical laws directly into the learning process by incorporating loss functions based on governing equations, initial and boundary conditions. The fidelity-based objective function is constructed to achieve the desired quantum state so as to develop an OCP solving algorithm. The proposed work is distinguished between deterministic drift-based training and post-training stochastic validation. Extensive numerical experiments are demonstrated near-unity deterministic fidelity. Further, a systematic architecture study validated that a moderately deep network yielded optimal performance with reduced computational cost. Fidelity dissipation analysis confirmed that the method maintained accuracy even under strong damping. These findings established the PINN as an efficient and physically consistent framework for quantum optimal control in open systems.