Fetching the paper…

Reading the bibliography…

2025

The AI Cosmologist I: An Agentic System for Automated Data Analysis

Moss, Adam

Original
PDF
Post on X

Understand

We present the AI Cosmologist, an agentic system designed to automate cosmological/astronomical data analysis and machine learning research workflows.

This implements a complete pipeline from idea generation to experimental evaluation and research dissemination, mimicking the scientific process typically performed by human researchers.
The system employs specialized agents for planning, coding, execution, analysis, and synthesis that work together to develop novel approaches.
Unlike traditional auto machine-learning systems, the AI Cosmologist generates diverse implementation strategies, writes complete code, handles execution errors, analyzes results, and synthesizes new approaches based on experimental outcomes.

The AI Cosmologist I: An Agentic System for Automated Data Analysis · Around

Built on

Baron, D.: Machine Learning in Astronomy: a practical overview. arXiv e-prints, 1904–07248 (2019) https://doi.org/10.48550/arXiv.1904.07248 arXiv:1904.07248 [astro-ph.IM]
Original
1904
Earlier work this paper cites.
Villaescusa-Navarro, F., et al
Original
1909
Earlier work this paper cites.
Collister, A.A., Lahav, O.: ANNz: Estimating Photometric Redshifts Using Artificial Neural Networks. PASP 116
2004
Earlier work this paper cites.
Laureijs, R., et al.: Euclid Definition Study Report (2011) arXiv:1110.3193 [astro-ph.CO]
Original
2011
Earlier work this paper cites.
Bloom, J.S., Richards, J.W., Nugent, P.E., Quimby, R.M., Kasliwal, M.M., Starr, D.L., Poznanski, D., Ofek, E.O., Cenko, S.B., Butler, N.R., Kulkarni, S.R., Gal-Yam, A., Law, N.: Automating Discovery and Classification of Transients and Variable Stars in the Synoptic Survey Era. PASP 124
Original
2012
Earlier work this paper cites.
Thornton, C., Hutter, F., Hoos, H.H., Leyton-Brown, K.: Auto-weka: Combined selection and hyperparameter optimization of classification algorithms. In: KDD, pp. 847–855 (2013)
2013
Earlier work this paper cites.
Willett, K.W., et al
Original
2013
Earlier work this paper cites.
Dieleman, S., Willett, K.W., Dambre, J.: Rotation-invariant convolutional neural networks for galaxy morphology prediction. MNRAS 450
Original
2015
Earlier work this paper cites.
Feurer, M., al.: Efficient and robust automated machine learning. In: NIPS, pp. 2962–2970 (2015)
2015
Earlier work this paper cites.
Aghamousa, A., et al.: The DESI Experiment Part I: Science,Targeting, and Survey Design (2016) arXiv:1611.00036 [astro-ph.IM]
Original
2016
Earlier work this paper cites.
Olson, R.S., Bartley, N., Urbanowicz, R.J., Moore, J.H.: Evaluation of a tree-based pipeline optimization tool for automating data science. In: Genetic and Evolutionary Computation Conference (GECCO) Companion, pp. 485–492 (2016)
2016
Earlier work this paper cites.
Petrillo, C.E., Tortora, C., Chatterjee, S., Vernardos, G., Koopmans, L.V.E., Verdoes Kleijn, G., Napolitano, N.R., Covone, G., Schneider, P., Grado, A., McFarland, J.: Finding strong gravitational lenses in the Kilo Degree Survey with Convolutional Neural Networks. MNRAS 472
Original
2017
Earlier work this paper cites.
Zoph, B., Le, Q.V.: Neural architecture search with reinforcement learning. In: ICLR (2017)
2017
Earlier work this paper cites.
Finn, C., Abbeel, P., Levine, S.: Model-agnostic meta-learning for fast adaptation of deep networks. In: Proceedings of the 34th International Conference on Machine Learning (ICML), pp. 1126–1135 (2017)
2017
Earlier work this paper cites.
George, D., Huerta, E.A.: Deep neural networks to enable real-time multimessenger astrophysics. Phys. Rev. D 97
Original
2018
Earlier work this paper cites.
Butler, K.T., Davies, D.W., Cartwright, H., Isayev, O., Walsh, A.: Machine learning for molecular and materials science. Nature 559
2018
Earlier work this paper cites.

Similar

Ivezić, v., et al
Original
2019
Cited alongside, same era.
Bacon, D.J., et al
Original
2019
Cited alongside, same era.
Reichstein, M., Camps-Valls, G., Stevens, B., Jung, M., Denzler, J., Carvalhais, N., Prabhat: Deep learning and process understanding for data-driven Earth system science. Nature 566
2019
Cited alongside, same era.
Hutter, F., Kotthoff, L., Vanschoren, J.: Automated machine learning: Methods, systems, challenges. Automated Machine Learning (2019)
2019
Cited alongside, same era.
Elsken, T., Metzen, J.H., Hutter, F.: Neural architecture search: A survey. Journal of Machine Learning Research 20
2019
Cited alongside, same era.

Then

