Have a personal or library account? Click to login
Accelerating Atmosphere Modeling: Neural Network Enhancements for Faster NRLMSISE Calculations Cover

Accelerating Atmosphere Modeling: Neural Network Enhancements for Faster NRLMSISE Calculations

Open Access
|Oct 2025

Figures & Tables

Figure 1.

Distribution of input points based on naive uniform distribution of longitude, latitude, and altitude from 500 to 1500 km with the Earth radius taken as 6378 km. There are clearly visible cluster on poles, which should be removed.
Distribution of input points based on naive uniform distribution of longitude, latitude, and altitude from 500 to 1500 km with the Earth radius taken as 6378 km. There are clearly visible cluster on poles, which should be removed.

Figure 2.

Distribution of input points based on proper distribution of longitude, latitude, and altitude from 500 to 1500 km with the Earth radius taken as 6378 km
Distribution of input points based on proper distribution of longitude, latitude, and altitude from 500 to 1500 km with the Earth radius taken as 6378 km

Figure 3.

Distribution of argon (Ar) number densities before (left) and after (right) normalizations. Here, M stands for 1 million and k for 1 thousand.
Distribution of argon (Ar) number densities before (left) and after (right) normalizations. Here, M stands for 1 million and k for 1 thousand.

Figure 4.

Distribution of hydrogen (H) before (left) and after (right) normalizations. Here, M stands for 1 million.
Distribution of hydrogen (H) before (left) and after (right) normalizations. Here, M stands for 1 million.

Figure 5.

Validation loss on test dataset vs. epochs: MSE and REL (mean relative loss of all output features). Here, “k” means thousand.
Validation loss on test dataset vs. epochs: MSE and REL (mean relative loss of all output features). Here, “k” means thousand.

Periodic inputs selected for the model and their ranges for dataset generation

DOY Sin and CosSeconds Sin and CosLong Sin and CosLST Sin and CosLat SinLat Cos
[−1;1][−1;1][−1;1][−1; 1][−1;1][0;1]

Quantile summary (K) of temperature outputs used for network training_ Exospheric and altitude temperature quantile data are the same in taken precision_

Min5th25th50th75th95thMax
Temp.4.51e+028.41e+021.03e+031.19e+031.37e+031.64e+032.24e+03

Final relative percentage loss for each restored number density output_

HeON2O2ArHNO (anom)
0.171%0.414%0.560%0.648%0.850%0.137%0.401%0.168%

Neural network structure_ Biases enabled in each layer_

Layer indexLayer typeInput shapeOutput shapeParametersActivation function
0Dense1464960Hardswish
1Dense64644160Hardswish
2Dense64644160Hardswish
3Dense64644160Hardswish
4Dense6410650No activation
Total parameter numbers14 090

Quantile summary (cm−3) of neutral number-density outputs used for network training_ O(anom) is anomalous atomic oxygen_

HeON2O2ArHNO (anom)
Min2.99e+021.25e-081.19e-233.42e-291.11e-401.37e+036.84e-103.58e+01
5th6.63e+045.70e+012.75e-051.24e-082.23e-148.56e+034.05e+001.00e+03
25th2.16e+055.02e+031.05e-011.18e-043.29e-091.67e+043.58e+024.57e+03
50th4.76e+051.01e+052.52e+015.58e-029.26e-062.96e+047.40e+031.38e+04
75th1.08e+062.08e+065.06e+032.40e+011.86e-025.63e+041.16e+054.03e+04
95th2.83e+063.77e+078.74e+057.98e+033.17e+011.56e+052.49e+061.69e+05
Max1.04e+074.45e+089.67e+072.19e+061.40e+059.57e+062.65e+082.35e+06

Non-periodic inputs selected for the model and their ranges for dataset generation before normalization

Alt, kmF10.7, sfuF10.7 81a, sfuAp, nT
[500;1500][56;504][61;300][0;208]

Final relative percentage loss for restored temperatures and total mass density

Temp. exosphericTemp. at heightTotal mass density (ρTotal)
0.0469%0.0475%0.245%

Evaluation times of compared models for different data_ M stands for million_

Number of NRLMSISE calculationsC++ NRLMSISE time, secondsNeural network NRLMSISE time, seconds
CPU 1 threadCPU 1 threadCPU 12 threadsGPU CUDA
100 0000.262 (1× reference)0.087 (3.0× boost)0.079 (3.3× boost)0.142 (1.85× boost)
500 0001.314 (1× reference)0.310 (4.2× boost)0.192 (6.8× boost)0.143 (9.19× boost)
1M2.566 (1× reference)0.596 (4.3× boost)0.324 (7.9× boost)0.157 (16.3× boost)
1M × 10 times25.598 (1× reference)5.646 (4.5× boost)2.709 (9.5× boost)0.166 (154.2× boost)
1M × 100 times258.448 (1× reference)57.424 (4.5× boost)26.68 (9.7× boost)1.034 (250.0× boost)
DOI: https://doi.org/10.2478/arsa-2025-0007 | Journal eISSN: 2083-6104 | Journal ISSN: 1509-3859
Language: English
Page range: 121 - 136
Submitted on: Apr 18, 2025
|
Accepted on: Oct 1, 2025
|
Published on: Oct 6, 2025
In partnership with: Paradigm Publishing Services
Publication frequency: 4 issues per year

© 2025 Volodymyr Kashyn, Vasyl Choliy, published by Polish Academy of Sciences, Space Research Centre
This work is licensed under the Creative Commons Attribution 4.0 License.