ORLM: A Customizable Framework in Training Large Models for Automated Optimization Modeling

TL;DR

This paper introduces ORLM, an open-source framework for training large language models tailored to optimization modeling, addressing data scarcity and privacy issues, and demonstrating competitive performance on industry-relevant benchmarks.

Contribution

It presents a semi-automated data synthesis framework, OR-Instruct, for customizable training of open-source LLMs in optimization modeling, and introduces IndustryOR, a new industrial benchmark.

Findings

ORLM models outperform existing benchmarks in optimization tasks

Synthesized data significantly improves LLM capabilities in OR modeling

Scaling law and reinforcement learning can further enhance ORLM performance

Abstract

Optimization modeling plays a critical role in the application of Operations Research (OR) tools to address real-world problems, yet they pose challenges and require extensive expertise from OR experts. With the advent of large language models (LLMs), new opportunities have emerged to streamline and automate such task. However, current research predominantly relies on closed-source LLMs such as GPT-4, along with extensive prompt engineering techniques. This reliance stems from the scarcity of high-quality training datasets for optimization modeling, resulting in elevated costs, prolonged processing times, and privacy concerns. To address these challenges, our work is the first to propose a viable path for training open-source LLMs that are capable of optimization modeling and developing solver codes, eventually leading to a superior ability for automating optimization modeling and solving. Particularly, we design the {\sc OR-Instruct}, a semi-automated data synthesis framework for optimization modeling that enables customizable enhancements for specific scenarios or model types. This work also introduces IndustryOR, the first industrial benchmark for evaluating LLMs in solving practical OR problems. We train several 7B-scale open-source LLMs using synthesized data (dubbed ORLMs{https://github.com/Cardinal-Operations/ORLM}), which exhibit significantly enhanced optimization modeling capabilities, achieving competitive performance across the NL4OPT, MAMO, and IndustryOR benchmarks. Additionally, our experiments highlight the potential of scaling law and reinforcement learning to further enhance the performance of ORLMs. The workflows and human-machine interaction paradigms of ORLMs in practical industrial applications are also discussed in the paper.