RUBIX: Differentiable forward modelling of galaxy spectral data cubes for gradient-based parameter estimation

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Schaible, Anna Lena, Çakır, Ufuk, Buck, Tobias, Mack, Harald, Obreja, Aura, Oguz, Nihat, Oliver, William H., Cărămizaru, Horea-Alexandru
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866918213212700672
author Schaible, Anna Lena
Çakır, Ufuk
Buck, Tobias
Mack, Harald
Obreja, Aura
Oguz, Nihat
Oliver, William H.
Cărămizaru, Horea-Alexandru
author_facet Schaible, Anna Lena
Çakır, Ufuk
Buck, Tobias
Mack, Harald
Obreja, Aura
Oguz, Nihat
Oliver, William H.
Cărămizaru, Horea-Alexandru
contents Although integral-field spectroscopy enables spatially resolved spectral studies of galaxies, bridging particle-based simulations to observations remains slow and non-differentiable. We present RUBIX, a JAX-based pipeline that models mock integral-field unit (IFU) cubes for galaxies end-to-end and calculates gradients with respect to particle inputs. Our implementation is purely functional, sharded, and differentiable throughout. We validate the gradients against central finite differences and demonstrate gradient-based parameter estimation on controlled setups. While current experiments are limited to basic test cases, they demonstrate the feasibility of differentiable forward modelling of IFU data. This paves the way for future work scaling up to realistic galaxy cubes and enabling machine learning workflows for IFU-based inference. The source code for the RUBIX software is publicly available under https://github.com/AstroAI-Lab/rubix.
format Preprint
id arxiv_https___arxiv_org_abs_2511_17110
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle RUBIX: Differentiable forward modelling of galaxy spectral data cubes for gradient-based parameter estimation
Schaible, Anna Lena
Çakır, Ufuk
Buck, Tobias
Mack, Harald
Obreja, Aura
Oguz, Nihat
Oliver, William H.
Cărămizaru, Horea-Alexandru
Astrophysics of Galaxies
Although integral-field spectroscopy enables spatially resolved spectral studies of galaxies, bridging particle-based simulations to observations remains slow and non-differentiable. We present RUBIX, a JAX-based pipeline that models mock integral-field unit (IFU) cubes for galaxies end-to-end and calculates gradients with respect to particle inputs. Our implementation is purely functional, sharded, and differentiable throughout. We validate the gradients against central finite differences and demonstrate gradient-based parameter estimation on controlled setups. While current experiments are limited to basic test cases, they demonstrate the feasibility of differentiable forward modelling of IFU data. This paves the way for future work scaling up to realistic galaxy cubes and enabling machine learning workflows for IFU-based inference. The source code for the RUBIX software is publicly available under https://github.com/AstroAI-Lab/rubix.
title RUBIX: Differentiable forward modelling of galaxy spectral data cubes for gradient-based parameter estimation
topic Astrophysics of Galaxies
url https://arxiv.org/abs/2511.17110