arXiv subjects

Justin Alsing

Publications and source records attributed to Justin Alsing.

At least 19 recordsLinked to original sources

arXiv2026

pop-cosmos: Forward modeling KiDS-1000 redshift distributions using realistic galaxy populations

The accuracy of the cosmological constraints from Stage~IV galaxy surveys will be limited by how well the galaxy redshift distributions can be inferred. We have addressed this challenging problem for the Kilo-Degree Survey (KiDS) cosmic shear sample by developing a forward-modeling framework with two main ingredients: (1) the \texttt{pop-cosmos} generative model for the evolving galaxy population, calibrated on \textit{Spitzer} IRAC $\textit{Ch.\,1}<26$ galaxies from COSMOS2020; and (2) a data model for noise and selection, machine-learned from the SURFS-based KiDS-Legacy-Like Simulations (SKiLLS). Applying KiDS tomographic binning to our synthetic photometric data, we infer redshift distributions in each of five bins directly from the population and data models, bypassing the need for spectroscopic reweighting. Keeping the data model fixed, we compare results using two different galaxy population models: \texttt{pop-cosmos}; and \texttt{shark}, the semi-analytic galaxy formation model used in SKiLLS. In the first ($0.1<z<0.3$) and last ($0.9<z<1.2$) tomographic bins we find systematic differences in the mean redshifts of $\Delta z\sim0.05$-$0.1$, comparable to the reported uncertainties from spectroscopic reweighting methods. This work paves the way for accurate redshift distribution calibration for Stage~IV surveys directly through forward modeling, thus providing an independent cross-check on spectroscopic-based calibrations which avoids their selection biases and incompleteness. We will use the \texttt{pop-cosmos} redshift distributions in an upcoming full KiDS cosmology reanalysis.

astro-ph.CO↗

arXiv2025

pop-cosmos: Star formation over 12 Gyr from generative modelling of a deep infrared-selected galaxy catalogue

We study star formation over 12 Gyr using pop-cosmos, a generative model trained on 26-band photometry of 420,000 COSMOS2020 galaxies (IRAC Ch.1 $<26$). The model learns distributions over 16 SPS parameters via score-based diffusion, matching observed colours and magnitudes. We compute the star formation rate density (SFRD) to $z=3.5$ by directly integrating individual galaxy SFRs. The SFRD peaks at $z=1.3\pm0.1$, with peak value $0.08\pm0.01$ M$_{\odot}$ yr$^{-1}$ Mpc$^{-3}$. We classify star-forming (SF) and quiescent (Q) galaxies using specific SFR $<10^{-11}$ yr$^{-1}$, comparing with $NUVrJ$ colour selection. The sSFR criterion yields up to 20% smaller quiescent fractions across $0 0.95$) during the most recent $\sim$300 Myr, then sharp decorrelation with earlier star-forming epochs, marking clear quenching transitions. Massive ($10<\log_{10}(M_*/$M$_{\odot})<11$) galaxies quench on a time-scale of $\sim1$ Gyr, with mass assembly concentrated in their first 3.5 Gyr. Finally, AGN activity (infrared luminosity) peaks as massive ($\sim10^{10.5}$ M$_\odot$) galaxies approach the transition between star-forming and quiescent states, declining sharply once quiescence is established. This provides evidence that AGN feedback operates in a critical regime during the $\sim1$ Gyr quenching transition.

astro-ph.GA↗

arXiv2025

pop-cosmos: Insights from generative modeling of a deep, infrared-selected galaxy population