Tarsitano, C., La Barbera, F., Vavilova, I., al.: Image feature extraction and galaxy classification: a novel approach using automated machine learning. Monthly Notices of the Royal Astronomical Society 511
2022
Later among the works it cites.
Nijkamp, E., Pang, B., Hayashi, H., Tu, L., Wang, H., Zhou, Y., Savarese, S., Xiong, C.: CodeGen: An Open Large Language Model for Code with Multi-Turn Program Synthesis. arXiv preprint arXiv:2203.13474 (2022)
Original
2022
Later among the works it cites.
Yao, S., Zhao, J., Yu, D., Du, N., Shafran, I., Narasimhan, K., Cao, Y.: ReAct: Synergizing Reasoning and Acting in Language Models. In: Proc. International Conference on Learning Representations (ICLR) (2023)
2023
Later among the works it cites.
Shinn, N., al.: Reflexion: Language agents with verbal reinforcement learning. arXiv preprint arXiv:2303.11366 (2023)
Original
2023
Later among the works it cites.

Beyond the bibliography

alphaXiv searches the wider corpus for related work and actual follow-ups.

Open on alphaXiv

alphaXiv is searching for related work…

Vanschoren, J.: Meta-learning: A survey. arXiv preprint arXiv:1810.03548 (2019)

2019

Cited alongside, same era.

Real, E., Aggarwal, A., Huang, Y., Le, Q.V.: Regularized evolution for image classifier architecture search. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 33, pp. 4780–4789 (2019)

2019

Cited alongside, same era.

Liu, H., Simonyan, K., Yang, Y.: DARTS: Differentiable architecture search. In: International Conference on Learning Representations (ICLR) (2019)

2019

Cited alongside, same era.

Cranmer, K., Brehmer, J., Louppe, G.: The frontier of simulation-based inference. PNAS 117

2020

Cited alongside, same era.

Brehmer, J., al.: Mining gold from implicit models to improve likelihood-free inference. PNAS 117

2020

Cited alongside, same era.

Jumper, J., Evans, R., Pritzel, A., Green, T., Figurnov, S., Ronneberger, O., Tunyasuvunakool, K., Bates, R., …, Hassabis, D.: Highly accurate protein structure prediction with AlphaFold. Nature 596

2021

Cited alongside, same era.

Zöller, M.-A., Huber, M.F.: Benchmark and survey of automated machine learning frameworks. Journal of artificial intelligence research 70

2021

Cited alongside, same era.

Fried, D., Aghajanyan, A., Lin, J., Wang, S.I., Wallace, E., Shi, F., Zhong, R., Yih, W.-t., Zettlemoyer, L., Lewis, M.: InCoder: A generative model for code infilling and synthesis. In: International Conference on Learning Representations (ICLR) (2023)

2023

Later among the works it cites.

Li, R., Allal, L.B., Zi, Y., Muennighoff, N., Kocetkov, D., Mou, C., Marone, M., Akiki, C., et al.: StarCoder: May the source be with you! arXiv preprint arXiv:2305.06161 (2023)

2023

Later among the works it cites.

Yao, S., Zhao, J., Yu, D., Du, N., Shafran, I., Narasimhan, K., Cao, Y.: ReAct: Synergizing reasoning and acting in language models. Proceedings of the 40th International Conference on Machine Learning (ICML), 11837–11852 (2023)

2023

Later among the works it cites.

Shinn, N., Cassano, F., Berman, E., Gopinath, A., Narasimhan, K., Yao, S.: Reflexion: Language agents with verbal reinforcement learning. In: Advances in Neural Information Processing Systems 36 (NeurIPS 2023), pp. 41895–41908 (2023)

2023

Later among the works it cites.

Shen, Y., Song, K., Tan, X., Li, D., Lu, W., Zhuang, Y.: HuggingGPT: Solving AI Tasks with ChatGPT and its Friends in Hugging Face. arXiv preprint arXiv:2303.17580 (2023)

2023

Later among the works it cites.

Boiko, D.A., MacKnight, R., Kline, B., Gomes, G.: Autonomous chemical research with large language models. Nature 624

2023

Later among the works it cites.

Xu, M., al.: Expert iteration with language models for scientific discovery. arXiv preprint arXiv:2303.03494 (2023)

2023

Later among the works it cites.

Yang, J., Jimenez, C., Wettig, A., Lieret, K., Yao, S., Narasimhan, K., Press, O.: Swe-agent: Agent-computer interfaces enable automated software engineering. Advances in Neural Information Processing Systems 37

2024

Later among the works it cites.

M. Bran, A., Cox, S., Schilter, O., Baldassari, C., White, A.D., Schwaller, P.: Augmenting large language models with chemistry tools. Nature Machine Intelligence 6

2024

Later among the works it cites.

Laverick, A., Surrao, K., Zubeldia, I., Bolliet, B., Cranmer, M., Lewis, A., Sherwin, B., Lesgourgues, J.: Multi-Agent System for Cosmological Parameter Analysis (2024)

2024

Later among the works it cites.

Charnock, T., Moss, A.: Deep Recurrent Neural Networks for Supernovae Classification. ApJ 837

2041

Closest in time.