| --- |
| tags: |
| - math |
| license: apache-2.0 |
| datasets: |
| - oumi-ai/MetaMathQA-R1 |
| language: |
| - en |
| metrics: |
| - accuracy |
| base_model: |
| - deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B |
| pipeline_tag: text-generation |
| --- |
| |
| [](https://github.com/oumi-ai/oumi) |
| [](https://github.com/oumi-ai/oumi) |
|
|
| [](https://oumi.ai/docs/en/latest/index.html) |
| [](https://oumi.ai/blog) |
| [](https://discord.gg/oumi) |
|
|
| # oumi-ai/MiniMath-R1-1.5B |
|
|
| <!-- Provide a quick summary of what the model is/does. --> |
| Supervised fine-tune of [DeepSeek-R1-Distill-Qwen-1.5B](https://huggingface.co/deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B) using [oumi-ai/MetaMathQA-R1](https://huggingface.co/datasets/oumi-ai/MetaMathQA-R1). |
|
|
| Achieves **44.4% accuracy on MMLU-Pro-Math**, the **highest of any model with <=1.5B parameters**. |
|
|
| Improves the base model's accuracy by **+6 points**. |
|
|
|
|
| - **Developed by:** [Oumi AI](https://oumi.ai/) |
| - **Model type:** Small Language Model |
| - **Language(s) (NLP):** English |
| - **License:** [Apache 2.0](https://opensource.org/license/apache-2-0) |
| - **Finetuned from model:** [DeepSeek-R1-Distill-Qwen-1.5B](https://huggingface.co/deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B) |
| - **Demo:** [Fine-Tuning Notebook](https://github.com/oumi-ai/oumi/blob/307436bd98706cb9ce7b0bbf31204770af2b7c8c/notebooks/Oumi%20-%20MiniMath-R1-1.5B.ipynb) |
|
|
| ## Uses |
|
|
| <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. --> |
| Use as a conversational assistant for solving math problems with an exposed thought process. |
|
|
| ## Out-of-Scope Use |
|
|
| <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. --> |
| Smaller LLMs have limited capabilities and should be used with caution. Avoid using this model for purposes outside of mathematics. |
|
|
| ## Bias, Risks, and Limitations |
|
|
| <!-- This section is meant to convey both technical and sociotechnical limitations. --> |
| This model was finetuned with DeepSeek-R1 data on top of an R1-distill model, so any biases or risks associated with those models may be present. |
|
|
| ## Training Details |
|
|
| ### Training Data |
|
|
| <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. --> |
| Training data: [oumi-ai/MetaMathQA-R1](https://huggingface.co/datasets/oumi-ai/MetaMathQA-R1) |
|
|
| ### Training Procedure |
|
|
| <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. --> |
| Training notebook: [Fine-Tuning Notebook](https://github.com/oumi-ai/oumi/blob/307436bd98706cb9ce7b0bbf31204770af2b7c8c/notebooks/Oumi%20-%20MiniMath-R1-1.5B.ipynb) |
|
|
| ## Evaluation |
|
|
| <!-- This section describes the evaluation protocols and provides the results. --> |
|
|
|
|
| ## Environmental Impact |
|
|
| <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly --> |
|
|
| - **Hardware Type:** H100 |
| - **Hours used:** 0.8 (0.1 * 8 GPUs) |
| - **Cloud Provider:** Google Cloud Platform |
| - **Compute Region:** us-east5 |
| - **Carbon Emitted:** 0.07 kg |
|
|
| ## Citation |
|
|
| <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. --> |
|
|
| ``` |
| @misc{miniMathR1_2025, |
| author = {Jeremiah Greer}, |
| title = {MiniMath-R1-1.5B}, |
| month = {February}, |
| year = {2025}, |
| url = {https://huggingface.co/oumi-ai/MiniMath-R1-1.5B} |
| } |
| |
| @software{oumi2025, |
| author = {Oumi Community}, |
| title = {Oumi: an Open, End-to-end Platform for Building Large Foundation Models}, |
| month = {January}, |
| year = {2025}, |
| url = {https://github.com/oumi-ai/oumi} |
| } |
| ``` |