Modeling early-universe energy injection with Dense Neural Networks

The Big Picture

Imagine trying to trace every ripple in a pond after throwing a stone. Not just the first splash, but every secondary wave bouncing off every edge, for billions of years. That’s roughly the challenge physicists face when modeling the early universe. If dark matter (the invisible substance making up most of the universe’s mass) occasionally transforms into bursts of high-energy particles, those particles slam into the hot, charged gas that filled the young cosmos, triggering cascades of secondary particles that ripple through cosmic history.

Understanding those cascades could unlock clues about dark matter’s identity. But the computational cost of tracking them has long been a bottleneck.

The code package DarkHistory was built to handle this problem. It computes how the temperature and electrical state of the gas between galaxies (the intergalactic medium, or IGM) evolve when dark matter injects energy it wouldn’t under standard physics. But the original version carries a heavy price: roughly 18 gigabytes of pre-computed lookup tables that must sit in memory during every simulation. For researchers without access to powerful lab computers, that’s a serious obstacle.

Yitian Sun and Tracy Slatyer at MIT found a smarter way to store all that information. They trained compact neural networks to replace the tables entirely, shrinking the memory footprint by a factor of roughly 400 without meaningfully sacrificing accuracy.

Key Insight: Dense Neural Networks can replace multi-gigabyte physics lookup tables in cosmological simulations, achieving comparable accuracy at a fraction of the storage and memory cost, and making advanced dark matter research accessible on a laptop.

How It Works

The heart of DarkHistory is a set of transfer functions: mathematical objects that take an input bundle of particles (say, a burst of photons from annihilating dark matter) and output the secondary particles produced after one step of cosmic evolution. Think of them as detailed recipes. Given this mix of energetic particles at this moment in cosmic history, here’s what you get after the universe expands a little further. The problem is storage: each transfer function table runs about 1.5 gigabytes, and a standard run needs 12 loaded simultaneously.

Sun and Slatyer’s approach is clean. Instead of storing a giant pre-computed matrix, they train a Dense Neural Network (DNN), a standard multi-layer network where every unit connects to every unit in adjacent layers, to learn each transfer function as a smooth mathematical mapping.

The network takes five inputs: the energy of the incoming particle, the energy of the outgoing particle, the current redshift (a measure of how much the universe has expanded, used as a proxy for cosmic time), the fraction of hydrogen atoms that have lost their electrons, and the equivalent fraction for helium. It outputs a single number: the log of the transfer function value at that point. To reconstruct the full matrix, you evaluate the DNN across the grid of relevant energies.

The training process:

1. Generate a dense grid of training data from the original high-fidelity tables
2. Train separate DNNs for each transfer function type (photon-to-photon, electron-to-photon, etc.)
3. Validate against held-out data, confirming that temperature and ionization histories reproduce the original results to within a few percent

The DNN-based DarkHistory reproduces the hydrogen ionization history and IGM temperature to sub-percent accuracy in the regions that matter most (when species are more than 10% ionized) and to within a few percent everywhere else.

The approach also reproduces subtle distortions in the CMB (Cosmic Microwave Background, the faint glow of ancient light still detectable today), matching the original to within 10%. Current experimental uncertainties from CMB observations are themselves larger, so the approximation is more than adequate for real science.

There’s a bonus baked into the neural network approach: automatic interpolation. The original tables were defined on a fixed grid, and values between grid points required manual interpolation. DNNs are continuous functions. They return sensible values at any parameter combination within the training range. Future versions of DarkHistory could work with flexible binning schemes or sparse training data, opening doors that rigid lookup tables keep shut.

Why It Matters

The practical payoff is immediate: a researcher with a standard laptop can now run DarkHistory without loading 18 gigabytes into RAM. But the deeper point is about scalability.

The original table-based approach has a fundamental flaw. Adding a new physical parameter (say, a dependence on helium temperature or the density of ordinary matter) requires multiplying the table size by the number of steps in that new parameter. Storage demands explode exponentially.

Neural networks don’t work that way. Adding a new input increases the network’s complexity only modestly, roughly a constant-size increase in neurons regardless of how finely you sample the new parameter. That’s the difference between a method that scales and one that hits a wall.

This work fits into a broader wave of machine learning applications in theoretical physics, where neural networks increasingly stand in for expensive computations: full simulations of particle collisions, galaxy formation, and more. Sun and Slatyer aren’t using AI to find new patterns in observations. They’re compressing and generalizing hard-won physical knowledge so more scientists can use it. Future extensions of DarkHistory, from dark matter decaying into multiple channels to position-dependent energy injection, now have a clear computational path forward.

Bottom Line: By swapping 18 GB of lookup tables for a handful of compact neural networks, this work makes dark matter cosmology simulations far more accessible while laying the groundwork for a scalable DarkHistory that can handle more complex physics without exponentially growing storage demands.