A Probabilistic Autoencoder for Galaxy SED Reconstruction and Redshift Estimation: Application to Mock SPHEREx Spectrophotometry

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Feder, Richard M., Parker, Liam, Seljak, Uroš
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910073407668224
author Feder, Richard M.
Parker, Liam
Seljak, Uroš
author_facet Feder, Richard M.
Parker, Liam
Seljak, Uroš
contents We present a probabilistic autoencoder (PAE) framework for galaxy spectral energy distribution (SED) modeling and redshift estimation, applied to synthetic SPHEREx 102-band spectrophotometry. Our PAE learns a compact latent representation of rest-frame galaxy SEDs transformed to a simple Gaussian base density using a normalizing flow, combined with an explicit forward model enabling joint Bayesian inference over intrinsic SED parameters and redshift with well-defined priors. In controlled tests on simulated SPHEREx spectra, our PAE improves on template fitting (TF) in source recovery, outlier rate, and posterior calibration, with trade-offs in redshift performance that depend on the assumed priors. A simple cut on the ratio of PAE and TF uncertainties identifies sources that are overwhelmingly TF outliers, which can be used to clean existing TF samples while retaining the vast majority of well-recovered sources. By directly profiling over PAE latent variables, we show these cases correspond to shallow likelihood surfaces where the PAE's continuous SED manifold produces broader likelihoods that more faithfully reflect the lack of constraining power in the data, whereas the TF discrete model grid yields artificially confident but incorrect redshift estimates. Lastly, we present an alternative, simulation-based inference approach using a Transformer encoder and conditional normalizing flow, which provides similar redshift performance to the PAE but with $\sim200\times$ faster inference throughput. Our implementation, \texttt{PAESpec}, is publicly available and provides a foundation for principled redshift estimation in modern photometric surveys.
format Preprint
id arxiv_https___arxiv_org_abs_2603_24668
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle A Probabilistic Autoencoder for Galaxy SED Reconstruction and Redshift Estimation: Application to Mock SPHEREx Spectrophotometry
Feder, Richard M.
Parker, Liam
Seljak, Uroš
Instrumentation and Methods for Astrophysics
Cosmology and Nongalactic Astrophysics
We present a probabilistic autoencoder (PAE) framework for galaxy spectral energy distribution (SED) modeling and redshift estimation, applied to synthetic SPHEREx 102-band spectrophotometry. Our PAE learns a compact latent representation of rest-frame galaxy SEDs transformed to a simple Gaussian base density using a normalizing flow, combined with an explicit forward model enabling joint Bayesian inference over intrinsic SED parameters and redshift with well-defined priors. In controlled tests on simulated SPHEREx spectra, our PAE improves on template fitting (TF) in source recovery, outlier rate, and posterior calibration, with trade-offs in redshift performance that depend on the assumed priors. A simple cut on the ratio of PAE and TF uncertainties identifies sources that are overwhelmingly TF outliers, which can be used to clean existing TF samples while retaining the vast majority of well-recovered sources. By directly profiling over PAE latent variables, we show these cases correspond to shallow likelihood surfaces where the PAE's continuous SED manifold produces broader likelihoods that more faithfully reflect the lack of constraining power in the data, whereas the TF discrete model grid yields artificially confident but incorrect redshift estimates. Lastly, we present an alternative, simulation-based inference approach using a Transformer encoder and conditional normalizing flow, which provides similar redshift performance to the PAE but with $\sim200\times$ faster inference throughput. Our implementation, \texttt{PAESpec}, is publicly available and provides a foundation for principled redshift estimation in modern photometric surveys.
title A Probabilistic Autoencoder for Galaxy SED Reconstruction and Redshift Estimation: Application to Mock SPHEREx Spectrophotometry
topic Instrumentation and Methods for Astrophysics
Cosmology and Nongalactic Astrophysics
url https://arxiv.org/abs/2603.24668