Data underlying the study on Reinforcing Prediction of Atomic Energy Level Transitions with Optimized PINN-Bi-LSTM

DOI:10.4121/65174478-e69c-400b-ab41-a2f897ceb407.v1
The DOI displayed above is for this specific version of this dataset, which is currently the latest. Newer versions may be published in the future. For a link that will always point to the latest version, please use
DOI: 10.4121/65174478-e69c-400b-ab41-a2f897ceb407

Datacite citation style

Kang, Wentao (2025): Data underlying the study on Reinforcing Prediction of Atomic Energy Level Transitions with Optimized PINN-Bi-LSTM. Version 1. 4TU.ResearchData. dataset. https://doi.org/10.4121/65174478-e69c-400b-ab41-a2f897ceb407.v1
Other citation styles (APA, Harvard, MLA, Vancouver, Chicago, IEEE) available at Datacite

Dataset

Accurate prediction of atomic energy level transitions is essential for advancements in quantum spectroscopy, quantum device optimization, and atomic dynamics research. Traditional methods often encounter challenges such as high computational cost or limited physical interpretability. To overcome these limitations, the Physics-Informed Bi-LSTM (PINN-Bi-LSTM) model was proposed. This framework integrates Bidirectional Long Short-Term Memory (Bi-LSTM) networks with Physics-Informed Neural Networks (PINNs). The Bi-LSTM component captures temporal patterns in atomic state evolution from time-series data (e.g., spectral emission records), while the PINN component enforces quantum mechanical constraints (e.g., energy conservation, transition probability rules) through a physics-based loss term. The model was validated using both synthetic and experimental datasets and compared against benchmark methods, including ab initio simulations, standalone Bi-LSTMs, and LSTM–ARIMA hybrids. Results demonstrated that the PINN-Bi-LSTM model achieved superior accuracy (0.92), lower mean squared error (0.01), and stronger generalizability to complex systems (e.g., multi-electron atoms, dynamically perturbed states) compared to conventional approaches. Furthermore, the model maintained robust performance in noisy or sparse datasets, attributed to its dual emphasis on data efficiency and physical rigor. This work advances quantum state prediction by unifying temporal modeling capabilities with fundamental physics, offering applications in real-time spectroscopy and quantum device optimization.

History

  • 2025-09-29 first online, published, posted

Publisher

4TU.ResearchData

Format

script/.py; spreadsheet/.csv; MATLAB/.m

Funding

Organizations

School of Information Engineering, Beijing Institute of Graphic Communication

DATA

Files (2)