OptMATH: A Scalable Bidirectional Data Synthesis Framework for Optimization Modeling
About
Despite the rapid development of large language models (LLMs), a fundamental challenge persists: the lack of high-quality optimization modeling datasets hampers LLMs' robust modeling of practical optimization problems from natural language descriptions (NL). This data scarcity also contributes to the generalization difficulties experienced by learning-based methods. To address these challenges, we propose a scalable framework for synthesizing a high-quality dataset, named OptMATH. Starting from curated seed data with mathematical formulations (MF), this framework automatically generates problem data (PD) with controllable complexity. Then, a back-translation step is employed to obtain NL. To verify the correspondence between the NL and the PD, a forward modeling step followed by rejection sampling is used. The accepted pairs constitute the training part of OptMATH. Then a collection of rejected pairs is identified and further filtered. This collection serves as a new benchmark for optimization modeling, containing difficult instances whose lengths are much longer than these of NL4OPT and MAMO. Through extensive experiments, we demonstrate that models of various sizes (0.5B-32B parameters) trained on OptMATH achieve superior results on multiple modeling benchmarks, thereby validating the effectiveness and scalability of our approach. Our dataset is publicly available at https://github.com/AuroraLHL/OptMATH.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Optimization Modeling | OptMATH | Accuracy Rate (AR)34.7 | 60 | |
| Optimization Modeling | NL4OPT | Accuracy (pass@1)95.9 | 53 | |
| Optimization modeling and solving | IndustryOR | Pass@1 (SA)19 | 42 | |
| Optimization modeling and solving | NLP4LP | SA Score68.6 | 34 | |
| Optimization Modeling | OptiBench | -- | 34 | |
| Optimization Modeling | ComplexOR | SA33.33 | 28 | |
| Optimization Modeling | IndustryOR | Accuracy (pass@1)19 | 23 | |
| Optimization Modeling | ComplexOR | Accuracy (pass@1)33.33 | 21 | |
| Optimization Modeling | MAMO Complex | ER64.46 | 19 | |
| Optimization modeling and solving | NL4OPT | Solution Accuracy78.7 | 19 |