An efficient model to estimate thermospheric mass density based on an artificial intelligence approach

Modeling the mass density of thermosphere is essential for understanding the upper atmospheric dynamics and the support of satellites and space station. Such modeling has traditionally relied on either empirical approaches or first-principles physics-based frameworks. The empirical models are computationally efficient with relatively lower accuracy, while the physics-based models are more accurate with the cost of computation time. In this study, a data-driven deep learning model based on a modified U-Net architecture is proposed to estimate the global thermospheric mass density at altitudes of 100 to 500 km. This model directly utilizes input features including time, spatial coordinates, geomagnetic indices, F10.7 solar flux, and solar wind speed. To improve the model performance, we have introduced three main components: a Gated Recurrent Unit (GRU)-enhanced attention mechanism for spatially adaptive feature refinement, a height-adaptive normalization technique to mitigate altitude-induced bias, and a hybrid loss function combining mean absolute error with Laplacian loss to preserve both global structure and fine-scale details. The proposed model achieves accuracy comparable to physics-based models such as TIEGCM, with percentage errors typically below 5%, while the simulation time has been dramatically reduced from tens of minutes to a few seconds. This framework provides an efficient and accurate tool for reconstructing the global thermospheric density and can be potentially utilized for real-time estimation of the thermosphere density under varying geomagnetic and solar conditions.