papers

Publications (20)

cs.LG2025

Solving Bayesian inverse problems with diffusion priors and off-policy RL

Luca Scimeca, Siddarth Venkatraman, Moksh Jain +14

This paper presents a practical application of Relative Trajectory Balance (RTB), a recently introduced off-policy reinforcement learning (RL) objective that can asymptotically sol…

astro-ph.IM2025

Pixellated Posterior Sampling of Point Spread Functions in Astronomical Images

Connor Stone, Ronan Legin, Alexandre Adam +4

We introduce a novel framework for upsampled Point Spread Function (PSF) modeling using pixel-level Bayesian inference. Accurate PSF characterization is critical for precision meas…

astro-ph.CO2024

Inpainting Galaxy Counts onto N-Body Simulations over Multiple Cosmologies and Astrophysics

Antoine Bourdin, Ronan Legin, Matthew Ho +3

Cosmological hydrodynamical simulations, while the current state-of-the art methodology for generating theoretical predictions for the large scale structures of the Universe, are a…

cs.LG2025

Amortizing intractable inference in diffusion models for vision, language, and control

Siddarth Venkatraman, Moksh Jain, Luca Scimeca +12

Diffusion models have emerged as effective distribution estimators in vision, language, and reinforcement learning, but their use as priors in downstream tasks poses an intractable…

astro-ph.IM2023

Echoes in the Noise: Posterior Samples of Faint Galaxy Surface Brightness Profiles with Score-Based Likelihoods and Priors

Alexandre Adam, Connor Stone, Connor Bottrell +3

Examining the detailed structure of galaxy populations provides valuable insights into their formation and evolution mechanisms. Significant barriers to such analysis are the non-t…

cs.LG2025

Improved off-policy training of diffusion samplers

Marcin Sendera, Minsu Kim, Sarthak Mittal +6

We study the problem of training diffusion models to sample from a distribution with a given unnormalized density or energy function. We benchmark several diffusion-structured infe…

astro-ph.GA2023

Unraveling the Mysteries of Galaxy Clusters: Recurrent Inference Deconvolution of X-ray Spectra

Carter Rhea, Julie Hlavacek-Larrondo, Ralph Kraft +3

In the realm of X-ray spectral analysis, the true nature of spectra has remained elusive, as observed spectra have long been the outcome of convolution between instrumental respons…

astro-ph.IM2023

Pixelated Reconstruction of Foreground Density and Background Surface Brightness in Gravitational Lensing Systems using Recurrent Inference Machines

Alexandre Adam, Laurence Perreault-Levasseur, Yashar Hezaveh +1

Modeling strong gravitational lenses in order to quantify the distortions in the images of background sources and to reconstruct the mass density in the foreground lenses has been…

astro-ph.IM2026

Probabilistic Interpolation of Sagittarius A*'s Multi-Wavelength Light Curves Using Diffusion Models

Gabriel Sasseville, Julie Hlavacek-Larrondo, Daryl Haggard +3

Understanding the variability of Sagittarius A* (Sgr A*) requires coordinated, multi-wavelength observations that span the electromagnetic spectrum. In this work, we focus on data…

astro-ph.IM2022

Posterior samples of source galaxies in strong gravitational lenses with score-based priors

Alexandre Adam, Adam Coogan, Nikolay Malkin +4

Inferring accurate posteriors for high-dimensional representations of the brightness of gravitationally-lensed sources is a major challenge, in part due to the difficulties of accu…

astro-ph.GA2023

The search for the lost attractor

Mario Pasquato, Syphax Haddad, Pierfrancesco Di Cintio +8

N-body systems characterized by inverse square attractive forces may display a self similar collapse known as the gravo-thermal catastrophe. In star clusters, collapse is halted by…

astro-ph.GA2024

Deconvolving X-ray Galaxy Cluster Spectra Using a Recurrent Inference Machine

Carter Rhea, Julie Hlavacek-Larrondo, Alexandre Adam +4

Recent advances in machine learning algorithms have unlocked new insights in observational astronomy by allowing astronomers to probe new frontiers. In this article, we present a m…

astro-ph.GA2025

The spatially-resolved effect of mergers on the stellar mass assembly of MaNGA galaxies

Eirini Angeloudi, Marc Huertas-Company, Jesús Falcón-Barroso +3

Understanding the origin of stars within a galaxy - whether formed in-situ or accreted from other galaxies (ex-situ) - is key to constraining its evolution. Spatially resolving the…

astro-ph.IM2023

Beyond Gaussian Noise: A Generalized Approach to Likelihood Analysis with non-Gaussian Noise

Ronan Legin, Alexandre Adam, Yashar Hezaveh +1

Likelihood analysis is typically limited to normally distributed noise due to the difficulty of determining the probability density function of complex, high-dimensional, non-Gauss…

astro-ph.IM2024

Caustics: A Python Package for Accelerated Strong Gravitational Lensing Simulations

Connor Stone, Alexandre Adam, Adam Coogan +9

Gravitational lensing is the deflection of light rays due to the gravity of intervening masses. This phenomenon is observed in a variety of scales and configurations, involving any…

astro-ph.IM2025

IRIS: A Bayesian Approach for Image Reconstruction in Radio Interferometry with expressive Score-Based priors

Noé Dia, M. J. Yantovski-Barth, Alexandre Adam +4

Inferring sky surface brightness distributions from noisy interferometric data in a principled statistical framework has been a key challenge in radio astronomy. In this work, we i…

astro-ph.IM2023

Bayesian Imaging for Radio Interferometry with Score-Based Priors

Noe Dia, M. J. Yantovski-Barth, Alexandre Adam +5

The inverse imaging task in radio interferometry is a key limiting factor to retrieving Bayesian uncertainties in radio astronomy in a computationally effective manner. We use a sc…

astro-ph.IM2022

Pixelated Reconstruction of Gravitational Lenses using Recurrent Inference Machines

Alexandre Adam, Laurence Perreault-Levasseur, Yashar Hezaveh

Modeling strong gravitational lenses in order to quantify the distortions in the images of background sources and to reconstruct the mass density in the foreground lenses has tradi…

astro-ph.IM2025

Tackling the Problem of Distributional Shifts: Correcting Misspecified, High-Dimensional Data-Driven Priors for Inverse Problems

Gabriel Missael Barco, Alexandre Adam, Connor Stone +2

Bayesian inference for inverse problems hinges critically on the choice of priors. In the absence of specific prior information, population-level distributions can serve as effecti…

astro-ph.GA2025

Mind the Information Gap: Unveiling Detailed Morphologies of z 0.5-1.0 Galaxies with SLACS Strong Lenses and Data-Driven Analysis

Ronan Legin, Connor Stone, Alexandre Adam +5

We present new state-of-the-art lens models for strong gravitational lensing systems from the Sloan Lens ACS (SLACS) survey, developed within a Bayesian framework that employs high…