arXiv subjects

Richard D. P. Grumitt

Publications and source records attributed to Richard D. P. Grumitt.

7 recordsLinked to original sources

arXiv2026

Interpreting map-based $E$/$B$ spectral properties of CMB foregrounds

Map-space $E$/$B$ decompositions of linear polarization are attractive for foreground and CMB analyses because they separate parity families: $B$-family patterns directly contaminate primordial tensor searches, while $E$-family patterns trace coherent Galactic structures. However, the $E$/$B$ transform is not fully local and can induce apparent spectral complexity even when the underlying sky is spectrally simple in $\underline{P}=Q+iU$. We quantify this effect for synchrotron emission using complex log--Taylor and moment expansions for $\underline{P}$, its spin-preserving projections $\underline{P}_E$ and $\underline{P}_B$, and its more standard scalar projections $E$ and $B$. We relate the coefficients of these expansions to physical mechanisms such as line-of-sight mixing, synchrotron ageing, and Faraday effects. Using simple sky models, we show how $\underline{P}_E$ and $\underline{P}_B$ reorganize the spectral behaviour of $\underline{P}$ into parity families with a clear geometric meaning. They retain interpretable amplitudes and angles and satisfy the closure relation $\underline{P}=\underline{P}_E+\underline{P}_B$, which extends to all moment orders. By contrast, scalar quantities such as $|E|$ and $|B|$ show larger induced variability of effective spectral parameters and enhanced spectral complexity, while $E+iB$ lacks interpretable polarization amplitude and angle. Finally, we present simple CMB-oriented applications: three-frequency diagnostics to test whether the sky is better described by a power law in $P$ or by separate effective laws in $(P_E,P_B)$, and idealized ILC and masking examples showing that the preferred map-space field depends on where foreground simplicity and residual contamination reside. This framework provides practical diagnostics for choosing foreground modelling, cleaning, and masking strategies in Galactic and CMB $B$-mode analyses.

astro-ph.CO↗

arXiv2025

Modeling Foreground Spatial Variations in 21 cm Gaussian Process Component Separation

Gaussian processes (GPs) have been extensively utilized as nonparametric models for component separation in 21 cm data analyses. This exploits the distinct spectral behavior of the cosmological and foreground signals, which are modeled through the GP covariance kernel. Previous approaches have employed a global GP kernel along all lines of sight (LoS). In this work, we study Bayesian approaches that allow for spatial variations in foreground kernel parameters, testing them against simulated HI intensity mapping observations. We consider a no-pooling (NP) model, which treats each LoS independently by fitting for separate covariance kernels, and a hierarchical Gaussian Process (HGP) model that allows for variation in kernel parameters between different LoS, regularized through a global hyperprior. We find that accounting for spatial variations in the GP kernel parameters results in a significant improvement in cosmological signal recovery, achieving up to a 30% reduction in the standard deviation of the residual distribution and improved model predictive performance. Allowing for spatial variations in GP kernel parameters also improves the recovery of the HI power spectra and wavelet scattering transform coefficients. Whilst the NP model achieves superior recovery as measured by the residual distribution, it demands extensive computational resources, faces formidable convergence challenges, and is prone to overfitting. Conversely, the HGP model strikes a balance between the accuracy and robustness of the signal recovery. Further improvements to the HGP model will require more physically motivated modeling of foreground spatial variations.

astro-ph.CO↗

arXiv2025

$\texttt{synax}$: A Differentiable and GPU-accelerated Synchrotron Simulation Package

We introduce synax, a novel library for automatically differentiable simulation of Galactic synchrotron emission. Built on the JAX framework, synax leverages JAX's capabilities, including batch acceleration, just-in-time compilation, and hardware-specific optimizations (CPU, GPU, TPU). Crucially, synax uses JAX's automatic differentiation (AD) mechanism, enabling precise computation of derivatives with respect to any model parameters. This feature facilitates powerful inference algorithms, such as Hamiltonian Monte Carlo (HMC) and gradient-based optimization, which enables inference over models that would otherwise be computationally prohibitive. In its initial release, synax supports synchrotron intensity and polarization calculations down to GHz frequencies, alongside several models of the Galactic magnetic field (GMF), cosmic ray (CR) spectra, and thermal electron density fields. We demonstrate the transformative potential of AD for tasks involving full posterior inference using gradient-based techniques or Maximum Likelihood Estimation (MLE) optimization. Notably, we show that GPU acceleration brings a twenty-fold enhancement in efficiency, while HMC achieves a two-fold improvement over standard random walk Metropolis-Hastings (RWMH) when performing inference over a four-parameter test model. HMC still works on a more complex, 16-parameter model while RWMH fails to converge. Additionally, we showcase the application of synax in optimizing the GMF based on the Haslam 408 MHz map, achieving residuals with a standard deviation below 1 K.

