Cosmo-PINN: A Physics-Informed Neural Network for Cosmological Reconstruction

Abstract: We introduce Cosmo-PINN, a Physics-Informed Neural Network for reconstruction of the cosmological theory. In this work we demonstrate the application of the Cosmo-PINN in the reconstruction of the dark energy equation of state parameter $w_{DE}\left( z\right) $ directly from late-time cosmological observations. This framework overcomes the main limitation shared by Gaussian Process and Artificial Neural Network reconstruction approaches, where the recovered solution is driven by the data and it is not necessarily true that it is physically consistent, by embedding the cosmological constraints directly into the loss function as hard constraints, ensuring that the reconstructed quantities satisfy the physical laws at every point during the training. For the training of the network, we employed background data, and specifically the Baryon Acoustic Oscillation from DESI DR2, the Cosmic Chronometers and three different Supernova compilations, while we simultaneously introduce the cosmological parameters $H_{0},~Ω{m0}$ and $r{\mathrm{drag}}$ as trained parameters. The reconstruction shows that the trained $w_{DE}\left( z\right) $ crosses the phantom divide within the redshift range $z=0.27-0.42$ in agreement with the value obtained by the Chevallier-Polarski-Linder model. In the quintessence scenario, for large redshifts the dark energy $Ω_{DE}\left( z\right) $ provides a pressureless nonzero contribution to the cosmological fluid suggesting a unified scenario. Finally, we demonstrate the significance of imposing the physical constraints within the loss function by comparing the Cosmo-PINN reconstruction against a purely data-driven neural network with the same architecture.