Pith. Sign in

REVIEW

2Flow2SN: Dusty Supernova Cosmology with Photometric Redshifts and Contamination using Neural Simulation-Based Inference

Cosmology and Nongalactic Astrophysics · Submitted 21 Oct 2026 · 239 references

Pith's verdict: A well-constructed modular SBI framework for photometric SN cosmology, but the central claim about amortisation over the Hubble diagram is not tested for the most relevant parameter—M_int0—which could break the second flow.

T0 · AI review

How this review was made: An AI referee (deepseek/deepseek-v4-flash) read the paper on October 23, 2026 and answered a fixed list of questions about it. T0 is the starting level every paper gets. The levels describe how deeply Pith has checked the paper, never the authors.

  1. T0 AI review An AI read the paper and answered a fixed list of questions.
  2. T1 Audit Automatic checks that the references and files the paper relies on exist and match.
  3. T2 Compute The paper's code or calculations re-run with the same results.
  4. T3 Formal statement The claims written out in a formal math language a computer can read.
  5. T4 Proof Those claims proved, and the proof checked by computer.

How Pith checks papers →

· 3 major objections · 6 minor · Ask about this paper · The compiled record · Read the full review →
Challenge this review Comment (0) Read on arXiv ↗
Cite ▾
arXiv ID2610.30244 (version 1)
Pith IDORQKXNPE, Pith's permanent record of this paper and its review
CitationBenjamin M. Boyd, Kaisey S. Mandel, Lisa Kelsey, Madeleine Ginolin, Matthew Grayling, Maximilian Autenrieth, Stephen Thorp. "2Flow2SN: Dusty Supernova Cosmology with Photometric Redshifts and Contamination using Neural Simulation-Based Inference." arXiv:2610.30244 (2026). https://pith.science/paper/ORQKXNPE
Cite this reviewBibTeX for Pith's review of this paper
Claim paper

Why Pith says this

Highlight any text to comment on it.

Referee report
The paper

The paper claims

FlowSN2 recovers cosmology from photometric SNe under selection and contamination.

The reading

This paper presents FlowSN2, a modular statistical framework that combines hierarchical Bayesian modelling with simulation-based inference to perform cosmological parameter estimation from photometric Type Ia supernova samples. The key innovation is reframing the problem from a regression of distances given spectroscopic redshifts to a joint inference over light curve summaries and photometric redshifts, treating both as data with uncertainties. FlowSN2 uses normalising flows to approximate intractable likelihoods for observed data and latent distances, allowing amortised inference across different cosmologies without retraining. The framework also incorporates a dust model that separates intrinsic SN variations from extrinsic dust effects, and handles non-Ia contamination via a principled Bayesian resampling approach that does not require an explicit contaminant likelihood. In simulated tests, FlowSN2 recovers true cosmological and population parameters, with posteriors matching analytical solutions, and scales to over a hundred thousand supernovae without retraining.

What carries the argument

The two normalising flows: a first flow $q_{\phi_1}$ approximates the intractable likelihood of observed light curve summaries and photometric redshifts given latent distance modulus and redshift, while a second flow $q_{\phi_2}$ learns the latent distance modulus-redshift distribution under selection effects. The second flow is trained with a Gaussian scatter on distance moduli ($\sigma_\mu = 0.25$) to provide support over alternative cosmologies. The flows are trained once on simulations under a fiducial cosmology and can be reused at inference for any cosmology, achieving amortisation. Contamination is handled by modular Bayesian resampling which resamples class labels according to classification scores and averages the pure-Ia posteriors, avoiding an explicit contaminant likelihood.

Extended reading notes

Core claim

FlowSN2 demonstrates that by using two normalising flows—one to learn the distribution of observed light curve summaries and redshifts conditioned on latent distance modulus and redshift, and another to learn the latent distance-redshift distribution under selection—one can achieve amortised inference over the Hubble diagram, enabling parameter estimation for alternative cosmologies (e.g., $w_0 w_a \mathrm{CDM}$) without retraining. The method also successfully propagates photometric classification uncertainty through a modular Bayesian resampling approach that averages posteriors over resampled class allocations. The paper shows that the resulting posteriors are in close agreement with tractable analytical likelihood solutions and pure-Ia baselines, even in the presence of selection effects, photometric redshift uncertainties, and up to 18% non-Ia contamination.

Load-bearing premise

The Gaussian distance modulus smearing ($\sigma_\mu = 0.25$) applied during training is sufficient to provide support over alternative cosmologies at inference time; if the true cosmology or population parameters lie far outside this smeared training distribution, the normalising flow evaluations may become unreliable.

