npj Quantum Information (Feb 2025)
Direct entanglement detection of quantum systems using machine learning
Abstract
Abstract Entanglement plays a crucial role in advancing quantum technologies and exploring quantum many-body simulations. Here, we introduce a protocol aided by neural networks for measuring entanglement in both equilibrium and non-equilibrium states of local Hamiltonians, with a favorable amount of training data. Our numerical simulations across various Hamiltonian models and qubit configurations reveal that this approach can predict comprehensive entanglement metrics, such as Rényi entropy, for up to 100 qubits using only single-qubit and two-qubit Pauli measurements. Excitingly, future entanglement dynamics beyond the measurement window can be predicted based solely on previous single-qubit traces. Experimentally, we utilize a nuclear spin quantum processor and a neural network to measure entanglement in the ground and dynamical states of a one-dimensional spin chain. The results demonstrate the feasibility of our method in practical experiments. Therefore, our approach offers a promising method for experimentally measuring entanglement in systems with dozens to hundreds of qubits.