Manipulation of Spin Dynamics by Deep Reinforcement Learning Agent

Abstract

We implement the reinforcement learning agent for a spin-1 atomic system to prepare spin squeezed state from given initial state. Proximal policy gradient (PPO) algorithm is used to deal with continuous external control field and final optimized protocol is given by a stochastic policy. In both mean-field system and two-body quantum system, RL agent finds the optimal policies. In many-body quantum system, it also gives polices that outperform purely greedy policy and optimized adiabatic passage. These polices given by RL agent have good physical interpretability in phase space and may help us to understand quantum dynamics. In fact, RL could be highly versatile in quantum optimal control problems.

8pages, 8 figures

Cited by in corpus (5)

• Quantum optimal control of multi-level dissipative quantum systems with Reinforcement Learning
• Policy Gradient based Quantum Approximate Optimization Algorithm
• Deep reinforcement learning for quantum Hamiltonian engineering
• Robust optimization for quantum reinforcement learning control using partial observations
• Fast quantum gate design with deep reinforcement learning using real-time feedback on readout signals