We present an extension of the pop-cosmos model for the evolving galaxy population up to redshift $z\sim6$. The model is trained on distributions of observed colors and magnitudes, from 26-band photometry of $\sim420,000$ galaxies in the COSMOS2020 catalog with Spitzer IRAC $\textit{Ch. 1}<26$. The generative model includes a flexible distribution over 16 stellar population synthesis (SPS) parameters, and a depth-dependent photometric uncertainty model, both represented using score-based diffusion models. We use the trained model to predict scaling relationships for the galaxy population, such as the stellar mass function, star-forming main sequence, and gas-phase and stellar metallicity vs. mass relations, demonstrating reasonable-to-excellent agreement with previously published results. We explore the connection between mid-infrared emission from active galactic nuclei (AGN) and star-formation rate, finding high AGN activity for galaxies above the star-forming main sequence at $1\lesssim z\lesssim 2$. Using the trained population model as a prior distribution, we perform inference of the redshifts and SPS parameters for 429,669 COSMOS2020 galaxies, including 39,588 with publicly available spectroscopic redshifts. The resulting redshift estimates exhibit minimal bias ($\text{median}[\Delta_z]=-8\times10^{-4}$), scatter ($\sigma_\text{MAD}=0.0132$), and outlier fraction ($6.19\%$) for the full $0<z<6$ spectroscopic compilation. These results establish that pop-cosmos can achieve the accuracy and realism needed to forward-model modern wide--deep surveys for Stage IV cosmology. We publicly release pop-cosmos software, mock galaxy catalogs, and COSMOS2020 redshift and SPS parameter posteriors.

astro-ph.GA↗

arXiv2025

Impact of redshift distribution uncertainties on Lyman-break galaxy cosmological parameter inference

A significant number of Lyman-break galaxies (LBGs) with redshifts 3 < z < 5 are expected to be observed by the upcoming Vera C. Rubin Observatory Legacy Survey of Space and Time (LSST). This will enable us to probe the universe at higher redshifts than is currently possible with cosmological galaxy clustering and weak lensing surveys. However, accurate inference of cosmological parameters requires precise knowledge of the redshift distributions of selected galaxies, where the number of faint objects expected from LSST alone will make spectroscopic based methods of determining these distributions extremely challenging. To overcome this difficulty, it may be possible to leverage the information in the large volume of photometric data alone to precisely infer these distributions. This could be facilitated using forward models, where in this paper we use stellar population synthesis (SPS) to estimate uncertainties on LBG redshift distributions for a 10 year LSST (LSSTY10) survey. We characterise some of the modelling uncertainties inherent to SPS by introducing a flexible parameterisation of the galaxy population prior, informed by observations of the galaxy stellar mass function (GSMF) and cosmic star formation density (CSFRD). These uncertainties are subsequently marginalised over and propagated to cosmological constraints in a Fisher forecast. Assuming a known dust attenuation model for LBGs, we forecast constraints on the sigma8 parameter comparable to Planck cosmic microwave background (CMB) constraints.

astro-ph.CO↗

arXiv2024

pop-cosmos: Scaleable inference of galaxy properties and redshifts with a data-driven population model

We present an efficient Bayesian method for estimating individual photometric redshifts and galaxy properties under a pre-trained population model (pop-cosmos) that was calibrated using purely photometric data. This model specifies a prior distribution over 16 stellar population synthesis (SPS) parameters using a score-based diffusion model, and includes a data model with detailed treatment of nebular emission. We use a GPU-accelerated affine invariant ensemble sampler to achieve fast posterior sampling under this model for 292,300 individual galaxies in the COSMOS2020 catalog, leveraging a neural network emulator (Speculator) to speed up the SPS calculations. We apply both the pop-cosmos population model and a baseline prior inspired by Prospector-$\alpha$, and compare these results to published COSMOS2020 redshift estimates from the widely-used EAZY and LePhare codes. For the $\sim 12,000$ galaxies with spectroscopic redshifts, we find that pop-cosmos yields redshift estimates that have minimal bias ($\sim10^{-4}$), high accuracy ($\sigma_\text{MAD}=7\times10^{-3}$), and a low outlier rate ($1.6\%$). We show that the pop-cosmos population model generalizes well to galaxies fainter than its $r<25$ mag training set. The sample we have analyzed is $\gtrsim3\times$ larger than has previously been possible via posterior sampling with a full SPS model, with average throughput of 15 GPU-sec per galaxy under the pop-cosmos prior, and 0.6 GPU-sec per galaxy under the Prospector prior. This paves the way for principled modeling of the huge catalogs expected from upcoming Stage IV galaxy surveys.

