Search papers, labs, and topics across Lattice.
This paper introduces a type-aware retrieval-augmented generation (RAG) method for automated industrial optimization modeling, addressing the challenge of LLMs generating non-compilable models. The method constructs a domain-specific typed knowledge base from heterogeneous sources and uses dependency propagation over a knowledge graph to retrieve a minimal dependency-closed context. Experiments on demand response optimization and flexible job shop scheduling demonstrate the method's ability to generate solver-executable models where conventional RAG baselines fail, highlighting the importance of type-aware dependency closure.
LLMs can now generate solver-executable optimization models from natural language by retrieving only the necessary typed dependencies, enabling automation in complex industrial settings.
Automated industrial optimization modeling requires reliable translation of natural-language requirements into solver-executable code. However, large language models often generate non-compilable models due to missing declarations, type inconsistencies, and incomplete dependency contexts. We propose a type-aware retrieval-augmented generation (RAG) method that enforces modeling entity types and minimal dependency closure to ensure executability. Unlike existing RAG approaches that index unstructured text, our method constructs a domain-specific typed knowledge base by parsing heterogeneous sources, such as academic papers and solver code, into typed units and encoding their mathematical dependencies in a knowledge graph. Given a natural-language instruction, it performs hybrid retrieval and computes a minimal dependency-closed context, the smallest set of typed symbols required for solver-executable code, via dependency propagation over the graph. We validate the method on two constraint-intensive industrial cases: demand response optimization in battery production and flexible job shop scheduling. In the first case, our method generates an executable model incorporating demand-response incentives and load-reduction constraints, achieving peak shaving while preserving profitability; conventional RAG baselines fail. In the second case, it consistently produces compilable models that reach known optimal solutions, demonstrating robust cross-domain generalization; baselines fail entirely. Ablation studies confirm that enforcing type-aware dependency closure is essential for avoiding structural hallucinations and ensuring executability, addressing a critical barrier to deploying large language models in complex engineering optimization tasks.
