TensorX
返回文献探索

Paper · arXiv 2412.15084

AceMath: Advancing Frontier Math Reasoning with Post-Training and Reward Modeling

Zihan Liu, Yang Chen, Mohammad Shoeybi, Bryan Catanzaro, Wei Ping

13 upvotesDecember 19, 2024arXiv 预印本
AI 摘要
supervised fine-tuning (SFT)instruction-tunedmath modelsfine-tuningpromptssynthetically generated responsesvision-and-language modelsreward modelsAceMath-RewardBenchbenchmarkmath reasoning benchmarksACEMath-72B-InstructQwen2.5-Math-72B-InstructGPT-4oClaude-3.5mathematical problemspan highlightingfeature extractionfeature representationrm@8 scoremodel evaluation

Abstract

In this paper, we introduce AceMath, a suite of frontier math models that excel in solving complex math problems, along with highly effective reward models capable of evaluating generated solutions and reliably identifying the correct ones. To develop the instruction-tuned math models, we propose a supervised fine-tuning (SFT) process that first achieves competitive performance across general domains, followed by targeted fine-tuning for the math domain using a carefully curated set of prompts and synthetically generated responses. The resulting model, AceMath-72B-Instruct greatly outperforms Qwen2.5-Math-72B-Instruct, GPT-4o and Claude-3.5 Sonnet. To develop math-specialized reward model, we first construct AceMath-RewardBench, a comprehensive and robust benchmark for evaluating math reward models across diverse problems and difficulty levels. After that, we present a systematic approach to build our math reward models. The resulting model, AceMath-72B-RM, consistently outperforms state-of-the-art reward models. Furthermore, when combining AceMath-72B-Instruct with AceMath-72B-RM, we achieve the highest average rm@8 score across the math reasoning benchmarks. We will release model weights, training data, and evaluation benchmarks at: https://research.nvidia.com/labs/adlr/acemath

北京市昌平区探索星信息技术及软件开发工作室

京ICP备2026059466号
AceMath: Advancing Frontier Math Reasoning with Post-Training and Reward Modeling | TensorX