astro-ph.CO↗

arXiv2024

Scaling-laws for Large Time-series Models

Scaling laws for large language models (LLMs) have provided useful guidance in training ever larger models for predictable performance gains. Time series forecasting shares a similar sequential structure to language, and is amenable to large-scale transformer architectures. Here we show that foundational decoder-only time series transformer models exhibit analogous scaling-behavior to LLMs, with architectural details (aspect ratio and number of heads) having a minimal effect over broad ranges. We assemble a large corpus of heterogenous time series data on which to train, and establish for the first time power-law scaling with parameter count, dataset size, and training compute, spanning five orders of magnitude.

cs.LG↗

arXiv2024

pop-cosmos: A comprehensive picture of the galaxy population from COSMOS data

We present pop-cosmos: a comprehensive model characterizing the galaxy population, calibrated to $140,938$ ($r<25$ selected) galaxies from the Cosmic Evolution Survey (COSMOS) with photometry in $26$ bands from the ultra-violet to the infra-red. We construct a detailed forward model for the COSMOS data, comprising: a population model describing the joint distribution of galaxy characteristics and its evolution (parameterized by a flexible score-based diffusion model); a state-of-the-art stellar population synthesis (SPS) model connecting galaxies' instrinsic properties to their photometry; and a data-model for the observation, calibration and selection processes. By minimizing the optimal transport distance between synthetic and real data we are able to jointly fit the population- and data-models, leading to robustly calibrated population-level inferences that account for parameter degeneracies, photometric noise and calibration, and selection. We present a number of key predictions from our model of interest for cosmology and galaxy evolution, including the mass function and redshift distribution; the mass-metallicity-redshift and fundamental metallicity relations; the star-forming sequence; the relation between dust attenuation and stellar mass, star formation rate and attenuation-law index; and the relation between gas-ionization and star formation. Our model encodes a comprehensive picture of galaxy evolution that faithfully predicts galaxy colors across a broad redshift ($z<4$) and wavelength range.

astro-ph.GA↗

arXiv2024

Data-Space Validation of High-Dimensional Models by Comparing Sample Quantiles

We present a simple method for assessing the predictive performance of high-dimensional models directly in data space when only samples are available. Our approach is to compare the quantiles of observables predicted by a model to those of the observables themselves. In cases where the dimensionality of the observables is large (e.g. multiband galaxy photometry), we advocate that the comparison is made after projection onto a set of principal axes to reduce the dimensionality. We demonstrate our method on a series of two-dimensional examples. We then apply it to results from a state-of-the-art generative model for galaxy photometry (pop-cosmos; arXiv:2402.00935) that generates predictions of colors and magnitudes by forward simulating from a 16-dimensional distribution of physical parameters represented by a score-based diffusion model. We validate the predictive performance of this model directly in a space of nine broadband colors. Although motivated by this specific example, we expect that the techniques we present will be broadly useful for evaluating the performance of flexible, non-parametric population models of this kind, and other settings where two sets of samples are to be compared.

astro-ph.IM↗

arXiv2023

Measuring the nuclear equation of state with neutron star-black hole mergers

Gravitational-wave (GW) observations of neutron star-black hole (NSBH) mergers are sensitive to the nuclear equation of state (EOS). We present a new methodology for EOS inference with non-parametric Gaussian process (GP) priors, enabling direct constraints on the pressure at specific densities and the length-scale of correlations on the EOS. Using realistic simulations of NSBH mergers, incorporating both GW and electromagnetic (EM) selection to ensure sample purity, we find that a GW detector network operating at O5-sensitivities will constrain the radius of a $\unit[1.4]{M_{\odot}}$ NS and the maximum NS mass with $1.6\%$ and $13\%$ precision, respectively. With the same sample, the projected constraint on the length-scale of correlations in the EOS is $\geq~\unit[3.2]{MeV~fm^{-3}}$. These results demonstrate strong potential for insights into the nuclear EOS from NSBH systems, provided they are robustly identified.

