Score-matching neural networks for improved multi-band source separation
arxiv(2024)
摘要
We present the implementation of a score-matching neural network that
represents a data-driven prior for non-parametric galaxy morphologies. The
gradients of this prior can be included in the optimization routine of the
recently developed multi-band modeling framework Scarlet2, a redesign of the
Scarlet method currently employed as deblender in the pipelines of the
HyperSuprimeCam survey and the Rubin Observatory. The addition of the prior
avoids the requirement of nondifferentiable constraints, which can lead to
convergence failures we discovered in Scarlet. We present the architecture and
training details of our score-matching neural network and show with simulated
Rubin-like observations that Scarlet2 outperforms Scarlet in accuracy of total
flux and morphology estimates, while maintaining excellent performance for
colors. We also demonstrate significant improvements in the robustness to
inaccurate initializations. Scarlet2 is written in python, extendend by JAX and
equinox, and is fully GPU compatible. The implementation and data package of
the score model are publicly available at
https://github.com/pmelchior/scarlet2.
更多查看译文
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要