Control of stochastic quantum dynamics by differentiable programming

Schäfer, Frank and Sekatski, Pavel and Koppenhöfer, Martin and Bruder, Christoph and Kloc, Michal. (2021) Control of stochastic quantum dynamics by differentiable programming. Machine Learning: Science and Technology, 2 (3). 035004.

PDF - Published Version
Available under License CC BY (Attribution).


Official URL: https://edoc.unibas.ch/87298/

Downloads: Statistics Overview


Control of the stochastic dynamics of a quantum system is indispensable in fields such as quantum information processing and metrology. However, there is no general ready-made approach to the design of efficient control strategies. Here, we propose a framework for the automated design of control schemes based on differentiable programming. We apply this approach to the state preparation and stabilization of a qubit subjected to homodyne detection. To this end, we formulate the control task as an optimization problem where the loss function quantifies the distance from the target state, and we employ neural networks (NNs) as controllers. The system's time evolution is governed by a stochastic differential equation (SDE). To implement efficient training, we backpropagate the gradient information from the loss function through the SDE solver using adjoint sensitivity methods. As a first example, we feed the quantum state to the controller and focus on different methods of obtaining gradients. As a second example, we directly feed the homodyne detection signal to the controller. The instantaneous value of the homodyne current contains only very limited information on the actual state of the system, masked by unavoidable photon-number fluctuations. Despite the resulting poor signal-to-noise ratio, we can train our controller to prepare and stabilize the qubit to a target state with a mean fidelity of around 85%. We also compare the solutions found by the NN to a hand-crafted control strategy.
Faculties and Departments:05 Faculty of Science > Departement Physik > Physik > Theoretische Physik (Bruder)
UniBasel Contributors:Bruder, Christoph and Schäfer, Frank and Kloc, Michal
Item Type:Article, refereed
Article Subtype:Research Article
Publisher:IOP Publishing
Note:Publication type according to Uni Basel Research Database: Journal article
Identification Number:
edoc DOI:
Last Modified:12 Apr 2022 10:04
Deposited On:12 Apr 2022 10:04

Repository Staff Only: item control page