gr-qc↗

arXiv2023

Optimal simulation-based Bayesian decisions

We present a framework for the efficient computation of optimal Bayesian decisions under intractable likelihoods, by learning a surrogate model for the expected utility (or its distribution) as a function of the action and data spaces. We leverage recent advances in simulation-based inference and Bayesian optimization to develop active learning schemes to choose where in parameter and action spaces to simulate. This allows us to learn the optimal action in as few simulations as possible. The resulting framework is extremely simulation efficient, typically requiring fewer model calls than the associated posterior inference task alone, and a factor of $100-1000$ more efficient than Monte-Carlo based methods. Our framework opens up new capabilities for performing Bayesian decision making, particularly in the previously challenging regime where likelihoods are intractable, and simulations expensive.

stat.ML↗

arXiv2023

Fishnets: Information-Optimal, Scalable Aggregation for Sets and Graphs

Set-based learning is an essential component of modern deep learning and network science. Graph Neural Networks (GNNs) and their edge-free counterparts Deepsets have proven remarkably useful on ragged and topologically challenging datasets. The key to learning informative embeddings for set members is a specified aggregation function, usually a sum, max, or mean. We propose Fishnets, an aggregation strategy for learning information-optimal embeddings for sets of data for both Bayesian inference and graph aggregation. We demonstrate that i) Fishnets neural summaries can be scaled optimally to an arbitrary number of data objects, ii) Fishnets aggregations are robust to changes in data distribution, unlike standard deepsets, iii) Fishnets saturate Bayesian information content and extend to regimes where MCMC techniques fail and iv) Fishnets can be used as a drop-in aggregation scheme within GNNs. We show that by adopting a Fishnets aggregation scheme for message passing, GNNs can achieve state-of-the-art performance versus architecture size on ogbn-protein data over existing benchmarks with a fraction of learnable parameters and faster training time.

cs.LG↗

arXiv2022

Neural Stellar Population Synthesis Emulator for the DESI PROVABGS

The Probabilistic Value-Added Bright Galaxy Survey (PROVABGS) catalog will provide the posterior distributions of physical properties of $>10$ million DESI Bright Galaxy Survey (BGS) galaxies. Each posterior distribution will be inferred from joint Bayesian modeling of observed photometry and spectroscopy using Markov Chain Monte Carlo sampling and the [arXiv:2202.01809] stellar population synthesis (SPS) model. To make this computationally feasible, PROVABGS will use a neural emulator for the SPS model to accelerate the posterior inference. In this work, we present how we construct the emulator using the [arXiv:1911.11778] approach and verify that it can be used to accurately infer galaxy properties. We confirm that the emulator is in excellent agreement with the original SPS model with $\ll 1\%$ error and is $100\times$ faster. In addition, we demonstrate that the posteriors of galaxy properties derived using the emulator are also in excellent agreement with those inferred using the original model. The neural emulator presented in this work is essential in bypassing the computational challenge posed in constructing the PROVABGS catalog. Furthermore, it demonstrates the advantages of emulation for scaling sophisticated analyses to millions of galaxies.

astro-ph.GA↗

arXiv2022

Hierarchical Bayesian inference of photometric redshifts with stellar population synthesis models

We present a Bayesian hierarchical framework to analyze photometric galaxy survey data with stellar population synthesis (SPS) models. Our method couples robust modeling of spectral energy distributions with a population model and a noise model to characterize the statistical properties of the galaxy populations and real observations, respectively. By self-consistently inferring all model parameters, from high-level hyper-parameters to SPS parameters of individual galaxies, one can separate sources of bias and uncertainty in the data.We demonstrate the strengths and flexibility of this approach by deriving accurate photometric redshifts for a sample of spectroscopically-confirmed galaxies in the COSMOS field, all with 26-band photometry and spectroscopic redshifts. We achieve a performance competitive with publicly-released photometric redshift catalogs based on the same data. Prior to this work, this approach was computationally intractable in practice due to the heavy computational load of SPS model calls; we overcome this challenge using with neural emulators. We find that the largest photometric residuals are associated with poor calibration for emission line luminosities and thus build a framework to mitigate these effects. This combination of physics-based modeling accelerated with machine learning paves the path towards meeting the stringent requirements on the accuracy of photometric redshift estimation imposed by upcoming cosmological surveys. The approach also has the potential to create new links between cosmology and galaxy evolution through the analysis of photometric datasets.

