Explainable Universe

The Big Picture

The Universe around us is shaped by material we can see, like stars and galaxies, and by that which remains invisible. One of the biggest mysteries in modern science is Dark Matter, an unseen substance that makes up most of the matter in the cosmos. Though we can’t observe it directly, its gravitational pull shapes how galaxies form, how gas and stars move, and how the Universe evolves. To understand this hidden matter, researchers in the Explainable Universe working group use supercomputers to build virtual Universes and then explore the impact that different kinds of Dark Matter would have on cosmic structure and formation history of the cosmos.

But, simulating the cosmos is only half the story. Real astronomical data are messy, incomplete, and uncertain. To make sense of this complexity, our team is also developing AI-powered inference tools that can learn the characteristics, features, and patters from these simulated Universes and then help us draw clear connections with data taken from real telescope observations. These tools are designed not only to make accurate predictions but help scientists build trust in the results.

Looking ahead, the Explainable Universe team will expand these virtual Universes to include many different scenarios for how the cosmos might work, and will train AI systems to compare these scenarios with real data. By combining cutting-edge cosmological simulations with explainable AI methods, we aim to peel back the layers of mystery around Dark Matter and the forces that shape our Universe.

Research Overview

The Explainable Universe working group at CosmicAI focuses on linking physical theories of cosmic structure formation with interpretable machine learning methods to enable robust, explainable inference from complex astronomical data. This effort is organized around two complementary thrusts: (1) the generation of large, systematically varied suites of cosmological simulations to probe the nature of Dark Matter and cosmic structure formation, and (2) the development of explainable AI methods for scientific inference under realistic, imperfect data conditions.

On the astrophysics side, the group is using commonly employed codes, including AREPO, RAMSES, and GIZMO, to generate controlled cosmological simulation suites. In distinct contrast to previous studies (e.g., like Illustris, IllustrisTNG, Simba, FIREBox, etc.), these simulation suites include thousands of simulations that systematically vary the input physical assumptions (e.g., the properties of Dark Matter) to create maps linking changed physical assumptions to identifiable impacts on galaxy formation and galaxy properties. By producing a broad library of realizations, we can trace how fundamental physics shapes observable structures in the Universe. This work is closely intertwined with the broader DREAMS collaboration and will scale toward larger, more diverse parameter spaces in upcoming runs.

On the AI side, the group develops simulation-based inference (SBI) pipelines that pair these simulation suites with machine learning models that draw probabilistic, interpretable conclusions about underlying cosmological parameters. Methods under development include graph neural networks for learning structural relationships between galaxies and their environments, uncertainty-aware inference models to quantify confidence, and explainability techniques such as saliency mapping and feature attribution to identify which physical signatures most strongly constrain model parameters. The emphasis is on explainability by design — ensuring that models provide insight into why they reach specific conclusions.

A central challenge in cosmology is that observational data are incomplete and noisy, and are subject to biases that can obscure underlying physics. Our inference framework explicitly addresses this through robustness testing, domain adaptation, and uncertainty calibration. These steps allow us to draw trustworthy scientific inferences even when real data deviate from idealized training sets. This capability is particularly crucial for leveraging upcoming large-scale survey missions, which will provide unprecedented but complex data.

Looking ahead, the Explainable Universe working group will expand both sides of this pipeline: scaling astrophysical simulation suites to encompass more physical models and initial conditions, and advancing AI frameworks to support explainable, uncertainty-aware inference at survey scale. This approach aims to turn astronomical observations into transparent, scientifically grounded constraints on the nature of Dark Matter and cosmic structure formation.

Projects

Simulation Development

Astro Projects

Publications/Works in Progress

The DREAMS Project: Disentangling the Impact of Halo-to-Halo Variance and Baryonic Feedback on Milky Way Dark Matter Density Profiles

Garcia, Alex M., Rose, J., Torrey, P., et al. (including Ou, X., Bhowmick, A., Farahi, A., Kallivayalil, N.

Linking Warm Dark Matter to Merger Tree Histories via Deep Learning Networks

Leisher, I., Torrey, P., Garcia, A., et al. (including Rose, J., Farahi, A., Kallivayalil, N.)

The DREAMS Project: Dark Matter Annihilation and Decay Signals from the Galactic Center
(in progress)

Garcia, Alex M., et al.

Team

  • Paul Torrey

    Astro Lead
    University of Virginia

  • Arya Farahi

    AI Lead
    UT Austin

  • Jeff Phillips

    Faculty Researcher
    University of Utah

  • Judy Fox

    Faculty Researcher
    University of Virginia

  • Alessandro Rinaldo

    Faculty Researcher
    UT Austin

  • Nitya Kallivayalil

    Faculty Researcher
    University of Virginia

  • Jonah Rose

    Postdoctoral Scholar
    Princeton University

  • Aklant Bhowmick

    Postdoctoral Scholar
    University of Virginia

  • Niusha Ahvazi

    Postdoctoral Scholar
    University of Virginia

  • Xiaowei Ou

    Postdoctoral Scholar
    University of Virginia

  • Alex Garcia

    Graduate Student
    University of Virginia

  • Jonathan Kho

    Graduate Student
    University of Virginia

  • Md. Khairul Islam

    Graduate Student
    University of Virginia

  • Walter Wang

    Graduate Student
    UT Austin

  • Giti Doolabi

    Graduate Student
    University of Virginia

  • Saiyang Zhang

    Graduate Student
    UT Austin

  • Nevena Gligić

    Graduate Student
    UT Austin

  • Akhil Kotturi

    Graduate Student
    UT Austin

  • Yanbo Pan

    Graduate Student
    University of Virginia

  • Shuting Xu

    Graduate Student
    Shuting Xu