Unsupervised learning for the systematic identification of nondispersive wave packets in driven helium
Nondispersive wave packets in driven helium are long-lived quantum states that follow classical resonant orbits without spreading. Their identification typically requires detailed analysis of phase-space structures and extensive exploration of parameter regimes. In this work, we introduce an unsupervised learning approach to automate the identification of physically relevant states in the driven helium atom. Using a Floquet-based description, quantum states are computed and represented as probability distributions in configuration and phase space, which serve as input to a convolutional neural network that constructs a low-dimensional embedding of the data. Clustering in the embedding space reveals distinct classes of quantum states. By combining geometric analysis, physical parameter inspection, and time-evolution studies, we identify clusters corresponding to frozen planet states and nondispersive wave packets. The method successfully recovers known NDWP regimes without prior labeling, demonstrating that the learned representation captures physically meaningful structures in a systematic and automated manner. These results establish unsupervised representation learning as an effective tool for the systematic analysis of complex quantum datasets.