astro-ph.IM↗

arXiv2022

Measuring the thermal and ionization state of the low-$z$ IGM using likelihood free inference

We present a new approach to measure the power-law temperature density relationship $T=T_0 (\rho / \bar{\rho})^{\gamma -1}$ and the UV background photoionization rate $\Gamma_{\rm HI}$ of the IGM based on the Voigt profile decomposition of the Ly$\alpha$ forest into a set of discrete absorption lines with Doppler parameter $b$ and the neutral hydrogen column density $N_{\rm HI}$. Previous work demonstrated that the shape of the $b$-$N_{\rm HI}$ distribution is sensitive to the IGM thermal parameters $T_0$ and $\gamma$, whereas our new inference algorithm also takes into account the normalization of the distribution, i.e. the line-density d$N$/d$z$, and we demonstrate that precise constraints can also be obtained on $\Gamma_{\rm HI}$. We use density-estimation likelihood-free inference (DELFI) to emulate the dependence of the $b$-$N_{\rm HI}$ distribution on IGM parameters trained on an ensemble of 624 Nyx hydrodynamical simulations at $z = 0.1$, which we combine with a Gaussian process emulator of the normalization. To demonstrate the efficacy of this approach, we generate hundreds of realizations of realistic mock HST/COS datasets, each comprising 34 quasar sightlines, and forward model the noise and resolution to match the real data. We use this large ensemble of mocks to extensively test our inference and empirically demonstrate that our posterior distributions are robust. Our analysis shows that by applying our new approach to existing Ly$\alpha$ forest spectra at $z\simeq 0.1$, one can measure the thermal and ionization state of the IGM with very high precision ($\sigma_{\log T_0} \sim 0.08$ dex, $\sigma_\gamma \sim 0.06$, and $\sigma_{\log \Gamma_{\rm HI}} \sim 0.07$ dex).

astro-ph.CO↗

arXiv2022

Forward modeling of galaxy populations for cosmological redshift distribution inference

We present a forward modeling framework for estimating galaxy redshift distributions from photometric surveys. Our forward model is composed of: a detailed population model describing the intrinsic distribution of physical characteristics of galaxies, encoding galaxy evolution physics; a stellar population synthesis model connecting the physical properties of galaxies to their photometry; a data-model characterizing the observation and calibration processes for a given survey; and, explicit treatment of selection cuts, both into the main analysis sample and subsequent sorting into tomographic redshift bins. This approach has the appeal that it does not rely on spectroscopic calibration data, provides explicit control over modeling assumptions, and builds a direct bridge between photo-$z$ inference and galaxy evolution physics. In addition to redshift distributions, forward modeling provides a framework for drawing robust inferences about the statistical properties of the galaxy population more generally. We demonstrate the utility of forward modeling by estimating the redshift distributions for the Galaxy And Mass Assembly (GAMA) and Vimos VLT Deep (VVDS) surveys, validating against their spectroscopic redshifts. Our baseline model is able to predict tomographic redshift distributions for GAMA and VVDS with a bias of $\Delta z \lesssim 0.003$ and $\Delta z \simeq 0.01$ on the mean redshift respectively -- comfortably accurate enough for Stage III cosmological surveys -- without any hyper-parameter tuning (i.e., prior to doing any fitting to those data). We anticipate that with additional hyper-parameter fitting and modeling improvements, forward modeling can provide a path to accurate redshift distribution inference for Stage IV surveys.

astro-ph.CO↗

arXiv2022

