Model Name: AstroSage-70B-20251009
Version: 2.0
Release Date: 2025-10-09
Developed by: AstroMLab (Tijmen de Haan, Yuan-Sen Ting, Tirthankar Ghosal, Tuan Dung Nguyen, Alberto Accomazzi, Emily Herron, Vanessa Lama, Azton Wells, Nesar Ramachandra, Rui Pan)
Corresponding Contact: Tijmen de Haan (tijmen.dehaan@gmail.com)
Funded by:
- Oak Ridge Leadership Computing Facility (OLCF), a DOE Office of Science User Facility at Oak Ridge National Laboratory (U.S. Department of Energy).
- Microsoft’s Accelerating Foundation Models Research (AFMR) program.
- World Premier International Research Center Initiative (WPI), MEXT, Japan.
- National Science Foundation (NSF).
- UChicago Argonne LLC, Operator of Argonne National Laboratory (U.S. Department of Energy).
License: Llama 3.1 Community License
Reference Paper: Tijmen de Haan et al. (2025). "AstroMLab 4: Benchmark-Topping Performance in Astronomy Q&A with a 70B-Parameter Domain-Specialized Reasoning Model" https://arxiv.org/abs/2505.17592
Model Details
Model Type: Autoregressive transformer-based LLM, specialized in astronomy, astrophysics, space science, astroparticle physics, cosmology, and astronomical instrumentation.
Base Model: Meta-Llama-3.1-70B
Model Architecture: AstroSage-70B is a fine-tuned derivative of the Meta-Llama-3.1-70B architecture, making no architectural changes. The Llama-3.1-70B-Instruct tokenizer is also used without modification.
Context Length: Fine-tuned on 8192-token sequences. Base model was trained to 128k context length.
Model Description
Overview: AstroSage-70B is a large-scale, domain-specialized language model tailored for research and education in astronomy, astrophysics, space science, cosmology, and astronomical instrumentation. It builds on the Llama-3.1-70B foundation model, enhanced through extensive continued pre-training (CPT) on a vast corpus of astronomical literature, further refined with supervised fine-tuning (SFT) on instruction-following datasets, and finally enhanced via parameter averaging (model merging) with other popular fine tunes. AstroSage-70B aims to achieve state-of-the-art performance on astronomy-specific tasks, providing researchers, students, and enthusiasts with an advanced AI assistant. This 70B parameter model represents a significant scaling up from the AstroSage-8B model. The primary enhancements from the AstroSage-8B model are:
- Stronger base model, higher parameter count for increased capacity
- Improved datasets
- Improved learning hyperparameters
- Reasoning capability (can be enabled or disabled at inference time)
Training Lineage
- Base Model: Meta-Llama-3.1-70B.
- Continued Pre-Training (CPT): The base model underwent 2.5 epochs of CPT (168k GPU-hours) on a specialized astronomy corpus (details below, largely inherited from AstroSage-8B) to produce AstroSage-70B-CPT. This stage imbues domain-specific knowledge and language nuances.
- Supervised Fine-Tuning (SFT): AstroSage-70B-CPT was then fine-tuned for 2.0 epochs (67k GPU-hours) using astronomy-relevant and general-purpose instruction-following datasets, resulting in AstroSage-70B-SFT.
- Final Mixture: The released AstroSage-70B model is created via parameter averaging / model merging:
- DARE-TIES
- AstroSage-70B-base designated as the "base model"
- 70% AstroSage-70B-SFT (density 0.7)
- 30% Llama-3.1-Instruct (density 0.5)
Intended Use: Like AstroSage-8B, this model can be used for a variety of LLM application, including
- Providing factual information and explanations in astronomy, astrophysics, cosmology, and instrumentation.
- Assisting with literature reviews and summarizing scientific papers.
- Answering domain-specific questions with high accuracy.
- Brainstorming research ideas and formulating hypotheses.
- Assisting with programming tasks related to astronomical data analysis.
- Serving as an educational tool for learning astronomical concepts.
- Potentially forming the core of future agentic research assistants capable of more autonomous scientific tasks.
We hope that with the enhanced intelligence and reasoning ability of AstroSage-70B compared to AstroSage-8B you can find additional use cases.
Bias, Risks, and Limitations
- Knowledge Cutoff: The model's knowledge cutoff is a mixture of the knowledge cutoff of the foundation model Llama-3.1-70B, which is December 2023, and the cutoff for the astro-ph papers used in the training dataset, which is January 2024. The model will not be aware of developments in astronomy after that cutoff.
- Hallucinations & Factual Accuracy: All LLMs, including AstroSage-70B, can generate outputs that sound plausible but are incorrect or nonsensical. All scientific or factual claims must be verified by users.
- Inherited Biases: The model may reflect biases present in its extensive training data (e.g., from scientific literature, general web data).
- Errors in Training Data: After training completed, the AstroSage Q&A SFT dataset was found to have a non-trivial proportion of references to "the article" or "the text", which are artifacts of how the synthetic training data was generated. The insufficient cleaning resulted in AstroSage-70B inadvertently being trained to occasionally refer to information that is not available in the current conversation. We find that this failure mode is however, fairly rare.
- Depth of Specialization: AstroSage-70B is highly specialized, having been trained on all astro-ph papers from April 2007 to January 2024. However, LLM training does not result in complete memorization of all facts. We find it excels in topics that are discussed in approximately five or more papers.
Chat Template
AstroSage-70B follows the Llama-3.1 chat template. For example:
<|begin_of_text|><|start_header_id|>system<|end_header_id|>
You are an expert in astronomy, astrophysics, space science, cosmology, and astronomical instrumentation. Your role is provide helpful, factual answers to the user's query.<|eot_id|><|start_header_id|>user<|end_header_id|>
Explain the ISW effect.<|eot_id|><|start_header_id|>assistant<|end_header_id|>
Enabling Reasoning (a.k.a. Thinking, Chain-of-Thought)
Like popular models such as o1, QwQ and DeepSeek-R1, AstroSage-70B is capable of reasoning through a problem before giving an answer. To enable this:
- Set the system prompt to:
detailed thinking on - Prefill the assistant response with
<think>
To enable reasoning for the example above, you would give
<|begin_of_text|><|start_header_id|>system<|end_header_id|>
detailed thinking on<|eot_id|><|start_header_id|>user<|end_header_id|>
Explain the ISW effect.<|eot_id|><|start_header_id|>assistant<|end_header_id|>
<think>