Feedback-Conditional 3D Reconstruction of Cosmic Baryons with Flow Matching
summary
The gist
The gist The model learns to predict the posterior distribution of gas density fields given stellar mass-weighted galaxy density, conditioned on cosmological parameters and the baryon suppression
In short
The model learns to predict 3D cosmic gas density fields from galaxy surveys by conditioning its prediction on galaxy density, cosmological parameters, and a measure of baryonic feedback via a power spectrum suppression term. This allows for the reconstruction of per-sightline Fast Radio Burst (FRB) dispersion measures and their uncertainties. The method generalizes well to large volumes and provides an inference path to constrain astrophysical feedback models.
Key concepts
- Dispersion Measure (DM)
- DM is a measure of the column density of cosmic baryons along a line of sight, dominated by the intergalactic medium. It is crucial for understanding the gas distribution probed by Fast Radio Bursts (FRBs). The model estimates DM at each sky position by integrating the reconstructed electron density field along these lines.
- Baryonic Feedback Power Spectrum ($sP(k)$)
- This term quantifies how baryonic physics, such as AGN feedback, redistributes matter across different spatial scales. It suppresses the power spectrum of matter on small scales (scales $k ext{ around } 1-50 h ext{ Mpc}^{-1}$), reflecting the effect of feedback processes on galaxy formation and baryon distribution.
- Flow Matching Model
- This is a generative model approach that learns to map an initial, simple Gaussian distribution of data (like redshift space galaxies) to the complex target distribution (the real 3D gas density field). It achieves this by learning a velocity field that transports samples along a linear interpolation path between the two distributions.
- IllustrisTNG/CAMELS50
- These are large-scale cosmological simulations used as the training ground for the generative model. They provide realistic mock data of galaxy and gas distributions across various cosmological parameters and different feedback scenarios, enabling the model to learn how galaxy properties relate to underlying cosmic baryon fields.
Terminology used across episodes
This episode discusses
- Feedback-Conditional 3D Reconstruction of Cosmic Baryons with Flow Matching · Paper Radio
- A Catalog of Local Universe Fast Radio Bursts from CHIME/FRB and the KKO Outrigger
- Learning the Universe with the 2nd Generation of CAMELS: Varying 35 parameters of the IllustrisTNG model in (50Mpc/h) cubed boxes
- Evidence for large baryonic feedback at low and intermediate redshifts from kinematic Sunyaev-Zel'dovich observations with ACT and DESI photometric galaxies
- The DSA-2000 -- A Radio Survey Camera
- BaryonBridge: Stochastic Interpolant Model for Fast Hydrodynamical Simulations
- Reconstruction of Continuous Cosmological Fields from Discrete Tracers with Graph Neural Networks
- Flow Matching for Generative Modeling
- Fast Baryonic Field Painting for Sunyaev-Zel'dovich Analyses: Transfer Function vs. Hybrid Effective Field Theory
- The CGM with local universe FRBs: evidence of strong AGN feedback in a massive elliptical galaxy
- Can denoising diffusion probabilistic models generate realistic astrophysical fields?
- Probabilistic reconstruction of Dark Matter fields from biased tracers using diffusion models
- U-Net: Convolutional Networks for Biomedical Image Segmentation
- The Commensal Real-time ASKAP Fast Transient incoherent-sum survey
- Signatures of Suppressed Matter Clustering revealed by Fast Radio Bursts
- Revealing the 3D Cosmic Web through Gravitationally Constrained Neural Fields
The paper
Feedback-Conditional 3D Reconstruction of Cosmic Baryons with Flow Matching · Read on arXiv
Maryam Hussaini, Aizhan Akhmetzhanova, Carolina Cuesta-Lazaro, Liam Connor
Center for Astrophysics | Harvard & Smithsonian · Department of Physics, Harvard University
The complex relationship between galaxies and diffuse gas at large scales depends strongly on astrophysical feedback and cosmology. The dispersion measure (DM) of Fast Radio Bursts (FRBs) is a column of cosmic baryon density and is dominated by the intergalactic medium, which is not necessarily captured by traditional halo model prescriptions. We introduce a generative model to reconstruct 3D gas density fields from galaxy surveys, including redshift space distortions. The probabilistic model is trained on the CAMELS50 simulation suite (IllustrisTNG) and learns to predict the posterior distribution of gas fields conditioned on stellar mass-weighted galaxy density, cosmological parameters, the minimum stellar mass threshold for the galaxy sample, and the baryon suppression power spectrum as a measure of baryonic feedback. Our method uses the 3D gas reconstruction to estimate per-sightline FRB DMs and their posteriors. We validate the model on test boxes by showing agreement between predicted and true DM across individual lines of sight and the overall DM distribution. The model generalizes to a box size about,4 cubed times larger in volume than the training boxes and can cover the volumes spanned by cosmological FRB sightlines. The model's 3D gas density predictions vary with input power spectrum and feedback scenarios, offering an inference path to constrain feedback models from FRBs and other probes such as X-rays, the SZ effect, and weak lensing.
Transcript
Introduction to the show: ident: Astrophysics Radio. Generated commentary on the latest astrophysics papers.
Vera: Today's paper: "Feedback-Conditional 3D Reconstruction of Cosmic Baryons with Flow Matching".
Jocelyn: The gist The model learns to predict the posterior distribution of gas density fields given stellar mass-weighted galaxy density, conditioned on cosmological parameters and the baryon suppression spectrum.
Vera: First, who's behind it and why it matters.
Title and authors: Vera: So Jocelyn, we're looking at this paper now entitled "Feedback-Conditional three dee Reconstruction of Cosmic Baryons with Flow Matching <ref:2610.10741#pg1,Feedback-Conditional 3D Reconstruction of Cosmic Baryons with Flow Matching>." It seems like the core idea here is to use a generative model to reconstruct those three-dimensional gas density fields from the galaxy surveys we observe, including redshift space distortions.
Jocelyn: Right, Vera. I read the abstract again, and it sounds like they're not just trying to map where galaxies are; they’re trying to find out what the gas looks like in three dee space—the actual density distribution—and they want that result conditioned on several things: the galaxy density field, cosmological parameters, a minimum stellar mass threshold for their sample, and this baryon suppression power spectrum <ref:2610.10741#pg1,minimum stellar mass threshold for>.
Subrahmanyan: That suppression spectrum is key because it directly quantifies how baryonic physics, specifically feedback from things like Active Galactic Nuclei or AGN, redistributes matter across different scales. It's the mechanism that breaks the simple connection between dark matter and baryons that we usually expect in dark matter simulations.
Vera: Exactly. So, this model aims to predict a posterior distribution of these gas fields given the galaxy density field, cosmology, and that suppression spectrum. That means instead of giving us one single guess for the gas density, we get a whole range of possibilities—a distribution—which is really important for understanding the real physics out there.
Jocelyn: And they do this using a flow matching model to map a base Gaussian distribution onto the target data distribution. They train this by learning a velocity field that transports samples from that simple starting point to the complex, observed galaxy field, and they use a three dee conditional U-Net as their backbone for reconstructing these fields <ref:2610.10741#pg1>.
Subrahmanyan: The flow matching approach itself is interesting because it learns the transport mechanism directly without needing to write down an explicit forward model likelihood. This is different from some other methods that require you to assume a specific way the data relates to the underlying physics, which can introduce uncertainties if that assumption isn't perfect.
Title and authors: Vera: It sounds like they are trying to bypass some of those explicit likelihood problems by using this implicit likelihood approach, which they mention on page three <ref:2610.10741#pg1>. And the output they get is a gas density field that you can then integrate to predict the dispersion measures of Fast Radio Bursts, or FRBs.
Jocelyn: That integration step is what connects it back to what we actually observe with FRBs. Because the dispersion measure is essentially a column of cosmic baryon density, reconstructing that three dee gas field allows them to predict the distribution of DMs for those sightlines <ref:2610.10741#pg1,a column of cosmic baryon density>.
Subrahmanyan: That’s where the real power comes in, though. They're not just giving you a single guess; they are providing a posterior over these gas fields. If you sample from this posterior, each sample integrates to a different DM per voxel, so the spread across those samples tells you directly how uncertain you are about that sightline's DM.
Vera: That uncertainty quantification is what I find really compelling. Instead of just saying "the DM is X," they can tell us, "the DM could be anywhere in this range based on our models and data." It moves us closer to a true understanding of the underlying physics.
Jocelyn: And they’ve shown that the model generalizes well, even to much larger volumes than what they used for training, specifically mentioning TNG300-one <ref:2610.10741#pg1>. That suggests this isn't just a local fix but something scalable for mapping out larger cosmic structures relevant to FRB sightlines.
Subrahmanyan: Scaling up to volumes like TNG300-one is important because it tests whether the learned physics holds when you move away from the specific regions where the training data was generated <ref:2610.10741#pg1>. It shows that their understanding of how feedback shapes gas remains robust across different cosmic environments, as long as you condition on those key parameters.
Vera: And they’ve also found something interesting about how feedback itself affects the gas structure. They learned that stronger feedback leads to smoother gas distributions, while weaker feedback results in tighter filaments and denser nodes in the reconstruction. That’s a tangible result linking the model's output back to the physical processes we expect from galaxy evolution simulations.
Title and authors: Jocelyn: It feels like they are using this framework to put constraints on our models of AGN feedback by seeing how sensitive the gas reconstruction is to changes in that suppression spectrum. So, if we can see that relationship, it gives us a way to probe those feedback models more directly through these galaxy surveys.
Subrahmanyan: That’s a sophisticated way to connect simulation outputs with observational constraints. It’s moving beyond just matching galaxy positions and into constraining the physical drivers of how that gas is distributed across cosmic time and space.
Vera: So, to wrap up this discussion on the paper "Feedback-Conditional three dee Reconstruction of Cosmic Baryons with Flow Matching," what we have here is a probabilistic tool that lets us move from observing galaxies to inferring a full distribution of the underlying cosmic gas density, complete with uncertainty estimates based on cosmological parameters and feedback strength <ref:2610.10741#pg1,Feedback-Conditional 3D Reconstruction of Cosmic Baryons with Flow Matching>.
Jocelyn: It’s a way to bridge the gap between the complex physics happening inside halos and what we can actually measure through things like FRB dispersion measures across vast cosmic scales.
Subrahmanyan: Ultimately, it helps constrain models of galaxy formation by showing how variations in feedback sculpt the gas density field in a way that is observable through the distribution of galaxies.
Vera: And they’ve shown this method can work on very large volumes, which is a big step for applying these techniques to real cosmological surveys that look at huge swathes of space.
Jocelyn: It gives us a more robust way to think about the baryons in the universe, not just as something inside halos but as a large-scale structure whose distribution is shaped by complex astrophysical processes.
Subrahmanyan: This work provides a new data-driven path toward understanding how feedback regulates star formation and black hole growth across cosmic history.
Vera: That’s what we have with the paper "Feedback-Conditional three dee Reconstruction of Cosmic Baryons with Flow Matching <ref:2610.10741#pg1,Feedback-Conditional 3D Reconstruction of Cosmic Baryons with Flow Matching>." We'll be looking at this again next time, but for now, that’s what we’ve got.
The paper's summary: Vera: So, this paper is about taking those galaxy surveys and using this new flow matching method to figure out what the gas density field really looks like in three dimensions.
Jocelyn: Right, it’s basically a way to reconstruct that three dee gas map from the galaxy data we get, but it ties that reconstruction directly to the cosmological parameters and how much feedback is happening in those systems.
Subrahmanyan: The core idea is using this generative model to predict the posterior distribution of these gas fields, given what we see in galaxy density, plus all those physical conditions like cosmology and that suppression spectrum.
Vera: It’s not just giving us one guess for the gas structure, but a whole range of possibilities. That means we get a measure of how much we actually know about the distribution.
Jocelyn: And that uncertainty is really useful because it directly relates to something we can see in other experiments, like the dispersion measures of Fast Radio Bursts.
Subrahmanyan: Because each possible gas configuration you sample from that posterior maps to a different dispersion measure per voxel, so the spread tells you the uncertainty on that sightline’s DM.
Vera: They trained this using high-quality simulations, specifically IllustrisTNG, and they used this suppression spectrum as a way to quantify that feedback effect.
Jocelyn: And they showed how sensitive the reconstruction is to it; stronger feedback leads to smoother gas distributions, while weaker feedback makes the filaments tighter.
Subrahmanyan: So they’re using this model not just to reconstruct things, but also to constrain our physical models of AGN feedback by seeing how the gas responds when you change that suppression spectrum input.
Vera: They even tested it on much larger volumes than what they trained on, like TNG300-one which shows the method can work for huge cosmic areas relevant to FRB sightlines.
Jocelyn: That’s a big deal because those large volumes are exactly where we need to map out the gas distribution for these distant events.
Subrahmanyan: What this means is that we can use these galaxy surveys to probe how feedback regulates star formation and black hole growth across vast scales in the universe.
The paper's improvements: Vera: So, we’ve seen how this model reconstructs the gas field conditioned on galaxy density and feedback parameters, but what are they suggesting to make it better?
Jocelyn: They are focusing on making the flow matching part more robust across different cosmological scenarios.
Subrahmanyan: They want to ensure that when you condition on a different baryon suppression spectrum, say a stronger one versus a weaker one, the resulting gas reconstruction reflects that physical difference accurately.
Vera: It sounds like they are trying to build in an even tighter link between the input parameters—the cosmology and the feedback strength—and the final output gas map.
Jocelyn: Yes, they want to make sure it doesn't just learn a mapping for one specific feedback scenario, but generalizes better across the whole range of what those simulations produce.
Subrahmanyan: That’s crucial because if we can constrain how much feedback alters the gas structure based on these survey results, we get a much better handle on how black holes affect galaxy evolution overall.
Vera: They also mentioned something about improving the training setup itself. They're tweaking the way they sample those one thousand two hundred eighty-three voxel grids for the input galaxies.
Jocelyn: It’s about getting a more accurate representation of the actual galaxy positions before feeding them into the flow matching network, especially when dealing with varying stellar mass thresholds.
Subrahmanyan: From a theoretical standpoint, this means that the physics captured in their training set is being leveraged more efficiently to constrain the underlying baryonic processes.
Vera: And they’re pushing for better ways to handle the scaling up of this reconstruction, making sure it stays accurate when applied to even larger volumes like TNG300-one.
Jocelyn: So they are essentially aiming for a method that is not just accurate in a small box, but actually works reliably on the largest cosmic structures we care about.
Subrahmanyan: If this works as they suggest, it opens up a new way to use these galaxy surveys to map out the distribution of matter on scales far beyond what was possible before.
Conclusion: Vera: So, to wrap up this discussion on "Feedback-Conditional three dee Reconstruction of Cosmic Baryons with Flow Matching," we’ve seen how they use generative models and flow matching to map out the three dimensional gas density field from galaxy surveys, conditioned on feedback physics.
Jocelyn: It really shows how observational data, even when messy like galaxy distributions, can be used to infer the underlying physical state of the cosmic gas.
Subrahmanyan: This work connects those large scale structures we see in the sky to the detailed processes happening inside galaxies via that baryon suppression spectrum.
Vera: The implication is that we can move beyond just seeing where galaxies are and start reconstructing what they are made of, which is the gas itself, with a measure of how certain we are about it.
Jocelyn: That uncertainty quantification is pretty powerful because it gives us a realistic picture instead of just one single number for the DM distribution.
Subrahmanyan: It helps constrain our models of galaxy formation by showing exactly how feedback sculpts the gas density field across different scales and environments.
Vera: They’ve also shown that this method can be applied to much larger volumes, which is a big step for applying these techniques to real cosmological surveys that look at huge swathes of space.
Jocelyn: It means we can start using these galaxy maps to predict things like the dispersion measures of Fast Radio Bursts over vast cosmic distances.
Subrahmanyan: So this is a data-driven path toward understanding how feedback regulates star formation and black hole growth across cosmic history.
More episodes
- 2610.12394-Inflow-driven galaxy evolution - II: A hierarchy in the scaling relations of star-forming galaxies
- 2610.10750-Primordial black holes
- 2610.12462-Disk Structure May Determine AGN Variability and Explain the Accretion Disk Size Problem: No Broad Line Region Required
- 2610.10535-A weakly modelled view of the joint compact-binary mass plane: population structure and spectral-siren cosmology
- 2610.10669-Impact of site-dependent noise configurations on Einstein Telescope science at low frequency
- 2610.10674-Interpretation of Nanohertz Gravitational Waves via Cosmic Strings in Post-BBN Matter and Kination Domination
- 2610.10677-Primordial black holes as a natural consequence of scale-invariance
- 2610.10688-Matter-Antimatter Asymmetry from Dirac-Majorana Flip-Flop
- 2610.10695-The Origin of the Fermi Halo-like Emission: A New Model of the Fermi Bubbles
- 2610.10779-Debiasing Local Cluster Infall Reveals: Application to the Virgo Cluster