Challenges
Datasets
Workspaces
Papers
Discussions
Leaderboard
Log inSign up
Challenges
Datasets
Workspaces
Papers
Discussions
Leaderboard
Blog
Job Board
Q3AS

© 2026 Aqora Quantum S.A.S.

TermsPrivacyLegal Notice

IEEE Transactions on Artificial Intelligence

Robust Control of Uncertain Quantum Systems Based on Physics-Informed Neural Networks and Sampling Learning

Kai Zhang, Qi Yu, Sen Kuang

Published Jan 20, 2025• 7 citations
Publisher

Abstract

High-fidelity quantum control is one of the key elements in quantum computing and information processing. In view of possible inaccuracies in quantum system modeling and inevitable errors in control fields, the design of robust control fields is of great importance. In this article, we propose a neural network-based robust control strategy that incorporates physics-informed neural networks (PINNs) and sampling-based learning control techniques for uncertain closed and open quantum systems. We employ the gradient descent algorithm with momentum for the network training, where two methods including direct calculation and automatic differentiation are used to compute the gradient of the loss function with respect to network weights. The direct calculation method demonstrates the internal mechanism of the gradient computation, while the automatic differentiation technology is easier to utilize. We provide some guidelines for the parameter selection of the sampling learning algorithm in the PINN robust control scheme to ensure good control performance. In particular, for open quantum systems with uncertainties, we point out the necessity of fast control. Some simulation experiments are conducted on closed and open systems with uncertainties and the results show the effectiveness of the proposed PINN control scheme in achieving high-fidelity state transfer of uncertain quantum systems.

DOI
10.1109/tai.2025.3531330

Related papers

A Neural Network Approach to Sampling Based Learning Control for Quantum System with Uncertainty

Jingshi Li, Song Chen Song Chen, Yanzhao Cao +1 more

Robust control design for quantum systems with uncertainty is a key task for developing practical quantum technology. In this paper, we apply neural networks to learn the control of a quantum system with uncertainty. By exploiting the auto differentiation function developed for neural network models, our method avoids the manual computation of the gradient of the cost function as required in traditional methods. We implement our method using two algorithms. One uses neural networks to learn both the states and the controls and one uses neural networks to learn only the controls but solve the states by finite difference methods. Both algorithms incorporate the sampling-based learning process into the training of the networks. The performance of the algorithms is evaluated on a practical numerical example, followed by a detailed discussion about the advantage and trade-offs between our method and the other numerical schemes.

Communications in Computational Physics•Oct 25, 2021•2 citations

Physics-Informed Neural Networks for Gate Design using Quantum Optimal Control

Sofiia Lauten, Matthew Otten

Implementing quantum gates on quantum computers can require the application of carefully shaped pulses for high-fidelity operations. We explore the use of physics-informed neural networks (PINNs) for quantum optimal control to assess their usefulness in predicting such pulses. Our PINN is a feedforward neural network that utilizes an unsupervised learning approach, whose loss function includes terms that enforce the equations that govern the evolution of a quantum system, measure how close the learned unitary is to the target unitary operation, and ensure state normalization. We use a sinusoidal activation function and adopt variance-type weight initialization, tailored to our activation function. By analyzing the model's performance with important machine learning metrics, we demonstrate that the choice of our architecture is well-suited for this type of problem. We ensure that our network avoids the vanishing and exploding gradients with our relevant choices. We build two different PINNs, one based on the Schrödinger equation and another one based on the Lindblad equation. Our PINNs are able to discover high-fidelity two-qubit gate pulses for a variety of quantum operations, demonstrating its flexibility and robustness. We build two different PINNs, one based on the Schrödinger equation and another one based on the Lindblad equation. Our PINNs are able to discover high-fidelity two-qubit gate pulses for a variety of quantum operations, demonstrating its flexibility and robustness.

quant-pharXiv:2511.09463•Nov 13, 2025•0 citations

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

S Vijayaram, Pagavathigounder Balasubramaniam

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.

The Philosophical Magazine A Journal of Theoretical Experimental and Applied Physics•Jun 2, 2026•0 citations

Forked Physics-Informed Neural Networks for Non-Markovian Open Quantum Dynamics and Control

Zhao-Wei Wang, Kai-Yu Yuan, Feng-Hua Ren +1 more