Editorial extensions

If this is right

  • Photometric SN cosmology can be performed without spectroscopic redshifts, reducing reliance on expensive spectroscopy.
  • Bayesian model comparison for nested models (e.g., intrinsic mass step vs. dust, evolving dark energy vs. constant) can be done at negligible extra cost using Savage-Dickey ratios.
  • The framework can scale to LSST and Roman sample sizes (~100,000 SNe) without retraining, enabling robust cosmological constraints.
  • Selection effects are naturally accounted for through the simulation-based training, removing the need for parametric selection models or binning.
  • The modular approach to contamination protects inference from misspecified contaminant models while still using all available data.

Reading between the lines

Editorial extensions of the paper, not claims the author makes directly.

  • The success on simplified simulations with Gaussian photometric redshift uncertainties suggests that extending to non-Gaussian or multimodal redshift estimates (which the authors note is straightforward) could provide a path to handling the most realistic photometric redshift distributions.
  • Since the normalising flows are trained on pure-Ia simulations and contamination is handled by resampling, the method implicitly assumes that the classification scores are well-calibrated; in practice, calibrating photometric classifiers for high-redshift samples remains an open challenge.
  • The ability to perform model comparison without retraining could be applied to other nested debates in SN cosmology, such as the presence of a colour-luminosity slope evolution with redshift.
  • The reliance on a Gaussian smearing ($\sigma_\mu=0.25$) to cover alternative cosmologies may limit support for cosmologies far from the fiducial training point; the authors note this is sufficient for typical prior ranges but it could be tested with broader smearing.
Original abstract

Photometric Type Ia supernova (SN Ia) cosmology in the era of the Vera C. Rubin Observatory's Legacy Survey of Space and Time (LSST) demands statistical frameworks that simultaneously account for selection effects, photometric redshift uncertainties and non-Ia contamination at scale. We present FlowSN2, a modular statistical framework combining hierarchical Bayesian modelling and simulation-based inference (SBI), that reframes photometric SN cosmology from a regression task to a joint inference problem over light curve summaries and photometric redshifts. By employing neural likelihood estimation via normalising flows, FlowSN2 approximates the data distributions of selected objects in a form that can be reused across different cosmological models, achieving amortisation over the Hubble diagram. This amortisation enables inference across alternative cosmological parameterisations, such as $w_0w_a\mathrm{CDM}$, away from the fiducial training cosmology. The framework integrates a standardisation model that disentangles intrinsic SN variations from extrinsic host-galaxy dust effects and adopts a principled Bayesian resampling approach to propagate SN classification uncertainty, without requiring an explicit contaminant likelihood. Across simulated tests spanning five cosmological parameter sets, our framework recovers true cosmological and population parameters, with posteriors in close agreement with both tractable analytical solutions and pure-Ia baselines. Conveniently, the framework enables Bayesian model comparison at negligible additional computational cost, providing a valuable tool for addressing debates surrounding SN standardisation and evolving dark energy. Capable of scaling to over a hundred thousand SNe without retraining, FlowSN2 offers an interpretable and statistically principled solution for cosmological inference with upcoming wide-field photometric surveys.

The compiled record

The paper's own sentences, arranged, kept beside the review and separate from it. Each label is drawn from a closed vocabulary; the number beside it is the probability the classifier gave that label. Numbers in the tables are the authors', checked by code. No sentence in this record was written by a model.

Compiled by classifier-1.13.0 via TypeSafe API, question bank abstract-p1-20260923, 2026-10-23: 1 request, 7,265 input tokens.

What the abstract itself states, as probabilities: claims first 0.05 · states limitation 0.14 · human participants 0.01 · controlled comparison 0.73 · randomized assignment 0.02 · states quantitative result 0.75 · compares with prior results 0.89 · mentions code or data availability 0.04.

The abstract, sentence by sentence

arXiv listing abstract v1. Results and the main claim are underlined; .

