Tool-Augmented Agent for Closed-loop Optimization,Simulation,and Modeling Orchestration

arXiv cs.AI Papers

Summary

The paper introduces COSMO-Agent, a tool-augmented reinforcement learning framework that trains LLMs to perform closed-loop CAD-CAE optimization, iteratively generating parametric geometries and running simulations until constraints are satisfied, with a multi-constraint reward and a new industry-aligned dataset.

arXiv:2605.20190v1 Announce Type: new Abstract: Iterative industrial design-simulation optimization is bottlenecked by the CAD-CAE semantic gap: translating simulation feedback into valid geometric edits under diverse, coupled constraints. To fill this gap, we propose COSMO-Agent (Closed-loop Optimization, Simulation, and Modeling Orchestration), a tool-augmented reinforcement learning (RL) framework that teaches LLMs to complete the closed-loop CAD-CAE process. Specifically, we cast CAD generation, CAE solving, result parsing, and geometry revision as an interactive RL environment, where an LLM learns to orchestrate external tools and revise parametric geometries until constraints are satisfied. To make this learning stable and industrially usable, we design a multi-constraint reward that jointly encourages feasibility, toolchain robustness, and structured output validity. In addition, we contribute an industry-aligned dataset that covers 25 component categories with executable CAD-CAE tasks to support realistic training and evaluation. Experiments show that COSMO-Agent training substantially improves small open-source LLMs for constraint-driven design, exceeding large open-source and strong closed-source models in feasibility, efficiency, and stability.
Original Article
View Cached Full Text

Cached at: 05/22/26, 08:46 AM

