Papers › SageNet: Fast Neural Network Emulation of the Stiff-amplified Gravitational Waves from...
SageNet: Fast Neural Network Emulation of the Stiff-amplified Gravitational Waves from Inflation
Fan Zhang, Yifang Luo, Bohua Li, Ruihan Cao, Wenjin Peng, Joel Meyers, Paul R. Shapiro
The archive published only this paper's code-link row. Authors, date and abstract are from arXiv's metadata (CC0), read from the Kaggle arXiv metadata snapshot of 2026-09-12 where its title matched the archive's; the title is the archive's.
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 SageNet (Stiff-Amplified Gravitational-wave Emulator Network), a deep learning framework designed to replace conventional numerical solvers. SageNet employs a Long Short-Term Memory architecture to emulate the present-day energy density spectrum of the inflationary GWs with possible stiff amplification, Ω_(GW)(f). Trained on a data set of 25,689 numerically generated solutions, SageNet allows accurate reconstructions of Ω_(GW)(f) and generalizes well to a wide range of cosmological parameters; 89.3% 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 of spectral indices in Ω_(GW)(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 speed-up of ~10⁴ 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.
Code
Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.
Code Syntology ran Syntology
Not run by Syntology. Nothing on this page verifies that the listed code works.
Results from the paper archive 2025-07-28
No leaderboard rows for this paper in the archive.
Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections