COSMO-Agent:用强化学习打通CAD-CAE闭环,让LLM自动优化工业设计
工业设计仿真优化长期受困于CAD(计算机辅助设计)与CAE(计算机辅助工程)之间的“语义鸿沟”——仿真反馈难以直接转化为有效的几何修改。近日,arXiv上的一篇新论文提出了 COSMO-Agent(闭环优化、仿真与建模编排),一个基于工具增强强化学习(RL)的框架,旨在教会大语言模型(LLM)自主完成CAD-CAE的闭环流程。
核心思路:将CAD-CAE过程重构为RL环境
COSMO-Agent的核心创新在于将传统的CAD生成、CAE求解、结果解析和几何修正串联成一个交互式强化学习环境。LLM在环境中扮演“智能体”角色,学习调用外部工具(如CAD建模器、CAE求解器),并根据仿真反馈逐步调整参数化几何模型,直到满足所有约束条件。
为了确保学习过程稳定且具备工业实用性,研究团队设计了一种多约束奖励函数,同时鼓励几何可行性、工具链鲁棒性以及结构化输出的有效性。此外,他们还贡献了一个工业对齐的数据集,涵盖25个零部件类别及可执行的CAD-CAE任务,为模型训练和评估提供了真实场景支撑。
实验结果:小模型逆袭,超越GPT-4等闭源模型
实验表明,经过COSMO-Agent训练后,小型开源LLM在约束驱动设计任务上取得了显著提升。在可行性、效率和稳定性三个关键指标上,这些经过微调的小模型不仅超越了同级别开源模型,甚至超过了大型开源模型(如Llama-3 70B)和强大的闭源模型(如GPT-4)。这一结果令人惊讶,也印证了“工具增强+强化学习”策略对于弥补LLM在专业领域推理能力不足的潜力。
行业意义:加速工业设计仿真迭代
传统设计仿真优化依赖工程师手动解读CAE结果并修改CAD模型,过程繁琐且耗时。COSMO-Agent通过自动化解耦这一闭环,有望大幅缩短产品开发周期,尤其适用于需要反复迭代的零部件设计场景。不过,论文目前仅展示了在特定数据集上的效果,实际工业部署还需考虑模型泛化性、计算资源消耗以及与现有软件工具的集成问题。
总体而言,COSMO-Agent为LLM在工程领域的落地提供了一个新范式:不追求模型本身全能,而是通过强化学习让模型学会调用专业工具。这一思路对于解决AI在垂直行业的“最后一公里”问题具有重要参考价值。