astro-ph.IM↗

arXiv2024

Sequential Kalman Tuning of the $t$-preconditioned Crank-Nicolson algorithm: efficient, adaptive and gradient-free inference for Bayesian inverse problems

Ensemble Kalman Inversion (EKI) has been proposed as an efficient method for the approximate solution of Bayesian inverse problems with expensive forward models. However, when applied to the Bayesian inverse problem EKI is only exact in the regime of Gaussian target measures and linear forward models. In this work we propose embedding EKI and Flow Annealed Kalman Inversion (FAKI), its normalizing flow (NF) preconditioned variant, within a Bayesian annealing scheme as part of an adaptive implementation of the $t$-preconditioned Crank-Nicolson (tpCN) sampler. The tpCN sampler differs from standard pCN in that its proposal is reversible with respect to the multivariate $t$-distribution. The more flexible tail behaviour allows for better adaptation to sampling from non-Gaussian targets. Within our Sequential Kalman Tuning (SKT) adaptation scheme, EKI is used to initialize and precondition the tpCN sampler for each annealed target. The subsequent tpCN iterations ensure particles are correctly distributed according to each annealed target, avoiding the accumulation of errors that would otherwise impact EKI. We demonstrate the performance of SKT for tpCN on three challenging numerical benchmarks, showing significant improvements in the rate of convergence compared to adaptation within standard SMC with importance weighted resampling at each temperature level, and compared to similar adaptive implementations of standard pCN. The SKT scheme applied to tpCN offers an efficient, practical solution for solving the Bayesian inverse problem when gradients of the forward model are not available. Code implementing the SKT schemes for tpCN is available at \url{https://github.com/RichardGrumitt/KalmanMC}.

stat.CO↗

arXiv2023

Flow Annealed Kalman Inversion for Gradient-Free Inference in Bayesian Inverse Problems

For many scientific inverse problems we are required to evaluate an expensive forward model. Moreover, the model is often given in such a form that it is unrealistic to access its gradients. In such a scenario, standard Markov Chain Monte Carlo algorithms quickly become impractical, requiring a large number of serial model evaluations to converge on the target distribution. In this paper we introduce Flow Annealed Kalman Inversion (FAKI). This is a generalization of Ensemble Kalman Inversion (EKI), where we embed the Kalman filter updates in a temperature annealing scheme, and use normalizing flows (NF) to map the intermediate measures corresponding to each temperature level to the standard Gaussian. In doing so, we relax the Gaussian ansatz for the intermediate measures used in standard EKI, allowing us to achieve higher fidelity approximations to non-Gaussian targets. We demonstrate the performance of FAKI on two numerical benchmarks, showing dramatic improvements over standard EKI in terms of accuracy whilst accelerating its already rapid convergence properties (typically in $\mathcal{O}(10)$ steps).

stat.CO↗

arXiv2022

Deterministic Langevin Monte Carlo with Normalizing Flows for Bayesian Inference

We propose a general purpose Bayesian inference algorithm for expensive likelihoods, replacing the stochastic term in the Langevin equation with a deterministic density gradient term. The particle density is evaluated from the current particle positions using a Normalizing Flow (NF), which is differentiable and has good generalization properties in high dimensions. We take advantage of NF preconditioning and NF based Metropolis-Hastings updates for a faster convergence. We show on various examples that the method is competitive against state of the art sampling methods.

stat.ML↗

arXiv2022

Model-independent constraints on $Ω_m$ and $H(z)$ from the link between geometry and growth

We constrain the expansion history of the Universe and the cosmological matter density fraction in a model-independent way by exclusively making use of the relationship between background and perturbations under a minimal set of assumptions. We do so by employing a Gaussian process to model the expansion history of the Universe from present time to the recombination era. The expansion history and the cosmological matter density are then constrained using recent measurements from cosmic chronometers, Type-Ia supernovae, baryon acoustic oscillations, and redshift-space distortion data. Our results show that the evolution in the reconstructed expansion history is compatible with the \textit{Planck} 2018 prediction at all redshifts. The current data considered in this study can constrain a Gaussian process on $H(z)$ to an average $9.4 \%$ precision across redshift. We find $Ω_m = 0.224 \pm 0.066$, lower but statistically compatible with the \textit{Planck} 2018 cosmology. Finally, the combination of future DESI measurements with the CMB measurement considered in this work holds the promise of $8 \%$ average constraints on a model-independent expansion history as well as a five-fold tighter $Ω_m$ constraint using the methodology developed in this work.

astro-ph.CO↗