Photometric Type Ia supernova (SN Ia) cosmology in the era of the Vera C. Rubin Observatory's Legacy Survey of Space and Time (LSST) demands statistical frameworks that simultaneously account for selection effects, photometric redshift uncertainties and non-Ia contamination at scale. We present FlowSN2, a modular statistical framework combining hierarchical Bayesian modelling and simulation-based inference (SBI), that reframes photometric SN cosmology from a regression task to a joint inference problem over light curve summaries and photometric redshifts. By employing neural likelihood estimation via normalising flows, FlowSN2 approximates the data distributions of selected objects in a form that can be reused across different cosmological models, achieving amortisation over the Hubble diagram. This amortisation enables inference across alternative cosmological parameterisations, such as $w_0w_a\mathrm{CDM}$, away from the fiducial training cosmology. The framework integrates a standardisation model that disentangles intrinsic SN variations from extrinsic host-galaxy dust effects and adopts a principled Bayesian resampling approach to propagate SN classification uncertainty, without requiring an explicit contaminant likelihood. Across simulated tests spanning five cosmological parameter sets, our framework recovers true cosmological and population parameters, with posteriors in close agreement with both tractable analytical solutions and pure-Ia baselines. Conveniently, the framework enables Bayesian model comparison at negligible additional computational cost, providing a valuable tool for addressing debates surrounding SN standardisation and evolving dark energy. Capable of scaling to over a hundred thousand SNe without retraining, FlowSN2 offers an interpretable and statistically principled solution for cosmological inference with upcoming wide-field photometric surveys.

background contribution method result interpretation main claim

What the classifier reports for each sentence. The number is the probability it gave the label.
#sentencerolethe authors' own workstates the main conclusionstrength
s0 Photometric Type Ia supernova (SN Ia) cosmology in the era of the Vera C. background 1.00 0.04 0.02 unclear 0.98
s1 Rubin Observatory's Legacy Survey of Space and Time (LSST) demands statistical frameworks that simultaneously account for selection effects, photometric redshift uncertainties and non-Ia contamination at scale. background 0.94 0.06 0.03 asserted 1.00
s2 We present FlowSN2, a modular statistical framework combining hierarchical Bayesian modelling and simulation-based inference (SBI), that reframes photometric SN cosmology from a regression task to a joint inference problem over light curve summaries and photometric redshifts. contribution 0.96 0.93 0.23 asserted 1.00
s3 By employing neural likelihood estimation via normalising flows, FlowSN2 approximates the data distributions of selected objects in a form that can be reused across different cosmological models, achieving amortisation over the Hubble diagram. method 0.98 0.90 0.10 asserted 1.00
s4 This amortisation enables inference across alternative cosmological parameterisations, such as $w_0w_a\mathrm{CDM}$, away from the fiducial training cosmology. method 0.70 0.73 0.10 asserted 0.97
s5 The framework integrates a standardisation model that disentangles intrinsic SN variations from extrinsic host-galaxy dust effects and adopts a principled Bayesian resampling approach to propagate SN classification uncertainty, without requiring an explicit contaminant likelihood. method 0.96 0.86 0.13 asserted 1.00
s6 Across simulated tests spanning five cosmological parameter sets, our framework recovers true cosmological and population parameters, with posteriors in close agreement with both tractable analytical solutions and pure-Ia baselines. result 1.00 0.96 0.42 asserted 1.00
s7 Conveniently, the framework enables Bayesian model comparison at negligible additional computational cost, providing a valuable tool for addressing debates surrounding SN standardisation and evolving dark energy. interpretation 0.53 0.75 0.15 asserted 0.97
s8 Capable of scaling to over a hundred thousand SNe without retraining, FlowSN2 offers an interpretable and statistically principled solution for cosmological inference with upcoming wide-field photometric surveys. contribution 0.51 0.76 0.34 asserted 0.93

What kind of paper the classifier reports this is: method or tool 1.00 · evidence from simulation or computation 1.00.

What was examined, and what was not

sources read
  • arXiv listing abstract v1
not examined
  • full text
  • figures and tables
  • references and citing sentences
  • relations to other papers
levels run
  • P abstract
not run
  • S sections
  • T body sentences
  • C citations
  • R cross-paper relations

Every quoted sentence on this record is the authors'. Every label is from Pith's closed vocabulary, with the probability shown beside it. Every number in the measurements table is the authors', checked by code. Record as JSON