Marginal Post Processing of Bayesian Inference Products with Normalizing Flows and Kernel Density Estimators

Bayesian analysis has become an indispensable tool across many different cosmological fields including the study of gravitational waves, the Cosmic Microwave Background and the 21-cm signal from the Cosmic Dawn among other phenomena. The method provides a way to fit complex models to data describing key cosmological and astrophysical signals and a whole host of contaminating signals and instrumental effects modelled with `nuisance parameters'. In this paper, we summarise a method that uses Masked Autoregressive Flows and Kernel Density Estimators to learn marginal posterior densities corresponding to core science parameters. We find that the marginal or 'nuisance-free' posteriors and the associated likelihoods have an abundance of applications including; the calculation of previously intractable marginal Kullback-Leibler divergences and marginal Bayesian Model Dimensionalities, likelihood emulation and prior emulation. We demonstrate each application using toy examples, examples from the field of 21-cm cosmology and samples from the Dark Energy Survey. We discuss how marginal summary statistics like the Kullback-Leibler divergences and Bayesian Model Dimensionalities can be used to examine the constraining power of different experiments and how we can perform efficient joint analysis by taking advantage of marginal prior and likelihood emulators. We package our multipurpose code up in the pip-installable code margarine for use in the wider scientific community.

astro-ph.IM↗

arXiv2021

Lossless, Scalable Implicit Likelihood Inference for Cosmological Fields

We present a comparison of simulation-based inference to full, field-based analytical inference in cosmological data analysis. To do so, we explore parameter inference for two cases where the information content is calculable analytically: Gaussian random fields whose covariance depends on parameters through the power spectrum; and correlated lognormal fields with cosmological power spectra. We compare two inference techniques: i) explicit field-level inference using the known likelihood and ii) implicit likelihood inference with maximally informative summary statistics compressed via Information Maximising Neural Networks (IMNNs). We find that a) summaries obtained from convolutional neural network compression do not lose information and therefore saturate the known field information content, both for the Gaussian covariance and the lognormal cases, b) simulation-based inference using these maximally informative nonlinear summaries recovers nearly losslessly the exact posteriors of field-level inference, bypassing the need to evaluate expensive likelihoods or invert covariance matrices, and c) even for this simple example, implicit, simulation-based likelihood incurs a much smaller computational cost than inference with an explicit likelihood. This work uses a new IMNNs implementation in $\texttt{Jax}$ that can take advantage of fully-differentiable simulation and inference pipeline. We also demonstrate that a single retraining of the IMNN summaries effectively achieves the theoretically maximal information, enhancing the robustness to the choice of fiducial model where the IMNN is trained.

astro-ph.CO↗

arXiv2021

Non-parametric spatial curvature inference using late-universe cosmological probes

Inferring high-fidelity constraints on the spatial curvature parameter, $\Omega_{\rm K}$, under as few assumptions as possible, is of fundamental importance in cosmology. We propose a method to non-parametrically infer $\Omega_{\rm K}$ from late-Universe probes alone. Using Gaussian Processes (GP) to reconstruct the expansion history, we combine Cosmic Chronometers (CC) and Type Ia Supernovae (SNe~Ia) data to infer constraints on curvature, marginalized over the expansion history, calibration of the CC and SNe~Ia data, and the GP hyper-parameters. The obtained constraints on $\Omega_{\rm K}$ are free from parametric model assumptions for the expansion history, and are insensitive to the overall calibration of both the CC and SNe~Ia data (being sensitive only to relative distances and expansion rates). Applying this method to \textit{Pantheon} SNe~Ia and the latest compilation of CCs, we find $\Omega_{\rm K} = -0.03 \pm 0.26$, consistent with spatial flatness at the $\mathcal{O}(10^{-1})$ level, and independent of any early-Universe probes. Applying our methodology to future Baryon Acoustic Oscillations and SNe~Ia data from upcoming Stage IV surveys, we forecast the ability to constrain $\Omega_{\rm K}$ at the $\mathcal{O}(10^{-2})$ level.

astro-ph.CO↗