project
OpenMath-Nemotron - NVIDIAのオープンソース数学推論モデルシリーズ
OpenMath-Nemotronは、NVIDIAが提供する一連のオープンソースの数学的推論モデルであり、オリンピックレベルの問題を含む複雑な数学的問題を解決するために特別に設計されています。これらのモデルは、大規模なデータセットであるOpenMathReasoningに基づいています。
OpenMath-Nemotronとは何ですか?
OpenMath-Nemotronは、NVIDIAが提供するオープンソースの数学的推論モデルシリーズで、オリンピックレベルの問題を含む複雑な数学的問題を解決するために特別に設計されています。これらのモデルは、54万の固有の問題と320万の長い推論ソリューションを含む大規模なOpenMathReasoningデータセットでトレーニングされています。OpenMath-Nemotronシリーズには、OpenMath-Nemotron-1.5B、OpenMath-Nemotron-7B、OpenMath-Nemotron-14B、OpenMath-Nemotron-32B、およびOpenMath-Nemotron-14B-Kaggle(AIMO-2コンペティションで使用されたモデル)が含まれます。1.5Bバージョンは、特定のタスクにおいて14B DeepSeek-R1モデルを上回る性能を発揮します。
OpenMath-Nemotronの主な機能
- 複雑な数学的問題を解く基礎レベルからオリンピックレベルまで、幅広い数学の問題解決をサポートします。
- 長い推論能力段階的な思考に基づき、詳細な問題解決手順を作成する。
- マルチモーダル推論複数の推論方法をサポートしており、さまざまな種類の数学的問題に適しています。
OpenMath-Nemotronの技術的原理
- 大規模データセットこのデータセットは、54万件の固有の数学問題と320万件の長文推論ソリューションを含む大規模なデータセットであるOpenMathReasoningを使用してトレーニングされました。データはArt of Problem Solving(AoPS)コミュニティフォーラムから収集され、厳密なスクリーニングと処理を経ています。
- 思考連鎖(CoT)このモデルは、一連の中間的な問題解決ステップを生成することで、問題の解決策を段階的に導き出すことに基づいています。最終的な答えを出す前に、より深く考えることをモデルに促します。
- ツール統合型推論(TIR)このアプローチは、反復的な学習、生成、および品質フィルタリングに基づく長時間の推論とコード実行を統合します。モデルは必要に応じて計算のためのコードを促し、サンドボックス内でコードを実行して、より正確な解を取得します。
- モデルのトレーニングと最適化Qwen2.5-Baseモデルは、教師ありファインチューニング(SFT)技術を用いて学習され、CoT解生成、TIR解生成、GenSelectなど、複数のタスクをサポートします。AdamWオプティマイザとコサイン学習率減衰戦略を、シーケンスパッキングおよびコンテキスト並列化技術と組み合わせることで、長大な推論データに対する学習を大幅に高速化します。
- 推論最適化モデル推論の最適化はTensorRT-LLMに基づいており、動的なバッチ処理とint8やFP8などの様々な量子化技術をサポートすることで、推論速度を向上させ、レイテンシを削減します。
OpenMath-Nemotronプロジェクトのアドレス
- GitHubリポジトリ:https://github.com/NVIDIA/NeMo-Skills
- ハギングフェイスモデルライブラリ:https://huggingface.co/collections/nvidia/openmathreasoning
- arXiv技術論文:https://arxiv.org/pdf/2504.16891
OpenMath-Nemotronの応用シナリオ
- 数学教育これは、生徒と教師が数学の問題を解決し、学習成果を向上させるのに役立ちます。
- 競技トレーニングこれは、数学コンテストの参加者が問題解決戦略を練習し、最適化するのに役立ちます。
- 学術研究複雑な数学的問題の探求を支援し、学術研究に貢献する。
- 産業用途実務的な工学や金融における数学的問題を解決し、効率性を向上させる。
- AI開発基礎モデルとして、数学的推論を必要とするAIシステムの開発を推進する。