Date
Publisher
arXiv
We introduce Confucius3-Math, an open-source large language model with 14B
parameters that (1) runs efficiently on a single consumer-grade GPU; (2)
achieves SOTA performances on a range of mathematical reasoning tasks,
outperforming many models with significantly larger sizes. In particular, as
part of our mission to enhancing education and knowledge dissemination with AI,
Confucius3-Math is specifically committed to mathematics learning for Chinese
K-12 students and educators. Built via post-training with large-scale
reinforcement learning (RL), Confucius3-Math aligns with national curriculum
and excels at solving main-stream Chinese K-12 mathematical problems with low
cost. In this report we share our development recipe, the challenges we
encounter and the techniques we develop to overcome them. In particular, we
introduce three technical innovations: Targeted Entropy Regularization, Recent
Sample Recovery and Policy-Specific Hardness Weighting. These innovations
encompass a new entropy regularization, a novel data scheduling policy, and an
improved group-relative advantage estimator. Collectively, they significantly
stabilize the RL training, improve data efficiency, and boost performance. Our
work demonstrates the feasibility of building strong reasoning models in a
particular domain at low cost. We open-source our model and code at
https://github.com/netease-youdao/Confucius3-Math.
Study at a glance
- Question: Can a low-cost, lightweight AI model be developed to effectively solve Chinese K-12 math problems?
- Setting and data: Chinese K-12 national curriculum mathematics problems.
- Method: AI model development and training (reinforcement learning).
Why it matters
Enhancing education and knowledge dissemination with AI requires supporting mathematics learning for Chinese K-12 students and educators. To achieve this, models must overcome training challenges to solve mainstream national curriculum mathematical problems at low cost.
Key findings
- In Chinese K-12 mathematics, an open-source model with 14 billion parameters achieved top reasoning performance, outperforming significantly larger models.
- The tool aligns with the national curriculum and excels at solving mainstream Chinese K-12 mathematics problems.
- The model runs efficiently on a single consumer-grade graphics card, demonstrating that advanced reasoning tools can operate at low cost.
Summary drafted with AI from the paper’s abstract and may contain errors. See the full paper for details.
What is the application?
Why use AI?
Study design