# Tool-Augmented Agent for Closed-loop Optimization, Simulation, and Modeling Orchestration
Source: [https://arxiv.org/html/2605.20190](https://arxiv.org/html/2605.20190)
Liyuan Deng1,2Shujian Deng2111L\. Deng and S\. Deng contributed equally\.Yongkang Chen2Yongkang Dai1Zhihang Zhong2 Linyang Li2Xiao Sun2Yilei Shi1Huaxi Huang2222Y\. Shi and H\. Huang are co\-corresponding authors\. 1Northwestern Polytechnical University2Shanghai Artificial Intelligence Laboratory \{dly,yilei\_shi\}@mail\.nwpu\.edu\.cn\{dengshujian,huanghuaxi\}@pjlab\.org\.cn

###### Abstract

Iterative industrial design–simulation optimization is bottlenecked by the CAD–CAE semantic gap: translating simulation feedback into valid geometric edits under diverse, coupled constraints\. To fill this gap, we proposeCOSMO\-Agent\(Closed\-loopOptimization,Simulation, andModelingOrchestration\), a tool\-augmented reinforcement learning \(RL\) framework that teaches LLMs to complete the closed\-loop CAD–CAE process\. Specifically, we cast CAD generation, CAE solving, result parsing, and geometry revision as an interactive RL environment, where an LLM learns to orchestrate external tools and revise parametric geometries until constraints are satisfied\. To make this learning stable and industrially usable, we design a multi\-constraint reward that jointly encourages feasibility, toolchain robustness, and structured output validity\. In addition, we contribute an industry\-aligned data set that covers 25 component categories with executable CAD–CAE tasks to support realistic training and evaluation\. Experiments show that COSMO\-Agent training substantially improves small open\-source LLMs for constraint\-driven design, exceeding large open\-source and strong closed\-source models in feasibility, efficiency, and stability\.

![[Uncaptioned image]](https://arxiv.org/html/2605.20190v1/imgs/teaser1.png)

Figure 1:COSMO\-Agent performs closed\-loop CAD–CAE optimization by iteratively generating parametric geometry, running CAE simulation, extracting displacement/stress metrics, and updating design parameters until all constraints are satisfied\.
## 1Introduction

Modern industrial design is an iterative search for geometries that satisfy coupled and often competing constraints\. In functional component development, Computer\-Aided Design \(CAD\) specifies a parametric geometry with a feature/history tree, while Computer\-Aided Engineering \(CAE\) provides physics\-based verification through simulation \(e\.g\., finite element analysis\)\. Despite advances in automation,*closed\-loop*CAD–CAE iteration remains a practical bottleneck: engineers must translate high\-dimensional simulation feedback \(fields and aggregate metrics\) into low\-dimensional,*structured*CAD edits that remain executable under the original parametric history\. This translation is further complicated by heterogeneous toolchains that frequently fail mid\-pipeline, where regeneration breakdowns, meshing errors, and solver non\-convergence are common\. As a result, CAD–CAE optimization in practice becomes a long\-horizon sequential decision\-making problem under hard executability constraints and stochastic tool failures, rather than a clean continuous optimization problem\.

Existing automation strategies only partially address this setting\. Derivative\-free optimizers\[[13](https://arxiv.org/html/2605.20190#bib.bib10),[11](https://arxiv.org/html/2605.20190#bib.bib11),[3](https://arxiv.org/html/2605.20190#bib.bib12),[29](https://arxiv.org/html/2605.20190#bib.bib13)\]can adjust parameters to scalar objectives, but typically do not model*executability*and*failure recovery*as part of the optimization state\. Moreover, validity is usually enforced by rigid templates or hand\-crafted heuristics\. Differentiable or surrogate\-based methods can reduce expensive solver calls, but frequently rely on approximations that diverge from production CAD–CAE pipelines and do not directly produce history\-consistent executable edits in native parametric CAD\[[23](https://arxiv.org/html/2605.20190#bib.bib14),[18](https://arxiv.org/html/2605.20190#bib.bib15),[17](https://arxiv.org/html/2605.20190#bib.bib16),[12](https://arxiv.org/html/2605.20190#bib.bib17)\]\. Recent progress suggests an alternative: an LLM that serves as a controller, mapping tool feedback to structured tool invocations\[[38](https://arxiv.org/html/2605.20190#bib.bib8),[1](https://arxiv.org/html/2605.20190#bib.bib34)\], while tool\-use training improves API\-call fidelity\[[25](https://arxiv.org/html/2605.20190#bib.bib9),[22](https://arxiv.org/html/2605.20190#bib.bib18)\]\. However, prompting\-first agents remain brittle under regeneration/meshing/solver failures, while standard instruction tuning or RLHF mainly targets short\-horizon imitation/alignment rather than long\-horizon trial\-and\-error optimization driven by downstream simulation consequences\[[32](https://arxiv.org/html/2605.20190#bib.bib7),[31](https://arxiv.org/html/2605.20190#bib.bib21),[21](https://arxiv.org/html/2605.20190#bib.bib22),[6](https://arxiv.org/html/2605.20190#bib.bib23)\]\.

To address these challenges, we introduceCOSMO\-Agent\(Closed\-loopOptimization,Simulation, andModelingOrchestration\), a novel tool\-augmented reinforcement learning framework for reliable closed\-loop CAD–CAE iterative evolution\. We model CAD editing, regeneration, meshing, solving and result parsing as an interactive environment with explicit failure states\. An LLM policy operates over a structured action space of parametric edits\. Each proposed edit is validated by a CAD tool set and evaluated through simulation, and the agent iteratively revises actions based on tool feedback until all constraints are satisfied or the budget is exhausted\. To make learning stable in partially reliable pipelines, we optimize a multi\-constraint objective that jointly encourages \(i\)*feasibility*via constraint satisfaction, \(ii\)*robustness*via successful execution and recovery from toolchain failures, and \(iii\)*structured validity*aligned with parametric requirements, preventing reward hacking that produces numerically favorable but non\-executable designs\.

Finally, we contribute an industry\-aligned benchmark of∼\\sim20,000 executable CAD–CAE tasks across 25 component categories, with standardized interfaces and fixed tool\-call/retry budgets\. Each task provides an initial parametric CAD model, a toolchain configuration, and constraints spanning physics, geometry, and economics \(e\.g\., cost\), enabling reproducible evaluation of feasibility, efficiency \(iterations/tool calls\), and stability \(failure recovery\)\. Using this benchmark, we compare diverse open\-source and proprietary LLMs under a unified interface and fixed budgets\. Results show that COSMO\-Agent training markedly improves an 8B LLM, achieving higher feasibility, efficiency, and stability than most baselines under our protocol\.In summary, our contributions are as follows:

- •formulating closed\-loop CAD–CAE iterative evolution as a long\-horizon sequential decision\-making problem that explicitly models heterogeneous tools, hard executability constraints, and stochastic failure states\.
- •proposing COSMO\-Agent, a tool\-augmented RL framework with a multi\-constraint objective that grounds structured, executable parametric edits in downstream feedback, jointly optimizing feasibility, robustness to tool failures, and output validity\.
- •introducing an industry\-aligned executable CAD–CAE benchmark with standardized interfaces, fixed tool\-call and retry budgets, and constraints \(i\.e\., physics, geometry, and cost\)\.
- •demonstration of improved closed\-loop performance under these controlled budgets\.

## 2Related Work

### 2\.1CAD Model Generation

Learning\-based research on*parametric*CAD spans representations and generation strategies\. SketchGraphs\[[26](https://arxiv.org/html/2605.20190#bib.bib24)\]provides large\-scale constraint graphs from real CAD sketches\. Fusion 360 Gallery\[[34](https://arxiv.org/html/2605.20190#bib.bib25)\]introduces a programmatic CAD language with human design sequences and an interactive environment that formulates CAD construction as a sequential decision process\. JoinABLe\[[33](https://arxiv.org/html/2605.20190#bib.bib26)\]extends learning to CAD assemblies by releasing weakly supervised joint annotations\.

Recent work also leverages large models to synthesize or manipulate CAD*programs*\. LLM4CAD\[[16](https://arxiv.org/html/2605.20190#bib.bib39)\]studies multimodal LLMs for generating CAD programs from text and images, while Text\-to\-CadQuery\[[35](https://arxiv.org/html/2605.20190#bib.bib40)\]directly generates CadQuery code and improves executability and geometric quality via supervision and fine\-tuning\. OpenECAD\[[39](https://arxiv.org/html/2605.20190#bib.bib27)\]enables editable CAD through structured sketches and executable construction commands, and tool\-augmented agents such as CAD\-Assistant\[[19](https://arxiv.org/html/2605.20190#bib.bib41)\]iteratively execute and repair CAD commands via a CAD API\. Overall, prior systems largely emphasize geometric correctness, editability, or task completion, but rarely incorporate downstream CAE feedback and engineering acceptance constraints into a closed\-loop objective, or treat executability and failure recovery in real CAD pipelines as first\-class optimization targets\.

![Refer to caption](https://arxiv.org/html/2605.20190v1/x1.png)Figure 2:COSMO\-Agent: \(a\) overall closed\-loop framework, \(b\) MCP tool set for CAD–CAE optimization, and \(c\) training reward function\.
### 2\.2LLM Agents for Engineering Simulation

A growing body of work couples LLMs with engineering simulators to automate solver setup and execution\. In CFD, MetaOpenFOAM\[[5](https://arxiv.org/html/2605.20190#bib.bib29)\]adopts a multi\-agent architecture for OpenFOAM workflows, often using retrieval for configuration generation and error correction; CFDagent\[[36](https://arxiv.org/html/2605.20190#bib.bib30)\]similarly decomposes preprocessing, solving, and postprocessing into specialized agents with iterative debugging\. Other efforts build training data and fine\-tuned models for natural language to solver configuration, such as NL2FOAM\[[8](https://arxiv.org/html/2605.20190#bib.bib31)\]; recent end\-to\-end automation continues this direction \(e\.g\., Foam\-Agent\[[40](https://arxiv.org/html/2605.20190#bib.bib32)\]\), and finite\-element workflows have also been explored in ecosystems such as MOOSE\[[42](https://arxiv.org/html/2605.20190#bib.bib33)\]\. These systems demonstrate the promise of “LLM \+ tools” for generating simulation inputs, debugging configurations, and producing results\. However, they largely target completing \(or reproducing\) a simulation instance from a specification\. For multi\-round design optimization—iteratively editing geometry based on simulation outcomes until multiple coupled acceptance constraints are met—learnable closed\-loop strategies remain underexplored, especially under realistic toolchain instability\.

### 2\.3Tool\-augmented LLM Agents

For LLM agents, tool augmentation grounds decisions in executable actions and enables policy updates from observable tool feedback, which is crucial for multi\-step tasks with external dependencies\. ReAct\[[38](https://arxiv.org/html/2605.20190#bib.bib8)\], MRKL\[[14](https://arxiv.org/html/2605.20190#bib.bib35)\], and SayCan\[[1](https://arxiv.org/html/2605.20190#bib.bib34)\]exemplify reasoning interleaved with tool calls, while tool\-use training improves call timing and API\-call fidelity\[[25](https://arxiv.org/html/2605.20190#bib.bib9),[22](https://arxiv.org/html/2605.20190#bib.bib18)\]\.

Beyond prompting, recent work scales long\-horizon training and optimization via verifiable feedback and efficient rollouts\. InternBootcamp\[[15](https://arxiv.org/html/2605.20190#bib.bib36)\]provides verifiable task environments to support scalable RL and evaluation, HybridFlow\[[28](https://arxiv.org/html/2605.20190#bib.bib37)\]improves RLHF system efficiency for multi\-step behaviors, and MARTI\[[41](https://arxiv.org/html/2605.20190#bib.bib38)\]unifies multi\-agent training and inference with multi\-turn rollouts and verifier\-based workflows\. However, these frameworks do not directly resolve closed\-loop CAD–CAE optimization, where agents must produce structured, history\-consistent parametric edits under hard executability constraints and fixed tool\-call/retry budgets while remaining robust to stochastic toolchain failures\. COSMO\-Agent addresses this gap by training an LLM policy with explicit failure states and a multi\-constraint objective grounded in downstream CAE feedback and engineering acceptance constraints\.

## 3Methodology

As shown in Fig\.[2](https://arxiv.org/html/2605.20190#S2.F2), we introduce the COSMO\-Agent from three aspects: 1\) the general framework, 2\) the MCP tool design, and 3\) the reward module\.

### 3\.1General Framework

The general framework of COSMO\-Agent is as shown in Fig\.[2](https://arxiv.org/html/2605.20190#S2.F2)\(a\)\. We construct design requirements, constraints, initial geometric parameters and material parameters given by users into a prompt and input it into the LLM\. We denote each task instance as:

ℐ=\(c,𝐩0,η,δ,γ,κ,ℳ\)\\mathcal\{I\}=\\big\(c,\\mathbf\{p\}\_\{0\},\\eta,\\delta,\\gamma,\\kappa,\\mathcal\{M\}\\big\)\(1\)whereccdenotes the part category,𝐩0∈ℝd\\mathbf\{p\}\_\{0\}\\in\\mathbb\{R\}^\{d\}is the initial geometric parameter vector,η\\etaspecifies the simulation settings \(e\.g\., loads and boundary conditions\),δ\\deltais the maximum displacement threshold,γ\\gammais the maximum allowable von Mises stress threshold,κ\\kappais the cost threshold, andℳ\\mathcal\{M\}is the material library, with the stress limit being material\-dependent and provided by the library, i\.e\.,γ​\(mt\)=σallow​\(mt\)\\gamma\(m\_\{t\}\)=\\sigma\_\{\\text\{allow\}\}\(m\_\{t\}\)\. The LLM then formulates a round\-by\-round updated design plan\{\(𝐩t,mt\)\}t=0T\\\{\(\\mathbf\{p\}\_\{t\},m\_\{t\}\)\\\}\_\{t=0\}^\{T\}based on the input prompt\. At each roundtt, the design state is determined by geometric parameters𝐩t\\mathbf\{p\}\_\{t\}and a material choicemt∈ℳm\_\{t\}\\in\\mathcal\{M\}\. We then feed the design state into the LLM to get the invocation strategy for MCP tool as follow:

xt=\(c,𝐩t,mt\)x\_\{t\}=\(c,\\mathbf\{p\}\_\{t\},m\_\{t\}\)\(2\)The planner then invokes the MCP Tools\. These MCP Tools generate three\-dimensional CAD design files, and the CAE solver performs simulation solving and calculation in accordance with the specified conditions\. The MCP tool returns, under simulation settingη\\eta, a scalar feedback tuple:

Φ​\(xt;η\)=\(umax\(t\),σmax\(t\),C\(t\)\)\\Phi\(x\_\{t\};\\eta\)=\\big\(u\_\{\\max\}^\{\(t\)\},\\sigma\_\{\\max\}^\{\(t\)\},C^\{\(t\)\}\\big\)\(3\)whereumax\(t\)u\_\{\\max\}^\{\(t\)\}is the maximum displacement magnitude,σmax\(t\)\\sigma\_\{\\max\}^\{\(t\)\}is the maximum von Mises equivalent stress, andC\(t\)C^\{\(t\)\}is the cost metric\. The results are then fed back to the LLM for verifying whether they meet the design requirements\. Design feasibility is defined by the following constraints:

umax\(t\)≤δ,σmax\(t\)≤σallow​\(mt\),C\(t\)≤κu\_\{\\max\}^\{\(t\)\}\\leq\\delta,\\qquad\\sigma\_\{\\max\}^\{\(t\)\}\\leq\\sigma\_\{\\text\{allow\}\}\(m\_\{t\}\),\\qquad C^\{\(t\)\}\\leq\\kappa\(4\)whereσallow​\(mt\)\\sigma\_\{\\text\{allow\}\}\(m\_\{t\}\)denotes the allowable stress of materialmtm\_\{t\}from the material library\. If the requirements are not met, the planner re\-initiates another CAD–CAE iteration\. At each round, it records the user input and interaction history in memory, evaluates constraints using the latestΦ​\(xt;η\)\\Phi\(x\_\{t\};\\eta\), and outputs updated design parameters and material choice\(𝐩t\+1,mt\+1\)\(\\mathbf\{p\}\_\{t\+1\},m\_\{t\+1\}\)\. The update is guided by numerical feedback: displacement and stress are parsed from CAE results, and cost is computed from geometry and material properties\. The model then decides the next modification toward satisfying all constraints and triggers the next CAD generation and CAE verification\. Once all constraints are satisfied at some roundt∗t^\{\*\}, it returns the final design parameters/material configuration and terminates; otherwise, it proceeds until reaching the maximum number of rounds\.

### 3\.2MCP Tool Set

As shown in Fig\.[2](https://arxiv.org/html/2605.20190#S2.F2)\(b\), we expose the CAD–CAE toolchain via MCP, allowing COSMO\-Agent to trigger external computations and receive outputs through a unified, structured interface\. The tool set covers four stages of the closed\-loop iteration: \(i\) parametric geometry generation, \(ii\) finite element solving, \(iii\) result metric extraction, and \(iv\) cost estimation\.

##### Tool 1: CAD generator\.

The CAD generator maps a part category and a parameter vector to an executable solid geometry consumable by downstream solvers\. Given categoryccand parameters𝐩\\mathbf\{p\}, the tool produces a solidG​\(c,𝐩\)G\(c,\\mathbf\{p\}\)and exports a geometry file path as the primary output\. It also returns geometry metadata for boundary\-condition assignment, such as anchor points on designated functional faces, so that the semantics of loads and constraints remain consistent across different parameter settings\.

##### Tool 2: CAE solver\.

The CAE solver performs physics\-based analysis on the generated geometry\. The tool takes as input the geometry file path, material parameters \(e\.g\., Young’s modulus\), and load/boundary\-condition parameters \(e\.g\., pressure magnitude and fixed constraints\), and outputs a standard result file path together with solver logs\. To enable automated boundary\-condition assignment, anchor points in the geometry metadata are matched to target faces after geometry import\. Let the candidate face set be

ℱ=\{Fj\}\\mathcal\{F\}=\\\{F\_\{j\}\\\}\(5\)and for an anchor point𝐪\\mathbf\{q\}, the point\-to\-face distance isdist⁡\(𝐪,Fj\)\\operatorname\{dist\}\(\\mathbf\{q\},F\_\{j\}\)\. Faces satisfying

dist⁡\(𝐪,Fj\)≤ε\\operatorname\{dist\}\(\\mathbf\{q\},F\_\{j\}\)\\leq\\varepsilon\(6\)within toleranceε\\varepsilonare selected as load faces or constraint faces, which maintains consistent boundary\-condition locations across varying parameterizations\.

##### Tool 3: Result extractor\.

The result extractor converts field outputs in the solver result file into scalar metrics used for constraint checking\. Its input is the result file path produced by the CAE solver, and its outputs are the maximum displacementumaxu\_\{\\max\}and the maximum equivalent stressσmax\\sigma\_\{\\max\}\. For the nodal displacement vector𝐮i=\(ux,i,uy,i,uz,i\)\\mathbf\{u\}\_\{i\}=\(u\_\{x,i\},u\_\{y,i\},u\_\{z,i\}\), the displacement magnitude and maximum displacement are defined as

‖𝐮i‖2=ux,i2\+uy,i2\+uz,i2,umax=maxi⁡‖𝐮i‖2\\\|\\mathbf\{u\}\_\{i\}\\\|\_\{2\}=\\sqrt\{u\_\{x,i\}^\{2\}\+u\_\{y,i\}^\{2\}\+u\_\{z,i\}^\{2\}\},\\\\ u\_\{\\max\}=\\max\_\{i\}\\\|\\mathbf\{u\}\_\{i\}\\\|\_\{2\}\(7\)For the stress components\(σx​x,σy​y,σz​z,τx​y,τy​z,τz​x\)\(\\sigma\_\{xx\},\\sigma\_\{yy\},\\sigma\_\{zz\},\\tau\_\{xy\},\\tau\_\{yz\},\\tau\_\{zx\}\), we define

Δσ\\displaystyle\\Delta\_\{\\sigma\}=\(σx​x−σy​y\)2\+\(σy​y−σz​z\)2\+\(σz​z−σx​x\)2,\\displaystyle=\(\\sigma\_\{xx\}\-\\sigma\_\{yy\}\)^\{2\}\+\(\\sigma\_\{yy\}\-\\sigma\_\{zz\}\)^\{2\}\+\(\\sigma\_\{zz\}\-\\sigma\_\{xx\}\)^\{2\},\(8\)Δτ\\displaystyle\\Delta\_\{\\tau\}=τx​y2\+τy​z2\+τz​x2\\displaystyle=\\tau\_\{xy\}^\{2\}\+\\tau\_\{yz\}^\{2\}\+\\tau\_\{zx\}^\{2\}The von Mises equivalent stress is then computed as

σv=12​Δσ\+3​Δτ\\sigma\_\{v\}=\\sqrt\{\\tfrac\{1\}\{2\}\\Delta\_\{\\sigma\}\+3\\Delta\_\{\\tau\}\}\(9\)and the maximum equivalent stress is computed by

σmax=maxi⁡σv,i\\sigma\_\{\\max\}=\\max\_\{i\}\\sigma\_\{v,i\}\(10\)These outputs provide numerical observations for constraint evaluation and iterative updates\.

##### Tool 4: Cost calculator\.

The cost calculator provides a cost metric associated with geometry and material\. Its inputs include the geometry file path, the material densityρ​\(m\)\\rho\(m\), and the unit mass priceπ​\(m\)\\pi\(m\), and its output is the costCC\. The computation follows a volume–mass–price chain\. Let the solid volume obtained from the geometry beVm3V\_\{\\text\{m\}^\{3\}\}\(inm3\\text\{m\}^\{3\}\)\. The mass is computed as

M=ρ​\(m\)​Vm3M=\\rho\(m\)\\,V\_\{\\text\{m\}^\{3\}\}\(11\)and the final cost is

C=M⋅π​\(m\)C=M\\cdot\\pi\(m\)\(12\)Along with displacement and stress, this cost is used to check the cost constraint and guide subsequent updates\.

### 3\.3Reward Design

We continue training from Qwen3\-8B\[[37](https://arxiv.org/html/2605.20190#bib.bib43)\]so that the model can learn tool\-usage strategies over multi\-round interactions and produce structured outputs that are directly executable\. We adopt Generalized Reinforcement Policy Optimization \(GRPO\)\[[27](https://arxiv.org/html/2605.20190#bib.bib42)\]to optimize the policy, where the reward signal aligns multiple objectives within a single training target, including satisfying engineering constraints, reducing redundant tool invocations, and maintaining consistency between the reported outputs and the executed design state, as illustrated in Fig\.[2](https://arxiv.org/html/2605.20190#S2.F2)\(c\)\.

The reward is derived by parsing the rollout tool\-interaction logs, without performing additional CAE re\-evaluation\. Reward computation relies solely on the tool calls, tool responses recorded in the trajectory, and the model’s final output, thereby ensuring that the training\-time evaluation remains consistent with the actual execution of the toolchain\. The overall reward consists of three components:

R=Rcons\+Rstop\+RjsonR=R\_\{\\text\{cons\}\}\+R\_\{\\text\{stop\}\}\+R\_\{\\text\{json\}\}\(13\)
##### Constraint RewardRconsR\_\{\\text\{cons\}\}\.

From the trajectory log, we extract the metric triple\(umax,σmax,C\)\(u\_\{\\max\},\\sigma\_\{\\max\},C\)for each iteration, whereumaxu\_\{\\max\}andσmax\\sigma\_\{\\max\}are taken from the result extractor outputs andCCis taken from the cost calculator output\. The materialmmand geometric parameters𝐩\\mathbf\{p\}are taken from the most recent design proposal associated with that triple\. We use the last complete triple in the trajectory as the final evaluation target; if no complete triple exists, we setRcons=0R\_\{\\text\{cons\}\}=0\. For the final triple, we define the number of satisfied constraints as

N=𝕀​\[umax≤δ\]\+𝕀​\[C≤κ\]\+𝕀​\[σmax≤σallow​\(m\)\]N=\\mathbb\{I\}\[u\_\{\\max\}\\leq\\delta\]\+\\mathbb\{I\}\[C\\leq\\kappa\]\+\\mathbb\{I\}\[\\sigma\_\{\\max\}\\leq\\sigma\_\{\\text\{allow\}\}\(m\)\]\(14\)The constraint reward is defined in a piecewise manner:

Rcons=\{0\.00,N=00\.20,N=10\.50,N=21\.00,N=3R\_\{\\text\{cons\}\}=\\begin\{cases\}0\.00,&N=0\\\\ 0\.20,&N=1\\\\ 0\.50,&N=2\\\\ 1\.00,&N=3\\end\{cases\}\(15\)

##### Feasible\-Then\-Stop TermRstopR\_\{\\text\{stop\}\}\.

We locate the first time step at which a complete triple satisfying all three constraints appears in the trajectory, and denote its time index bytfeast\_\{\\text\{feas\}\}\. LetKKbe the number of tool events aftertfeast\_\{\\text\{feas\}\}\(including tool calls and tool responses\)\. We define

Rstop=−min⁡\(λ​K,λmax\)R\_\{\\text\{stop\}\}=\-\\min\(\\lambda K,\\lambda\_\{\\max\}\)\(16\)whereλ=0\.02\\lambda=0\.02andλmax=0\.10\\lambda\_\{\\max\}=0\.10\. If no complete triple satisfying all constraints appears in the trajectory, we setRstop=0R\_\{\\text\{stop\}\}=0\.

##### Structured\-Output Consistency TermRfmtR\_\{\\text\{fmt\}\}\.

To support downstream CAD generation and simulation reproducibility, the final output is required to be a structured JSON object \(including category, material, and geometric parameters\)\. If the final output contains a parsable JSON whose category/material/parameters are consistent with the design proposal that produced the final triple, we assign

Rfmt=0\.10R\_\{\\text\{fmt\}\}=0\.10\(17\)otherwise, we set

Rfmt=0R\_\{\\text\{fmt\}\}=0\(18\)

## 4Dataset Annotation

To better support the CAD–CAE closed\-loop optimization, we construct an industrial component dataset\. The dataset covers 25 common part categories in industrial design, including flat plate flanges, triangular brackets, hex thin nuts, and I\-beam cantilever beams\. In terms of scale, it contains 20,000 training samples, 200 test samples, and 100 generalization samples\. Among the 25 categories, 20 are used for training and testing, while the remaining 5 are reserved for generalization evaluation\. The generalization set is designed to assess the model’s transfer ability to unseen geometric templates and parameter semantics\.

The geometric data are generated from parametric templates\. For each part category, we implement a CadQuery\[[7](https://arxiv.org/html/2605.20190#bib.bib44)\]template that produces a STEP file given a set of geometric parameters\. We then perform finite element analysis under the specified material and boundary/loading conditions, and extract the maximum displacement magnitudeumaxu\_\{\\max\}and the maximum von Mises equivalent stressσmax\\sigma\_\{\\max\}\. The cost metricCCis computed from the geometry volume together with the material density and unit price\. Material properties are provided by a unified material libraryℳ\\mathcal\{M\}, includingE,ν,ρ,π,E,\\nu,\\rho,\\pi,andσallow\\sigma\_\{\\text\{allow\}\}, summarized in Table[1](https://arxiv.org/html/2605.20190#S4.T1)\. In particular,σallow\\sigma\_\{\\text\{allow\}\}is directly used as the upper bound in stress constraint checking\.

To construct optimization\-oriented targets, we annotate the constraint thresholds based on the ground\-truth metrics obtained from simulation\. For displacement and cost, we adopt a randomized reduction strategy: a standard reduction of 5%–10%, and an extreme reduction of 30% \(accounting for 10% of the dataset\)\. We further vary the difficulty by applying the reduction to one, two, or all three constraints \(“reduce 1 / 2 / 3 items”\), thereby producing a diverse set of constraint combinations\. After threshold generation, we perform a feasibility check to ensure that the target thresholds remain within the feasible range for the corresponding category, avoiding trivially infeasible targets\.

Beyond numerical annotations, we also construct prompts to drive interactive optimization\. Each prompt is formed by concatenating four parts in order: background and objectives \(Part 1\), initial design and boundary/loading description \(Part 2\), material library, tool usage rules, and termination conditions \(Part 3\), and the fixed JSON output requirement \(Part 4\)\. Parts 1 and 3 each include 10 stylistically different variants to increase linguistic diversity, while Parts 2 and 4 are customized for each category to ensure strict consistency in parameter semantics, boundary conditions, and JSON field definitions\. Finally, the model is required to output a single parsable JSON object containing the optimized geometric parameters and the selected material name, enabling reproducible CAD generation and CAE verification\.

Table 1:Material library used in our CAD–CAE toolchain\. For each material,EEdenotes Young’s modulus in MPa,ν\\nudenotes Poisson’s ratio,ρ\\rhodenotes mass density in kg/m3,π\\pidenotes unit mass price in ¥/kg, andσallow\\sigma\_\{\\text\{allow\}\}denotes the allowable stress in MPa\.
## 5Experiments

### 5\.1Experiment Settings

#### 5\.1\.1Implementation Details

COSMO\-Agent is built on the Qwen3\-8B backbone and trained with the Internbootcamp\[[15](https://arxiv.org/html/2605.20190#bib.bib36)\]framework for multi\-turn interactive rollouts and policy updates on 16×\\timesH200 \(144GB\) GPUs\. We use GRPO during training: for each prompt, we sample 8 rollout trajectories with temperature=1\.0 and top\-pp=0\.9; each instance allows up to 15 assistant turns to cover multi\-round iterations in the CAD–CAE closed loop\. For optimization, we set the actor learning rate to1×10−61\\times 10^\{\-6\}, adopt GRPO clipping with clip ratio low=0\.2 and high=0\.28, and apply gradient clipping with a norm of 1\.0\. The training batch size is 8 prompts per step, and dynamic batching is enabled to accommodate length variability from long contexts and multi\-turn interactions\. We also enable KL regularization to constrain policy deviation, with the KL coefficient set to 0\.001\. We choose CADQuery library as the CAD generator\. The CAE solver performs finite element analysis using the FreeCAD\[[24](https://arxiv.org/html/2605.20190#bib.bib45)\]FEM backend, where meshing is carried out by Gmsh\[[9](https://arxiv.org/html/2605.20190#bib.bib46)\]and linear static solving is performed by CalculiX\.

#### 5\.1\.2Compared Methods

We compare COSMO\-Agent with language models of different scales, covering both state\-of\-the\-art open\-source and closed\-source systems\. All models are evaluated under the same task inputs, the same CAD–CAE toolchain, the same maximum interaction\-turn limit, and a unified JSON output specification\. The open\-source baselines include Qwen3\-8B\[[37](https://arxiv.org/html/2605.20190#bib.bib43)\], Intern\-S1\-mini\[[4](https://arxiv.org/html/2605.20190#bib.bib47)\], Llama\-4\-Scout\[[20](https://arxiv.org/html/2605.20190#bib.bib50)\], Qwen3\-30B\[[30](https://arxiv.org/html/2605.20190#bib.bib48)\], Qwen3\-Next\[[30](https://arxiv.org/html/2605.20190#bib.bib48)\], and Intern\-S1\[[4](https://arxiv.org/html/2605.20190#bib.bib47)\]\. The closed\-source baselines include Claude\-Sonnet\-4\.5\[[2](https://arxiv.org/html/2605.20190#bib.bib51)\]and Gemini\-3\-Flash\[[10](https://arxiv.org/html/2605.20190#bib.bib49)\]\.

#### 5\.1\.3Evaluation Metrics

We evaluate all models in terms of task success and efficiency\. For each instance, the model’s final JSON output is parsed into executable geometric parameters and a material configuration\. We then reproduce the result with the same CAD–CAE toolchain to obtain the maximum displacement magnitudeumaxu\_\{\\max\}, the maximum equivalent stressσmax\\sigma\_\{\\max\}, and the costCC, based on which we compute the following metrics:

∙\\bulletFull Success Rate\(FSR\): the proportion of instances that satisfy all three constraints \(displacement, stress, and cost\)\. ∙\\bulletDisplacement Satisfaction Rate\(DSR\): the proportion of instances withumax≤δu\_\{\\max\}\\leq\\delta\. ∙\\bulletStress Satisfaction Rate\(SSR\): the proportion of instances withσmax≤σallow​\(m\)\\sigma\_\{\\max\}\\leq\\sigma\_\{\\text\{allow\}\}\(m\)\. ∙\\bulletCost Satisfaction Rate\(CSR\): the proportion of instances withC≤κC\\leq\\kappa\. ∙\\bulletModel Extract Output\(MEO\): the proportion of instances for which a valid and parsable final JSON can be successfully extracted from the model output\. ∙\\bulletAverage Score\(AS\): the average per\-instance composite score, consisting of \(i\) success/failure signals from tool responses, \(ii\) whether a valid structured JSON can be extracted, and \(iii\) the number of satisfied constraints\. ∙\\bulletAvg Tool Calls\(ATC\): the average number of tool invocations per instance during inference\.

### 5\.2Main Results

As shown in Table[2](https://arxiv.org/html/2605.20190#S5.T2), COSMO\-Agent \(8B\) achieves an FSR of 74\.5%, which is the best among all methods\. Compared with the strongest open\-source baseline intern\-S1 \(32\.0%\), it improves by 42\.5%\. Compared with the best closed\-source baseline Gemini\-3\-Flash \(67\.5%\), it improves by 7\.0%\. This indicates that COSMO\-Agent can more reliably produce feasible solutions that simultaneously satisfy the displacement, stress, and cost constraints\. In terms of constraint satisfaction, COSMO\-Agent achieves a displacement satisfaction rate of 87\.5%, a stress satisfaction rate of 76\.0%, and a cost satisfaction rate of 93\.5%, which are overall the best\. For structured outputs, COSMO\-Agent reaches an MEO of 100%, ensuring that the final results can be parsed into an executable JSON and thus improving the reliability of end\-to\-end reproduction\. From the optimization process perspective, COSMO\-Agent obtains an AS of 0\.6504, slightly lower than Gemini’s 0\.6802\. Since AS includes tool\-call rewards, Gemini uses more tool calls per successful case on average \(9\.32\), making it easier to accumulate process scores\. In contrast, COSMO\-Agent has an ATC of 6\.72, indicating that it requires fewer tool calls to reach feasible solutions and is more interaction\-efficient\. Overall, COSMO\-Agent achieves high success rates while maintaining good inference efficiency\.

Table 2:Main results on the test set\. Abbreviations:FSR\(Full Success Rate\),DSR\(Displacement Satisfaction Rate\),SSR\(Stress Satisfaction Rate\),CSR\(Cost Satisfaction Rate\),MEO\(Model Extract Output\),AS\(Average Score\),ATC\(Avg Tool Calls\)\.Table 3:Generalization results on the unseen\-category set\.
### 5\.3Generalization Performance

As shown in Table[3](https://arxiv.org/html/2605.20190#S5.T3), on the generalization set consisting of five unseen categories, COSMO\-Agent achieves an FSR of 75\.0%, which is broadly consistent with the main\-test result \(74\.5%\), indicating no obvious degradation on unseen templates\. Compared with the baselines, COSMO\-Agent remains significantly ahead: among open\-source methods, intern\-S1 achieves 35\.0%; among closed\-source methods, Gemini\-3\-Flash achieves 57\.0% and Claude\-Sonnet\-4\.5 achieves 38\.0%\.

On the generalization set, COSMO\-Agent still maintains a 100% format extraction success rate, ensuring stable end\-to\-end reproducibility\. In contrast, Gemini\-3\-Flash has a format extraction success rate of only 60\.0%, which directly reduces the effective end\-to\-end success proportion in evaluation\. From the optimization process perspective, COSMO\-Agent obtains an AS of 0\.6150, which is lower than Gemini \(0\.6977\) and Claude \(0\.6468\)\. This difference is consistent with the AS scoring rule: AS accumulates tool\-call rewards, and Gemini and Claude use more tool calls per successful case on average \(9\.44 and 11\.08, respectively\) than COSMO\-Agent \(6\.57\), making it easier to obtain higher process scores\. Nevertheless, on the more critical end\-to\-end metric FSR, COSMO\-Agent still maintains a clear advantage and also demonstrates high interaction efficiency on unseen categories\.

![Refer to caption](https://arxiv.org/html/2605.20190v1/x2.png)Figure 3:Visualized inference cases of COSMO\-Agent\.
### 5\.4Visualization Result

Fig\.[3](https://arxiv.org/html/2605.20190#S5.F3)illustratesCOSMO\-Agent’s dynamic reasoning through a structure–material co\-optimization of a sintered flange bushing\. To meet the cost constraint, it first switches the material to Carbon Steel–ASTM A105 and proposes a reduced\-size geometry, then iteratively invokes CAD generation, CAE simulation, result extraction, and cost calculation in a closed\-loop “generation–simulation–evaluation” workflow\. The first evaluation shows stress meeting the strength constraint, but displacement \(87\.25μ\\mum\) slightly exceeding the stiffness limit \(80\.21μ\\mum\), while cost \(¥36\.19\) remains within budget\.COSMO\-Agenttherefore improves stiffness via minor geometry updates and re\-verifies, and the final design satisfies strength, stiffness, and cost constraints\.

### 5\.5Ablation Studies

We conduct ablation studies to identify the key factors behind the performance gains of COSMO\-Agent: \(i\) whether GRPO\-based RL training is applied, and \(ii\) whether the reward is computed from toolchain rollout logs\. All other settings are kept identical, and the results are summarized in Table[4](https://arxiv.org/html/2605.20190#S5.T4)\.

##### Effect of RL training\.

Without RL \(w/o RL\), the model achieves an FSR of 26\.0%\. With RL training and the full reward, the FSR increases to 74\.5%, improving by 48\.5 percentage points\. The displacement satisfaction rate improves from 39\.5% to 87\.5%, the cost satisfaction rate improves from 65\.0% to 93\.5%, and the stress satisfaction rate improves from 72\.0% to 76\.0%\. These results indicate that RL training substantially strengthens the model’s ability to update parameters based on numerical feedback in closed\-loop interactions\.

##### Effect of rollout\-log\-based reward\.

We compare our rollout\-log\-based reward with a “final\-JSON re\-verification” baseline, where the reward is computed by parsing the model’s final JSON output, re\-running the CAD–CAE toolchain once with the reported\(𝐩,m\)\(\\mathbf\{p\},m\), and then scoring feasibility based on the re\-simulated metrics \(rather than using the rollout interaction logs\)\. Replacing our reward with this baseline yields an FSR of 36\.0%, which is much lower than the full COSMO\-Agent \(74\.5%\)\. Under this setting, the average tool calls drop to 2\.62\. By inspecting rollout logs, we find that the model tends to avoid calling tools and directly outputs a guessed JSON solution, weakening closed\-loop optimization\. In contrast, our rollout\-log\-based reward directly parses tool\-interaction logs and uses the last complete metric triple to compute returns, avoiding the high cost of re\-simulation and more effectively encouraging the model to follow the “call tools–read feedback–iterate” loop\.

Table 4:Ablation studies on the test set\. “w/o RL” disables GRPO training\. “w/o Rollout Reward” replaces the rollout\-log\-based reward with a final\-JSON re\-verification reward\.

## 6Conclusion

We presented COSMO\-Agent, a tool\-augmented reinforcement learning framework for reliable closed\-loop CAD–CAE iteration\. By modeling the CAD–CAE pipeline as an interactive environment and training an 8B LLM with GRPO and a rollout\-log\-based reward, COSMO\-Agent learns to generate structured, executable parametric edits and improve designs through multi\-round tool feedback\. The rollout\-log reward leverages tool execution logs to encourage proper tool use and constraint\-driven optimization without requiring additional expensive re\-simulations\. Experiments show that COSMO\-Agent significantly improves feasibility and interaction efficiency under fixed tool\-call and retry budgets, and it generalizes well to unseen component categories\.

In future work, we will extend COSMO\-Agent to richer design settings with contact, assembly, and multi\-part constraints, and to additional physics such as nonlinear materials and coupled multi\-physics\. We also plan to support alternative CAD and CAE backends and study scalability under larger action spaces, tighter budgets, and more diverse failure modes\. Finally, we will explore improved training curricula and robustness objectives to further strengthen long\-horizon reliability in practical pipelines\.

## References

- \[1\]M\. Ahn, A\. Brohan, N\. Brown, Y\. Chebotar, O\. Cortes, B\. David, C\. Finn, C\. Fu, K\. Gopalakrishnan, K\. Hausman,et al\.\(2022\)Do as I can, not as I say: grounding language in robotic affordances\.InConference on Robot Learning, CoRL 2022, 14\-18 December 2022, Auckland, New Zealand,Proceedings of Machine Learning Research, Vol\.205,pp\. 287–318\.Cited by:[§1](https://arxiv.org/html/2605.20190#S1.p2.1),[§2\.3](https://arxiv.org/html/2605.20190#S2.SS3.p1.1)\.
- \[2\]Anthropic\(2025\)Claude Sonnet 4\.5: System Card\.System cardAnthropic PBC\.External Links:[Link](https://assets.anthropic.com/m/12f214efcc2f457a/original/Claude-Sonnet-4-5-System-Card.pdf)Cited by:[§5\.1\.2](https://arxiv.org/html/2605.20190#S5.SS1.SSS2.p1.1)\.
- \[3\]C\. Audet and J\. E\. Dennis\(2006\)Mesh adaptive direct search algorithms for constrained optimization\.SIAM Journal on Optimization17\(1\),pp\. 188–217\.External Links:[Document](https://dx.doi.org/10.1137/040603371)Cited by:[§1](https://arxiv.org/html/2605.20190#S1.p2.1)\.
- \[4\]L\. Bai, Z\. Cai, M\. Cao, W\. Cao, C\. Chen, H\. Chen, K\. Chen, P\. Chen, Y\. Chen, Y\. Chen, Y\. Cheng, Y\. Cheng, P\. Chu, T\. Chu, E\. Cui, G\. Cui, L\. Cui, Z\. Cui, N\. Deng, N\. Ding, N\. Dong, P\. Dong, S\. Dou, S\. Du, H\. Duan, C\. Fan, B\. Gao, C\. Gao, J\. Gao, S\. Gao, Y\. Gao, Z\. Gao, J\. Ge, Q\. Ge, L\. Gu, Y\. Gu, A\. Guo, Q\. Guo, X\. Guo, C\. He, J\. He, Y\. Hong, S\. Hou, C\. Hu, H\. Hu, J\. Hu, M\. Hu, Z\. Hua, H\. Huang, J\. Huang, X\. Huang, Z\. Huang, Z\. Jiang, L\. Kong, L\. Li, P\. Li, P\. Li, S\. Li, T\. Li, W\. Li, Y\. Li, D\. Lin, J\. Lin, T\. Lin, Z\. Lin, H\. Liu, J\. Liu, J\. Liu, J\. Liu, K\. Liu, K\. Liu, K\. Liu, S\. Liu, S\. Liu, W\. Liu, X\. Liu, Y\. Liu, Z\. Liu, Y\. Lu, H\. Lv, H\. Lv, H\. Lv, Q\. Lv, Y\. Lv, C\. Lyu, C\. Ma, J\. Ma, R\. Ma, R\. Ma, R\. Ma, X\. Ma, Y\. Ma, Z\. Ma, S\. Mi, J\. Ning, W\. Ning, X\. Pang, J\. Peng, R\. Peng, Y\. Qiao, J\. Qiu, X\. Qu, Y\. Qu, Y\. Ren, F\. Shang, W\. Shao, J\. Shen, S\. Shen, C\. Song, D\. Song, D\. Song, C\. Su, W\. Su, W\. Sun, Y\. Sun, Q\. Tan, C\. Tang, H\. Tang, K\. Tang, S\. Tang, J\. Tong, A\. Wang, B\. Wang, D\. Wang, L\. Wang, R\. Wang, W\. Wang, W\. Wang, Y\. Wang, Z\. Wang, L\. Wu, W\. Wu, Y\. Wu, Z\. Wu, L\. Xiao, S\. Xing, C\. Xu, H\. Xu, J\. Xu, R\. Xu, W\. Xu, G\. Yang, Y\. Yang, H\. Ye, J\. Ye, S\. Ye, J\. Yu, J\. Yu, J\. Yu, F\. Yuan, B\. Zhang, C\. Zhang, C\. Zhang, H\. Zhang, J\. Zhang, Q\. Zhang, Q\. Zhang, S\. Zhang, T\. Zhang, W\. Zhang, W\. Zhang, Y\. Zhang, Z\. Zhang, H\. Zhao, Q\. Zhao, X\. Zhao, X\. Zhao, B\. Zhou, D\. Zhou, P\. Zhou, Y\. Zhou, Y\. Zhou, D\. Zhu, L\. Zhu, and Y\. Zou\(2025\)Intern\-s1: a scientific multimodal foundation model\.External Links:2508\.15763,[Link](https://arxiv.org/abs/2508.15763)Cited by:[§5\.1\.2](https://arxiv.org/html/2605.20190#S5.SS1.SSS2.p1.1)\.
- \[5\]Y\. Chen, X\. Zhu, H\. Zhou, and Z\. Ren\(2024\)MetaOpenFOAM: an llm\-based multi\-agent framework for cfd\.arXiv preprint arXiv:2407\.21320\.Cited by:[§2\.2](https://arxiv.org/html/2605.20190#S2.SS2.p1.1)\.
- \[6\]P\. F\. Christiano, J\. Leike, T\. B\. Brown, M\. Martic, S\. Legg, and D\. Amodei\(2017\)Completelyinforcement learning from human preferences\.InAdvances in Neural Information Processing Systems,pp\. 4299–4307\.Cited by:[§1](https://arxiv.org/html/2605.20190#S1.p2.1)\.
- \[7\]CadQueryExternal Links:[Document](https://dx.doi.org/10.5281/zenodo.14590990),[Link](https://doi.org/10.5281/zenodo.14590990)Cited by:[§4](https://arxiv.org/html/2605.20190#S4.p2.7)\.
- \[8\]Z\. Dong, Z\. Lu, and Y\. Yang\(2025\)Fine\-tuning a large language model for automating computational fluid dynamics simulations\.Theoretical and Applied Mechanics Letters,pp\. 100594\.Cited by:[§2\.2](https://arxiv.org/html/2605.20190#S2.SS2.p1.1)\.
- \[9\]C\. Geuzaine and J\. Remacle\(2009\)Gmsh: a 3\-d finite element mesh generator with built\-in pre\-and post\-processing facilities\.International journal for numerical methods in engineering79\(11\),pp\. 1309–1331\.Cited by:[§5\.1\.1](https://arxiv.org/html/2605.20190#S5.SS1.SSS1.p1.3)\.
- \[10\]Google DeepMind\(2025\-12\)Gemini 3 flash model card\.Google DeepMind\.Note:Model card for the Gemini 3 Flash generative AI modelOnline PDFExternal Links:[Link](https://storage.googleapis.com/deepmind-media/Model-Cards/Gemini-3-Flash-Model-Card.pdf)Cited by:[§5\.1\.2](https://arxiv.org/html/2605.20190#S5.SS1.SSS2.p1.1)\.
- \[11\]N\. Hansen and A\. Ostermeier\(2001\)Completely derandomized self\-adaptation in evolution strategies\.Evolutionary Computation9\(2\),pp\. 159–195\.External Links:[Document](https://dx.doi.org/10.1162/106365601750190398)Cited by:[§1](https://arxiv.org/html/2605.20190#S1.p2.1)\.
- \[12\]Y\. Hu, L\. Anderson, T\. Li, Q\. Sun, N\. Carr, J\. Ragan\-Kelley, and F\. Durand\(2020\)DiffTaichi: differentiable programming for physical simulation\.InInternational Conference on Learning Representations,External Links:[Link](https://openreview.net/forum?id=B1eB5xSFvr)Cited by:[§1](https://arxiv.org/html/2605.20190#S1.p2.1)\.
- \[13\]D\. R\. Jones, M\. Schonlau, and W\. J\. Welch\(1998\)Efficient global optimization of expensive black\-box functions\.Journal of Global Optimization13\(4\),pp\. 455–492\.External Links:[Document](https://dx.doi.org/10.1023/A%3A1008306431147)Cited by:[§1](https://arxiv.org/html/2605.20190#S1.p2.1)\.
- \[14\]E\. Karpas, O\. Abend, Y\. Belinkov, B\. Lenz, O\. Lieber, N\. Ratner, Y\. Shoham, H\. Bata, Y\. Levine, K\. Leyton\-Brown,et al\.\(2022\)MRKL systems: a modular, neuro\-symbolic architecture that combines large language models, external knowledge sources and discrete reasoning\.arXiv preprint arXiv:2205\.00445\.Cited by:[§2\.3](https://arxiv.org/html/2605.20190#S2.SS3.p1.1)\.
- \[15\]P\. Li, J\. Ye, Y\. Chen, Y\. Ma, Z\. Yu, K\. Chen, G\. Cui, H\. Li, J\. Chen, C\. Lyu, W\. Zhang, L\. Li, Q\. Guo, D\. Lin, B\. Zhou, and K\. Chen\(2025\)InternBootcamp technical report: boosting llm reasoning with verifiable task scaling\.External Links:2508\.08636,[Link](https://arxiv.org/abs/2508.08636)Cited by:[§2\.3](https://arxiv.org/html/2605.20190#S2.SS3.p2.1),[§5\.1\.1](https://arxiv.org/html/2605.20190#S5.SS1.SSS1.p1.3)\.
- \[16\]X\. Li, Y\. Sun, and Z\. Sha\(2024\)LLM4CAD: multi\-modal large language models for 3d computer\-aided design generation\.InInternational Design Engineering Technical Conferences and Computers and Information in Engineering Conference,Vol\.88407,pp\. V006T06A015\.Cited by:[§2\.1](https://arxiv.org/html/2605.20190#S2.SS1.p2.1)\.
- \[17\]Z\. Li, N\. B\. Kovachki, K\. Azizzadenesheli, B\. Liu, K\. Bhattacharya, A\. Stuart, and A\. Anandkumar\(2021\)Fourier neural operator for parametric partial differential equations\.InInternational Conference on Learning Representations,External Links:[Link](https://openreview.net/forum?id=c8P9NQVtmnO)Cited by:[§1](https://arxiv.org/html/2605.20190#S1.p2.1)\.
- \[18\]L\. Lu, P\. Jin, G\. Pang, Z\. Zhang, and G\. E\. Karniadakis\(2021\)Learning nonlinear operators via DeepONet based on the universal approximation theorem of operators\.Nature Machine Intelligence3,pp\. 218–229\.External Links:[Document](https://dx.doi.org/10.1038/s42256-021-00302-5)Cited by:[§1](https://arxiv.org/html/2605.20190#S1.p2.1)\.
- \[19\]D\. Mallis, A\. S\. Karadeniz, S\. Cavada, D\. Rukhovich, N\. Foteinopoulou, K\. Cherenkova, A\. Kacem, and D\. Aouada\(2025\)CAD\-assistant: tool\-augmented vllms as generic cad task solvers\.InProceedings of the IEEE/CVF International Conference on Computer Vision,pp\. 7284–7294\.Cited by:[§2\.1](https://arxiv.org/html/2605.20190#S2.SS1.p2.1)\.
- \[20\]Meta\(2025\-04\)Llama 4 Scout \(17B×\\times16E\) Instruct: Model Card\.Note:Online model cardModel release date: April 5, 2025\. Accessed: 2026\-01\-23External Links:[Link](https://build.nvidia.com/meta/llama-4-scout-17b-16e-instruct/modelcard)Cited by:[§5\.1\.2](https://arxiv.org/html/2605.20190#S5.SS1.SSS2.p1.1)\.
- \[21\]L\. Ouyang, J\. Wu, X\. Jiang, D\. Almeida, C\. L\. Wainwright, P\. Mishkin, C\. Zhang, S\. Agarwal, K\. Slama, A\. Ray,et al\.\(2022\)Training language models to follow instructions with human feedback\.InAdvances in Neural Information Processing Systems,External Links:[Link](https://proceedings.neurips.cc/paper_files/paper/2022/file/b1efde53be364a73914f58805a001731-Paper-Conference.pdf)Cited by:[§1](https://arxiv.org/html/2605.20190#S1.p2.1)\.
- \[22\]Y\. Qin, S\. Liang, Y\. Ye, K\. Zhu, L\. Yan, Y\. Lu, Y\. Lin, X\. Cong, X\. Tang, B\. Qian, S\. Zhao, L\. Hong, R\. Tian, R\. Xie, J\. Zhou, M\. Gerstein, D\. Li, Z\. Liu, and M\. Sun\(2024\)ToolLLM: facilitating large language models to master 16000\+ real\-world apis\.InInternational Conference on Learning Representations,External Links:[Link](https://openreview.net/forum?id=dHng2O0Jjr)Cited by:[§1](https://arxiv.org/html/2605.20190#S1.p2.1),[§2\.3](https://arxiv.org/html/2605.20190#S2.SS3.p1.1)\.
- \[23\]M\. Raissi, P\. Perdikaris, and G\. E\. Karniadakis\(2019\)Physics\-informed neural networks: A deep learning framework for solving forward and inverse problems involving nonlinear partial differential equations\.Journal of Computational Physics378,pp\. 686–707\.External Links:[Document](https://dx.doi.org/10.1016/j.jcp.2018.10.045)Cited by:[§1](https://arxiv.org/html/2605.20190#S1.p2.1)\.
- \[24\]\(Software\)FreeCADNote:Accessed: 2001–2017External Links:[Link](http://www.freecadweb.org/)Cited by:[§5\.1\.1](https://arxiv.org/html/2605.20190#S5.SS1.SSS1.p1.3)\.
- \[25\]T\. Schick, J\. Dwivedi\-Yu, R\. Dessì, R\. Raileanu, M\. Lomeli, L\. Zettlemoyer, N\. Cancedda, and T\. Scialom\(2023\)Toolformer: language models can teach themselves to use tools\.InAdvances in Neural Information Processing Systems \(NeurIPS\),Vol\.36,pp\. 68539–68551\.Cited by:[§1](https://arxiv.org/html/2605.20190#S1.p2.1),[§2\.3](https://arxiv.org/html/2605.20190#S2.SS3.p1.1)\.
- \[26\]A\. Seff, Y\. Ovadia, W\. Zhou, and R\. P\. Adams\(2020\)SketchGraphs: a large\-scale dataset for modeling relational geometry in computer\-aided design\.External Links:2007\.08506,[Document](https://dx.doi.org/10.48550/arXiv.2007.08506)Cited by:[§2\.1](https://arxiv.org/html/2605.20190#S2.SS1.p1.1)\.
- \[27\]Z\. Shao, P\. Wang, Q\. Zhu, R\. Xu, J\. Song, X\. Bi, H\. Zhang, M\. Zhang, Y\. K\. Li, Y\. Wu, and D\. Guo\(2024\)DeepSeekMath: pushing the limits of mathematical reasoning in open language models\.External Links:2402\.03300,[Link](https://arxiv.org/abs/2402.03300)Cited by:[§3\.3](https://arxiv.org/html/2605.20190#S3.SS3.p1.1)\.
- \[28\]G\. Sheng, C\. Zhang, Z\. Ye, X\. Wu, W\. Zhang, R\. Zhang, Y\. Peng, H\. Lin, and C\. Wu\(2024\)HybridFlow: a flexible and efficient rlhf framework\.arXiv preprint arXiv: 2409\.19256\.Cited by:[§2\.3](https://arxiv.org/html/2605.20190#S2.SS3.p2.1)\.
- \[29\]J\. Snoek, H\. Larochelle, and R\. P\. Adams\(2012\)Practical bayesian optimization of machine learning algorithms\.InAdvances in Neural Information Processing Systems,Vol\.25,pp\. 2960–2968\.Cited by:[§1](https://arxiv.org/html/2605.20190#S1.p2.1)\.
- \[30\]Q\. Team\(2025\)Qwen3 technical report\.External Links:2505\.09388,[Link](https://arxiv.org/abs/2505.09388)Cited by:[§5\.1\.2](https://arxiv.org/html/2605.20190#S5.SS1.SSS2.p1.1)\.
- \[31\]Y\. Wang, Y\. Kordi, S\. Mishra, A\. Liu, N\. A\. Smith, D\. Khashabi, and H\. Hajishirzi\(2023\)Self\-instruct: aligning language models with self\-generated instructions\.InProceedings of the 61st Annual Meeting of the Association for Computational Linguistics \(ACL\),Cited by:[§1](https://arxiv.org/html/2605.20190#S1.p2.1)\.
- \[32\]J\. Wei, X\. Wang, D\. Schuurmans, M\. Bosma, B\. Ichter, F\. Xia, E\. Chi, Q\. V\. Le, and D\. Zhou\(2022\)Chain\-of\-thought prompting elicits reasoning in large language models\.InAdvances in Neural Information Processing Systems \(NeurIPS\),Vol\.35,pp\. 24824–24837\.Cited by:[§1](https://arxiv.org/html/2605.20190#S1.p2.1)\.
- \[33\]K\. D\. D\. Willis, P\. K\. Jayaraman, H\. Chu, Y\. Tian, Y\. Li, D\. Grandi, A\. Sanghi, L\. Tran, J\. G\. Lambourne, A\. Solar\-Lezama, and W\. Matusik\(2022\)JoinABLe: learning bottom\-up assembly of parametric CAD joints\.InIEEE/CVF Conference on Computer Vision and Pattern Recognition, CVPR 2022, New Orleans, LA, USA, June 18\-24, 2022,pp\. 15828–15839\.Cited by:[§2\.1](https://arxiv.org/html/2605.20190#S2.SS1.p1.1)\.
- \[34\]K\. D\. D\. Willis, Y\. Pu, J\. Luo, H\. Chu, T\. Du, J\. G\. Lambourne, A\. Solar\-Lezama, and W\. Matusik\(2021\)Fusion 360 gallery: a dataset and environment for programmatic CAD construction from human design sequences\.ACM Trans\. Graph\.40\(4\),pp\. 54:1–54:24\.Cited by:[§2\.1](https://arxiv.org/html/2605.20190#S2.SS1.p1.1)\.
- \[35\]H\. Xie and F\. Ju\(2025\)Text\-to\-cadquery: a new paradigm for cad generation with scalable large model capabilities\.arXiv preprint arXiv:2505\.06507\.Cited by:[§2\.1](https://arxiv.org/html/2605.20190#S2.SS1.p2.1)\.
- \[36\]Z\. Xu, L\. Wang, C\. Wang, Y\. Chen, Q\. Luo, H\. Yao, S\. Wang, and G\. He\(2025\)CFDAgent: a language\-guided, zero\-shot multi\-agent system for complex flow simulation\.Physics of Fluids37\(11\)\.Cited by:[§2\.2](https://arxiv.org/html/2605.20190#S2.SS2.p1.1)\.
- \[37\]A\. Yang, A\. Li, B\. Yang, B\. Zhang, B\. Hui, B\. Zheng, B\. Yu, C\. Gao, C\. Huang, C\. Lv, C\. Zheng, D\. Liu, F\. Zhou, F\. Huang, F\. Hu, H\. Ge, H\. Wei, H\. Lin, J\. Tang, J\. Yang, J\. Tu, J\. Zhang, J\. Yang, J\. Yang, J\. Zhou, J\. Zhou, J\. Lin, K\. Dang, K\. Bao, K\. Yang, L\. Yu, L\. Deng, M\. Li, M\. Xue, M\. Li, P\. Zhang, P\. Wang, Q\. Zhu, R\. Men, R\. Gao, S\. Liu, S\. Luo, T\. Li, T\. Tang, W\. Yin, X\. Ren, X\. Wang, X\. Zhang, X\. Ren, Y\. Fan, Y\. Su, Y\. Zhang, Y\. Zhang, Y\. Wan, Y\. Liu, Z\. Wang, Z\. Cui, Z\. Zhang, Z\. Zhou, and Z\. Qiu\(2025\)Qwen3 technical report\.External Links:2505\.09388,[Link](https://arxiv.org/abs/2505.09388)Cited by:[§3\.3](https://arxiv.org/html/2605.20190#S3.SS3.p1.1),[§5\.1\.2](https://arxiv.org/html/2605.20190#S5.SS1.SSS2.p1.1)\.
- \[38\]S\. Yao, J\. Zhao, D\. Yu, N\. Du, I\. Shafran, K\. R\. Narasimhan, and Y\. Cao\(2023\)ReAct: synergizing reasoning and acting in language models\.InInternational Conference on Learning Representations \(ICLR\),Cited by:[§1](https://arxiv.org/html/2605.20190#S1.p2.1),[§2\.3](https://arxiv.org/html/2605.20190#S2.SS3.p1.1)\.
- \[39\]Z\. Yuan, J\. Shi, and Y\. Huang\(2024\)OpenECAD: an efficient visual language model for editable 3d\-cad design\.Computers & Graphics124,pp\. 104048\.Cited by:[§2\.1](https://arxiv.org/html/2605.20190#S2.SS1.p2.1)\.
- \[40\]L\. Yue, N\. Somasekharan, Y\. Cao, and S\. Pan\(2025\)Foam\-agent: towards automated intelligent cfd workflows\.arXiv preprint arXiv:2505\.04997\.Cited by:[§2\.2](https://arxiv.org/html/2605.20190#S2.SS2.p1.1)\.
- \[41\]K\. Zhang, R\. Liu, X\. Zhu, K\. Tian, S\. Zeng, G\. Jia, Y\. Fan, X\. Lv, Y\. Zuo, C\. Jiang, Z\. Liu, J\. Wang, Y\. Wang, R\. Zhao, E\. Hua, Y\. Wang, S\. Wang, J\. Gao, X\. Long, Y\. Sun, Z\. Ma, G\. Cui, L\. Bai, N\. Ding, B\. Qi, and B\. Zhou\(2025\)MARTI: a framework for multi\-agent llm systems reinforced training and inference\.Tsinghua University and Shanghai AI Lab\.External Links:[Link](https://github.com/TsinghuaC3I/MARTI)Cited by:[§2\.3](https://arxiv.org/html/2605.20190#S2.SS3.p2.1)\.
- \[42\]T\. Zhang, Z\. Liu, Y\. Xin, and Y\. Jiao\(2025\)MooseAgent: a llm based multi\-agent framework for automating moose simulation\.arXiv preprint arXiv:2504\.08621\.Cited by:[§2\.2](https://arxiv.org/html/2605.20190#S2.SS2.p1.1)\.

Similar Articles

COOPA: A Modular LLM Agent Architecture for Operations Research Problems

arXiv cs.LG

This paper introduces COOPA, a modular LLM agent architecture for operations research problems that combines iterative confidence-based modeling, element-level provenance, and multi-solver routing. Evaluated across eight LLM backbones and four baselines, COOPA achieves the best macro-average accuracy on six backbones and improves over the strongest baseline by up to 6.7 percentage points.

Memory-Augmented Reinforcement Learning Agent for CAD Generation

arXiv cs.AI

This paper proposes a memory-augmented reinforcement learning framework for CAD generation agents that integrates geometric kernel toolchains, dual-track memory, and dynamic utility retrieval to handle complex CAD models with long operation sequences and geometric constraints, achieving improved success rate and geometric consistency.