SageNet: Fast Neural Network Emulation of the Stiff-amplified Gravitational Waves from Inflation
Accurate modeling of the inflationary gravitational waves (GWs) requires time-consuming, iterative numerical integrations of differential equations to take into account their backreaction on the expansion history. To improve computational efficiency while preserving accuracy, we present the Stiff-amplified Gravitational-wave Emulator Network (SageNet), a deep learning framework designed to replace conventional numerical solvers (code available at https://github.com/YifangLuo/SageNet). SageNet employs a long short-term memory architecture to emulate the present-day energy density spectrum of the inflationary GWs with possible stiff amplification, Ω<SUB>GW</SUB>(f). Trained on a data set of 25,689 numerically generated solutions, SageNet allows accurate reconstructions of Ω<SUB>GW</SUB>(f) and generalizes well to a wide range of cosmological parameters; 90.9% of the test emulations with randomly distributed parameters exhibit errors of under 4%. In addition, SageNet demonstrates its ability to learn and reproduce the artificial, adaptive sampling patterns in numerical calculations, which implement denser sampling of frequencies around changes in spectral indices in Ω<SUB>GW</SUB>(f). The dual capability of learning both physical and artificial features of the numerical GW spectra establishes SageNet as a robust alternative to exact numerical methods. Finally, our benchmark tests show that SageNet reduces the computation time from tens of seconds to milliseconds, achieving a speedup of ~10<SUP>4</SUP> times over standard CPU-based numerical solvers with the potential for further acceleration on GPU hardware. These capabilities make SageNet a powerful tool for accelerating Bayesian inference procedures for extended cosmological models. In a broad sense, the SageNet framework offers a fast, accurate, and generalizable solution to modeling cosmological observables whose theoretical predictions demand costly differential equation solvers.
展开 ▾SageNet:暴胀产生的刚性放大引力波的快速神经网络模拟 · 提出SageNet深度学习框架,用LSTM模拟暴胀引力波能谱,比传统数值求解器快约10^4倍,精度高。
预印本 2025-04-05 · 接收 2025-06-14 · 刊出 2025-07-29 · 收录 2026-08-24