Physics-informed neural networks (PINNs) provide a pathway to reunify the simulation and control of quantum systems, in which these two tasks are typically decoupled in traditional strategies. However, most work remains confined to Markovian environments. When applied to non-Markovian systems, standard PINN architectures fail to converge reliably due to multi-objective optimization conflicts arising from the coupled differential equations. To address this fundamental limitation, we extend our previously proposed forked PINN (FPINN) by incorporating a dedicated control branch. By decoupling the optimization objectives at the gradient level via selective gradient flow, our method turns a previously intractable multi-task optimization into a well-conditioned one, allowing simulation and control to be optimized jointly without compromise. Numerical simulations on a two-qubit Heisenberg XXX model confirm that our framework faithfully reproduces the features of non-Markovian dynamics, including decoherence and information backflow. Taking a state-preparation task on the same model as an example, our FPINN achieves higher fidelity than gradient ascent pulse engineering, chopped random basis, and standard PINNs, with the advantage becoming more pronounced as the environment becomes more dissipative and more Markovian. The generated pulses are also noticeably smoother, which is advantageous for experimental implementation. Our framework thus provides a unified, end-to-end differentiable paradigm for simulation and control of open quantum systems, with potential implications for quantum computing, simulation, and control.

quant-pharXiv:2607.12567•Jul 15, 2026•0 citations

Adaptive Quantum Physics-Informed Neural Networks for Differential Equations with Applications to Fluid Dynamics

Fabio Pereira dos Santos, Renato Portugal, Julio de Castro Vargas Fernandes +1 more

Physics-informed neural networks (PINNs) have emerged as a versatile approach for solving nonlinear partial differential equations (PDEs), yet achieving high accuracy efficiently using these techniques remains challenging for high-dimensional or multiscale systems. Here, we present a hybrid quantum-classical framework that enhances Quantum PINNs (QPINNs) through adaptive collocation point sampling and loss-aware attention mechanisms. By dynamically prioritizing points in regions with large PDE residuals or steep solution gradients, our method mitigates the spectral bias inherent in conventional PINNs. Current Quantum Physics-Informed Neural Networks are commonly assumed to be limited by the expressive power of quantum circuits. In our work, we observed that, across diverse differential equations, optimization - not only expressivity - can be an important bottleneck. Furthermore, a trainable loss-weighting scheme balances contributions from physics residuals, boundary conditions, and data fidelity during training. Integrating these strategies with quantum computing techniques (including variational quantum circuits and quantum gradient estimation) can yield at least a 60% improvement in solution accuracy under specific regimes for benchmark fluid flows and reaction-diffusion systems. Finally, we argue that merely increasing model expressivity is insufficient for resolving complex PDEs via QPINNs, as they remain constrained by the structural optimization limitations of classical PINNs. This framework provides a scalable pathway for quantum-enhanced scientific machine learning, bridging physics-based modeling with emerging quantum computational capabilities.

arXiv:2608.00850•Aug 1, 2026•0 citations

Physics-Informed Neural Networks for Quantum Control

Ariel Norambuena, Marios Mattheakis, F.J. Gonzalez +1 more

Quantum control is a ubiquitous research field that has enabled physicists to delve into the dynamics and features of quantum systems, delivering powerful applications for various atomic, optical, mechanical, and solid-state systems. In recent years, traditional control techniques based on optimization processes have been translated into efficient artificial intelligence algorithms. Here, we introduce a computational method for optimal quantum control problems via physics-informed neural networks (PINNs). We apply our methodology to open quantum systems by efficiently solving the state-to-state transfer problem with high probabilities, short-time evolution, and using low-energy consumption controls. Furthermore, we illustrate the flexibility of PINNs to solve the same problem under changes in physical parameters and initial conditions, showing advantages in comparison with standard control techniques.

