Advancing Machine Learning Applications in Quantum Few-Body Systems

Abstract

This paper presents a general neural network framework for solving quantum few-body systems, extending prior methods to handle diverse particle masses, interaction types, and system configurations. Our architecture, which combines an adaptive step size with the Metropolis-Adjusted Langevin Algorithm for Monte Carlo sampling, accurately approximates the ground-state wave functions of systems featuring harmonic confinement, Gaussian two-body interactions, and including three-body forces. In ten-particle systems, it achieves lower relative energy errors (with respect to the reference values) than previous machine-learning methods. Leveraging GPU-accelerated computation, the method scales favorably with system size while maintaining robust convergence, reduced hyperparameter sensitivity, and stable training. Beyond accurate energy estimation, the model captures spatial distributions and correlation structures, offering physical insights about inter-particle structure. By unifying applicability across identical and nonidentical particles, the proposed approach establishes a versatile computational tool for exploring complex few-body quantum systems, with significant implications for advancing computational models in few-body quantum systems.