How DREAMS are made: Emulating Satellite Galaxy and Subhalo Populations with Diffusion Models and Point Clouds

The Big Picture

Imagine understanding traffic patterns in a city. You could deploy sensors everywhere, run a full physics simulation of every car, or use a statistical model trained on real data to generate realistic flows in seconds. Cosmologists face a similar problem: they need to know how galaxies cluster around dark matter halos (vast, invisible clumps of dark matter that act as gravitational scaffolding), but the most accurate tool available, hydrodynamic simulations, burns through computing resources at a staggering rate. These are supercomputer programs that model the full physics of gas, stars, and dark matter simultaneously.

The stakes are real. Rubin Observatory, the Roman Space Telescope, and DESI will soon deliver data on billions of galaxies. To extract physics from those observations, theorists need fast, accurate models of how galaxies populate dark matter halos. The gap between “fast but crude” and “accurate but slow” has been a stubborn bottleneck.

A team led by Tri Nguyen at MIT, with collaborators across IAIFI, Princeton, Harvard, and Flatiron, has built something that breaks that tradeoff: NeHOD, a neural network that generates realistic galaxy populations with hydrodynamic fidelity at a fraction of the computational cost.

Key Insight: NeHOD uses a diffusion model and Transformer to generate satellite galaxy populations as point clouds, achieving hydrodynamic-level accuracy while remaining as fast as traditional statistical models.

How It Works

The traditional workhorse for placing galaxies in simulations is the Halo Occupation Distribution (HOD), a set of statistical rules that say, roughly, “a halo of this mass probably hosts this many galaxies.” It’s fast, but it ignores spatial detail and the influence of baryonic physics, processes involving ordinary matter like supernova explosions blowing gas out of galaxies. At the other extreme, hydrodynamic simulations solve full fluid dynamics equations for gas, stars, and dark matter all at once. Beautiful physics, but a single run can take millions of CPU hours.

NeHOD sits between these extremes. It learns the mapping from dark matter halo properties to galaxy populations by training on 1,024 high-resolution zoom-in simulations from the DREAMS project (the TNG-Warm DM suite). Zoom-in simulations concentrate computing power on a single halo, and in this case they all model Milky Way-mass halos while varying two key parameters: warm dark matter particle mass and astrophysical feedback strength.

The architecture has two interlocking pieces:

1. A Transformer encoder reads the dark matter halo as a cloud of particle positions and compresses it into a compact summary of the halo’s structure, mass profile, and shape.
2. A variational diffusion model generates the satellite galaxy population conditioned on that summary and the simulation parameters. It learns to turn random noise into structured data step by step, iteratively refining a set of points into realistic galaxy positions, velocities, stellar masses, and concentration parameters.

Treating galaxies as a point cloud (individual points in space rather than values on a fixed grid) lets NeHOD avoid the resolution limits that constrain grid-based approaches. It resolves spatial scales down to the simulation’s native resolution, which is critical for studying small satellite galaxies orbiting close to their host.

NeHOD correctly reproduces the key statistics of satellite galaxy populations: the subhalo mass function, the stellar-to-halo mass relation, the concentration-mass relation, and spatial clustering statistics, all as a function of the simulation parameters. When warm dark matter is lighter and suppresses small-scale structure, NeHOD generates fewer low-mass satellites. When feedback parameters change, it adjusts stellar masses accordingly.

The model also captures correlations between properties that simpler models miss. In a real galaxy system, a satellite’s position, velocity, mass, and concentration all co-evolved under the same gravitational and baryonic forces. The diffusion model learns these joint distributions implicitly, with no need for the researcher to specify them by hand.

The computational payoff is large: populating a halo with NeHOD takes seconds on a GPU. The equivalent hydrodynamic simulation takes months on a supercomputer cluster.

Why It Matters

This work connects two major efforts in modern cosmology. Surveys like SAGA and ELVES are already cataloging satellite galaxies around Milky Way analogs, with Rubin and Roman set to expand that census by orders of magnitude. At the same time, physicists are hunting for signatures of warm dark matter, a hypothetical variant whose fast-moving particles would smooth out small clumps and leave detectable gaps in satellite galaxy counts. Detecting that signal requires running thousands of forward models across wide parameter spaces, and only a fast emulator like NeHOD makes that feasible.

Accurate subhalo populations also open doors for strong gravitational lensing studies, where a massive galaxy bends light from a distant source into arcs or rings. The exact pattern depends on the detailed matter distribution, including small satellites. They matter equally for satellite kinematics, using dwarf galaxy motions to infer the mass profile of their host halo. The authors flag both as target applications.

There is a broader lesson here too. Point-cloud diffusion models, borrowed from machine learning’s work on 3D object generation, turn out to be a natural fit for astrophysical structures that are inherently sparse and irregular in space.

Bottom Line: NeHOD delivers hydrodynamic-quality galaxy populations at HOD-level speed, unlocking fast, parameter-spanning exploration of dark matter physics and galaxy formation with a neural generative model trained on the DREAMS simulation suite.