Figures
Figure 1. Illustration of our simulated surveys used in experiments. Two inde￾pendent surveys are simulated. We first include a low redshift spectroscopic survey and then a high redshift photometric survey, with larger redshift uncer￾tainties and non-Ia contaminants. The top and right panels are the normalised marginal distributions. Both surveys have their own selection function in ob￾served redshift and apparent magnitude space, shown in the marginals with dashed lines. a given survey’s observational footprint or because the recorded data fails to satisfy the quality cuts. Hereafter, for…
Figure 2. The posterior of the global parameters of interest 𝚯 = (𝚯SN, 𝜉rate, 𝑪) conditioned on all the data, framed as a joint inference problem, is 𝑃(𝚯| 𝑫ˆ obs, 𝒁ˆ obs, 𝑰obs) ∝ 𝑃(𝑫ˆ obs, 𝒁ˆ obs| 𝑰obs, 𝚯)𝑃(𝚯| 𝑰obs), (21) where 𝑫ˆ obs = {𝒅ˆ 𝑠 } 𝑁obs SN 𝑠=1 denotes the observed summary statistics for the selected sample and 𝒁ˆ obs = {𝑧ˆ𝑠 } 𝑁obs SN 𝑠=1 denotes the redshift mea￾surements associated with the selected SNe. The selected sample is represented by the indicator vector of ones 𝑰obs = {𝐼𝑠 = 1} 𝑁obs SN 𝑠=1 . The combined data likelihood across all the observed SNe is denoted by 𝑃(𝑫ˆ obs,…
Figure 3. Illustration of the FlowSN2 pipeline. The first normalising flow 𝑞 𝝓1 1 learns the observed light curve summary statistics (𝑚ˆ 𝑠 , 𝑐ˆ𝑠 , 𝑥ˆ𝑠 ) and observed redshifts 𝑧ˆ𝑠 distributions under the influence of selection effects, conditioned on the latent redshift 𝑧𝑠 and distance modulus 𝜇𝑠. The second normalising flow 𝑞 𝝓2 2 learns the latent redshift and distance modulus distribution under the influence of selection. The latent redshift distribution under the influence of selection decreases at higher redshifts as Malmquist bias takes effect. The learnt 𝑞 𝝓2 2 represents a noisy…
Figure 4. Samples from our trained four-dimensional normalising flow 𝑞 𝝓1 1 corresponding to the photometric redshift survey. We condition on the high￾est redshift SNe to highlight the maximum impact of selection effects. The distribution of unseen validation data demonstrates that the normalising flow has captured the influence of selection, relative to the full population shown in grey. 3.1.3 SN Ia Only Results Prior to inference, we assess the quality of our learnt 𝑞 𝝓1 1 likeli￾hood approximation by comparing its samples against held-out data distributions as demonstrated in
Figure 5. FlowSN2 posteriors on all our cosmological and SN population parameters. These inferences are using a pure-Ia sample with photometric redshifts under the influence of selection. We also show posteriors that use the analytical likelihood solution (defined in Appendix A) for the simplified model and the naive likelihood that does not take selection effects into account. This simulation is from cosmology 𝑪2 showing FlowSN2 is still effective when inferring cosmological parameters away from the fiducial ΛCDM cosmology 𝑪fid = 𝑪1 used in training the normalising flows. The true value for…
Figure 6. Mock classification scores for SNe Ia and contaminants in the high￾redshift photometric sample used in our experiments. These are obtained using the known conditional class probabilities from the simulations. Further experimental details are in Section 3.2.1. rameters, while 𝑪5 are closer to the parameters inferred in the recent DESI results (Adame et al. 2025). For additional constraining power we perform these inferences with a CMB and BAO prior as described in Appendix D. We find that again the FlowSN2 is able to infer the true cosmology under selection effects, with biases less…
Figure 7. FlowSN2 𝑤0𝑤𝑎CDM posteriors under the influence of non-Ia contamination. We perform inference with all objects including contaminants to show how they can bias each parameter. We then include inferences using our default approach of modular Bayesian resampling (𝑀 = 5) of the data to account for contamination as described in Section 2.5. For comparison we include inferences on the pure SN Ia subsample using the true labels. Our experiments include a ∼18% post-selection contamination rate. Experimental details can be found in Section 3.2.1. to demonstrate how such model comparisons are…
Figure 8. Plot demonstrating the scalability of the FlowSN2 framework inferring cosmological parameters and key SN Ia parameters jointly on increasing sample sizes. These inferences were under the influence of selection effects, with 18% non-Ia contamination rate and with photometric redshifts. The straight lines represent the true parameters in the simulations. parameter at the mass threshold Mstep = 1010 𝑀⊙, following the qualitative picture of Brout & Scolnic (2021). Objects in high-mass hosts are simulated to have a larger 𝜏 (0.098 against 0.068) and a lower 𝑅𝐵 (3.0 against 3.7). The…

Figures reproduced from arXiv:2610.30244 by the authors.

References

Discussion

Disagree with the review, or know something it missed? Leave a comment, highlight any passage above to comment on it, or challenge this review formally.