CoolFace
Modelpublic

ReasoningTransferability/UniReason-Qwen3-14B-RL

sourceHugging Faceapache-2.0updated 1y agoView on Hugging Face
3likes34downloads
Model Card

UniReason-Qwen3-14B-RL

This model is associated with the research paper: "Does Math Reasoning Improve General LLM Capabilities? Understanding Transferability of LLM Reasoning"

๐Ÿ“„ Paper: 2507.00432 ๐Ÿ’ป Code: https://github.com/ReasoningTransfer/Transferability-of-LLM-Reasoning

Abstract

Math reasoning has become the poster child of progress in large language models (LLMs), with new models rapidly surpassing human-level performance on benchmarks like MATH and AIME. But as math leaderboards improve week by week, it is worth asking: do these gains reflect broader problem-solving ability or just narrow overfitting?

Model Description

This model is a RL-GRPO-tuned version of qwen3-14b focused on math-reasoning capabilities. The model was developed as part of research investigating the transferability of mathematical reasoning skills to general language tasks.

Key Research Questions Addressed:

  • โ€”Does math reasoning training improve general LLM capabilities?
  • โ€”How do different training methods (RL vs SFT) affect transferability?
  • โ€”What is the trade-off between specialized math performance and general capabilities?

Model Details

  • โ€”Base Model: qwen3-14b
  • โ€”Training Method: RL-GRPO
  • โ€”Primary Focus: math-reasoning
  • โ€”Training Data: Math-specific datasets
  • โ€”Architecture: Transformer-based language model
  • โ€”Parameters: 14B

Training Details

Training Method: RL-GRPO

Custom training methodology - see paper for details.

Datasets Used

  • โ€”Mathematical reasoning datasets
  • โ€”See paper for complete dataset list

Performance

Math Reasoning Benchmarks

  • โ€”MATH: See paper
  • โ€”AIME: See paper

General Capabilities

  • โ€”General QA: See paper
  • โ€”Code Generation: See paper
  • โ€”Instruction Following: See paper

For detailed performance metrics, please refer to the paper.

Usage

python
from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

# Load model and tokenizer
model_name = "ReasoningTransferability/UniReason-Qwen3-14B-RL"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(
    model_name,
    torch_dtype=torch.float16,
    device_map="auto"
)

# Example: Math reasoning
math_prompt = "Solve this step by step: What is the derivative of x^3 + 2x^2 - 5x + 1?"
inputs = tokenizer(math_prompt, return_tensors="pt")
outputs = model.generate(**inputs, max_length=512, temperature=0.7)
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(response)

# Example: General reasoning
general_prompt = "Explain the concept of supply and demand in economics."
inputs = tokenizer(general_prompt, return_tensors="pt")
outputs = model.generate(**inputs, max_length=512, temperature=0.7)
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(response)

Limitations and Biases

  • โ€”Specialization Trade-offs: As explored in the paper, models optimized for math reasoning may show reduced performance on general tasks
  • โ€”Training Method Dependencies: Performance characteristics vary significantly between RL and SFT training approaches
  • โ€”Domain Transfer: The extent of capability transfer from math to other domains is limited
  • โ€”Computational Requirements: Model requires significant computational resources for inference

Research Findings

Key findings from the associated paper:

  1. 1.RL vs SFT: RL-tuned models show better transfer to general domains compared to SFT-tuned models
  2. 2.Capability Trade-offs: Most math-specialized models fail to transfer gains to other domains
  3. 3.Forgetting: SFT-tuned models often forget general capabilities during math-focused training

Ethical Considerations

  • โ€”This model is intended for research purposes
  • โ€”Users should be aware of potential biases in mathematical and general reasoning
  • โ€”The model should not be used for making critical decisions without human oversight
  • โ€”Consider the environmental impact of large model inference

Citation

If you use this model in your research, please cite both the model and the associated paper:

bibtex
@misc{huan2025doesmathreasoningimprove,
      title={Does Math Reasoning Improve General LLM Capabilities? Understanding Transferability of LLM Reasoning}, 
      author={Maggie Huan and Yuetai Li and Tuney Zheng and Xiaoyu Xu and Seungone Kim and Minxin Du and Radha Poovendran and Graham Neubig and Xiang Yue},
      year={2025},
      eprint={2507.00432},
      archivePrefix={arXiv},
      primaryClass={cs.AI},
      url={https://arxiv.org/abs/2507.00432}, 
}

Contact

For questions about this model or the associated research, please:

  • โ€”Open an issue in this repository
  • โ€”Contact the paper authors
  • โ€”Reference the original paper: https://arxiv.org/abs/2507.00432

Acknowledgments

This work builds upon the research presented in "Does Math Reasoning Improve General LLM Capabilities? Understanding Transferability of LLM Reasoning" and uses the qwen3-14b architecture as its foundation.


Model uploaded on 2025-07-03