Translation and rotation equivariant normalizing flow (TRENF) for optimal cosmological analysis
Translation and rotation equivariant normalizing flow (TRENF) for optimal cosmological analysis
复制标题
用于最佳宇宙学分析的平移和旋转等变归一化流 (TRENF)
DOI:
10.1093/mnras/stac2010
复制
发表时间:
2022
影响因子:
4.8
通讯作者:
Seljak, Uroš
中科院分区:
文献类型:
--
作者:
Dai, Biwei;Seljak, Uroš
Our Universe is homogeneous and isotropic, and its perturbations obey translation and rotation symmetry. In this work, we develop translation and rotation equivariant normalizing flow (TRENF), a generative normalizing flow (NF) model which explicitly incorporates these symmetries, defining the data likelihood via a sequence of Fourier space-based convolutions and pixel-wise non-linear transforms. TRENF gives direct access to the high dimensional data likelihoodp(x|y) as a function of the labelsy, such as cosmological parameters. In contrast to traditional analyses based on summary statistics, the NF approach has no loss of information since it preserves the full dimensionality of the data. On Gaussian random fields, the TRENF likelihood agrees well with the analytical expression and saturates the Fisher information content in the labelsy. On non-linear cosmological overdensity fields fromN-body simulations, TRENF leads to significant improvements in constraining power over the standard power spectrum summary statistic. TRENF is also a generative model of the data, and we show that TRENF samples agree well with theN-body simulations it trained on, and that the inverse mapping of the data agrees well with a Gaussian white noise both visually and on various summary statistics: when this is perfectly achieved the resultingp(x|y) likelihood analysis becomes optimal. Finally, we develop a generalization of this model that can handle effects that break the symmetry of the data, such as the survey mask, which enables likelihood analysis on data without periodic boundaries.