Astro2Method: Benchmarking LLMs on Machine Learning Method Selection in Astronomy
Organizations: Department of Computer Science, University of Rochester Rochester, NY, USA
Abstract
Machine learning (ML) is increasingly used in astronomy, but choosing a suitable method for a scientific problem often requires expertise in both astronomy and ML. Large language models (LLMs) could help astronomers make such choices, yet this ability remains underexplored in existing benchmarks. We introduce Astro2Method, a benchmark of 500 expert-reviewed binary questions from 346 astronomy papers (2024/01-2026/06), based on clear performance differences and complete (Easy) or partial (Hard) explanations from the authors. Across five LLMs, direct prompting achieves 87.50-92.55% accuracy on Easy questions but only 58.87-64.52% on Hard. Using historical papers, neither retrieval-augmented generation nor most agent methods with additional reasoning or retrieval (including ReAct, Self-Ask, and CRITIC) bring consistent improvement. Agents required to search the web improve accuracy mainly when they find the papers the questions are based on, with no consistent improvement otherwise. Providing the source paper's pipeline description with method names masked improves Hard accuracy, matching or exceeding web search. Together, these results suggest that relevant experimental information helps distinguish ML methods, but inferring it from related studies and domain knowledge remains challenging for LLMs in astronomy.
Figures & tables
| Model | Overall | Easy | Hard |
|---|---|---|---|
| GPT-5.6 Luna | 82.40 | 89.63 | 60.48 |
| GPT-6 Luna | 80.40 | 87.50 | 58.87 |
| GPT-6 Astra | 85.60 | 92.55 | 64.52 |
| DeepSeek V4.1 Flash | 82.20 | 89.10 | 61.29 |
| Qwen 3.8 Flash | 82.60 | 89.89 | 60.48 |
| GPT-5.6 Luna | DeepSeek V4.1 Flash | Qwen 3.8 Flash | |||||||
| Method | Overall | Easy | Hard | Overall | Easy | Hard | Overall | Easy | Hard |
| [0pt][24pt] Direct prompting | 82.40 | 89.63 | 60.48 | 82.20 | 89.10 | 61.29 | 82.60 | 89.89 | 60.48 |
| Text RAG ( ) | |||||||||
| BM25 | 80.20 | 87.50 | 58.06 | 77.60 | 84.84 | 55.65 | 81.60 | 88.83 | 59.68 |
| E5 | 80.60 | 88.03 | 58.06 | 81.40 | 87.77 | 62.10 | 81.00 | 88.30 | 58.87 |
| RRF | 82.00 | 88.83 | 61.29 | 81.40 | 88.03 | 61.29 | 81.60 | 88.83 | 59.68 |
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
| Journal | Downloaded papers | Source papers | Questions | Historical retrieval |
|---|---|---|---|---|
| Monthly Notices of the Royal Astronomical Society (MNRAS) | 171 | 48 | 70 | 413 |
| The Astrophysical Journal (ApJ) | 247 | 72 | 105 | 241 |
| Astronomy & Astrophysics (A&A) | 222 | 70 | 102 | 176 |
| Physical Review D (PhRvD) | 61 | 14 | 17 | 89 |
| The Astrophysical Journal Supplement Series (ApJS) | 148 | 67 | 94 | 64 |
| The Astronomical Journal (AJ) | 74 | 31 | 51 | 65 |
Model Selection
Model selection chooses models, features, or configurations that balance predictive fit, generalization, complexity, and deployment constraints.
See all 96 papers