Interpretable Neural Marked Statistics for Cosmological Inference

arXiv:2606.11295v1 Announce Type: cross Abstract: Recovering cosmological information beyond the power spectrum is a central goal for upcoming cosmological surveys, since late-time non-Gaussian signal in the matter density cannot be accessed through two-point statistics alone. Marked statistics fold part of this information back into the two-point level by reweighting the field with non-linear functions. We propose a neural marking scheme to generalize this process through a set of interpretable, physically motivated transformations that directly allow to interpret the gain in cosmological information at the morphological level. We employ a contrastive learning objective to align learnable marked summaries with the underlying cosmological parameters. At k_{max}=0.2,hMpc^{-1}, our neural mark tightens the marginalized constraint on sigma_8 by 2.9imes and on Omega_m by 1.8imes compared to classical marks, breaking the Omega_m-sigma_8 degeneracy at the Fisher information level. It further reduces the parameter MSE across our cosmological parameter prior by 1.45imes over the best classical mark. The learned latent geometry aligns with the Omega_m and sigma_8 directions in parameter space, indicating that the contrastive objective recovers the dominant axes of cosmological information. Our approach opens the door to more powerful, interpretable summary statistics for cosmological inference.

Source: arXiv cs.LG | 2026-06-11