arXiv:2206.06287•Physical Review Letters•Jan 3, 2024•71 citations
High-fidelity quantum control is one of the key elements in quantum computing and information processing. In view of possible inaccuracies in quantum system modeling and inevitable errors in control fields, the design of robust control fields is of great importance. In this article, we propose a neural network-based robust control strategy that incorporates physics-informed neural networks (PINNs) and sampling-based learning control techniques for uncertain closed and open quantum systems. We employ the gradient descent algorithm with momentum for the network training, where two methods including direct calculation and automatic differentiation are used to compute the gradient of the loss function with respect to network weights. The direct calculation method demonstrates the internal mechanism of the gradient computation, while the automatic differentiation technology is easier to utilize. We provide some guidelines for the parameter selection of the sampling learning algorithm in the PINN robust control scheme to ensure good control performance. In particular, for open quantum systems with uncertainties, we point out the necessity of fast control. Some simulation experiments are conducted on closed and open systems with uncertainties and the results show the effectiveness of the proposed PINN control scheme in achieving high-fidelity state transfer of uncertain quantum systems.
Robust control design for quantum systems with uncertainty is a key task for developing practical quantum technology. In this paper, we apply neural networks to learn the control of a quantum system with uncertainty. By exploiting the auto differentiation function developed for neural network models, our method avoids the manual computation of the gradient of the cost function as required in traditional methods. We implement our method using two algorithms. One uses neural networks to learn both the states and the controls and one uses neural networks to learn only the controls but solve the states by finite difference methods. Both algorithms incorporate the sampling-based learning process into the training of the networks. The performance of the algorithms is evaluated on a practical numerical example, followed by a detailed discussion about the advantage and trade-offs between our method and the other numerical schemes.
Implementing quantum gates on quantum computers can require the application of carefully shaped pulses for high-fidelity operations. We explore the use of physics-informed neural networks (PINNs) for quantum optimal control to assess their usefulness in predicting such pulses. Our PINN is a feedforward neural network that utilizes an unsupervised learning approach, whose loss function includes terms that enforce the equations that govern the evolution of a quantum system, measure how close the learned unitary is to the target unitary operation, and ensure state normalization. We use a sinusoidal activation function and adopt variance-type weight initialization, tailored to our activation function. By analyzing the model's performance with important machine learning metrics, we demonstrate that the choice of our architecture is well-suited for this type of problem. We ensure that our network avoids the vanishing and exploding gradients with our relevant choices. We build two different PINNs, one based on the Schrödinger equation and another one based on the Lindblad equation. Our PINNs are able to discover high-fidelity two-qubit gate pulses for a variety of quantum operations, demonstrating its flexibility and robustness. We build two different PINNs, one based on the Schrödinger equation and another one based on the Lindblad equation. Our PINNs are able to discover high-fidelity two-qubit gate pulses for a variety of quantum operations, demonstrating its flexibility and robustness.
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.
Physics-informed neural networks (PINNs) provide a pathway to reunify the simulation and control of quantum systems, in which these two tasks are typically decoupled in traditional strategies. However, most work remains confined to Markovian environments. When applied to non-Markovian systems, standard PINN architectures fail to converge reliably due to multi-objective optimization conflicts arising from the coupled differential equations. To address this fundamental limitation, we extend our previously proposed forked PINN (FPINN) by incorporating a dedicated control branch. By decoupling the optimization objectives at the gradient level via selective gradient flow, our method turns a previously intractable multi-task optimization into a well-conditioned one, allowing simulation and control to be optimized jointly without compromise. Numerical simulations on a two-qubit Heisenberg XXX model confirm that our framework faithfully reproduces the features of non-Markovian dynamics, including decoherence and information backflow. Taking a state-preparation task on the same model as an example, our FPINN achieves higher fidelity than gradient ascent pulse engineering, chopped random basis, and standard PINNs, with the advantage becoming more pronounced as the environment becomes more dissipative and more Markovian. The generated pulses are also noticeably smoother, which is advantageous for experimental implementation. Our framework thus provides a unified, end-to-end differentiable paradigm for simulation and control of open quantum systems, with potential implications for quantum computing, simulation, and control.
Physics-informed neural networks (PINNs) have emerged as a versatile approach for solving nonlinear partial differential equations (PDEs), yet achieving high accuracy efficiently using these techniques remains challenging for high-dimensional or multiscale systems. Here, we present a hybrid quantum-classical framework that enhances Quantum PINNs (QPINNs) through adaptive collocation point sampling and loss-aware attention mechanisms. By dynamically prioritizing points in regions with large PDE residuals or steep solution gradients, our method mitigates the spectral bias inherent in conventional PINNs. Current Quantum Physics-Informed Neural Networks are commonly assumed to be limited by the expressive power of quantum circuits. In our work, we observed that, across diverse differential equations, optimization - not only expressivity - can be an important bottleneck. Furthermore, a trainable loss-weighting scheme balances contributions from physics residuals, boundary conditions, and data fidelity during training. Integrating these strategies with quantum computing techniques (including variational quantum circuits and quantum gradient estimation) can yield at least a 60% improvement in solution accuracy under specific regimes for benchmark fluid flows and reaction-diffusion systems. Finally, we argue that merely increasing model expressivity is insufficient for resolving complex PDEs via QPINNs, as they remain constrained by the structural optimization limitations of classical PINNs. This framework provides a scalable pathway for quantum-enhanced scientific machine learning, bridging physics-based modeling with emerging quantum computational capabilities.
Quantum control is a ubiquitous research field that has enabled physicists to delve into the dynamics and features of quantum systems, delivering powerful applications for various atomic, optical, mechanical, and solid-state systems. In recent years, traditional control techniques based on optimization processes have been translated into efficient artificial intelligence algorithms. Here, we introduce a computational method for optimal quantum control problems via physics-informed neural networks (PINNs). We apply our methodology to open quantum systems by efficiently solving the state-to-state transfer problem with high probabilities, short-time evolution, and using low-energy consumption controls. Furthermore, we illustrate the flexibility of PINNs to solve the same problem under changes in physical parameters and initial conditions, showing advantages in comparison with standard control techniques.