AI-Powered Reconstruction of Dark Matter Velocity Fields from Redshift-Space Halo Distribution

In the study of cosmology and galaxy evolution, the peculiar velocity and density field of dark matter (DM) play a crucial role in studying many issues. Here, we propose a UNet-based deep learning to reconstruct the real-space DM velocity field from the spatial distribution of a sparse sample of DM halos in redshift space. By comparing and testing various properties, we demonstrate that the reconstructed velocity field is in good agreement with the actual situation. At $k<0.3~h/{\rm Mpc}$, the reconstruction of various velocity field components, including velocity magnitude and divergence, outperforms traditional linear perturbation theory. Additionally, the effects of redshift space distortions (RSD) are well corrected using the UNet model. Compared to the true real-space power spectra, the UNet reconstruction provides an unbiased estimate of the density, velocity, and momentum fields, remaining consistent within $2\sigma$ level. We also demonstrate that the UNet model remains effective even with limited information about halo masses. Thus, our proposed UNet model has a wide range of applications in various aspects of cosmology, such as RSD, cosmic web analysis, the kinetic Sunyaev-Zel'dovich effect, BAO reconstruction, and so on.