中文

BlenderLLM:通过自我改进训练用于计算机辅助设计的大语言模型

人机交互 2024-12-20 v1 人工智能

摘要

尽管大语言模型(LLM)在其他领域取得了显著进展,但其在计算机辅助设计(CAD)中的应用仍是一个有待探索的领域。在本文中,我们提出了 BlenderLLM,这是一个利用自我改进方法专门为 CAD 任务训练 LLM 的新框架。为此,我们开发了专门的训练数据集 BlendNet,并引入了全面的评估套件 CADBench。我们的结果表明,现有模型在生成准确的 CAD 脚本方面存在显著局限。然而,通过最少的基于指令的微调和迭代自我改进,BlenderLLM 在 CAD 脚本生成的功能和准确性上均显著超越了这些模型。本研究为 LLM 在 CAD 中的应用奠定了坚实基础,同时展示了自我改进模型在推进 CAD 自动化方面的变革潜力。我们鼓励进一步探索和采用这些方法,以推动该领域的创新。数据集、模型、基准测试和源代码已在 https://github.com/FreedomIntelligence/BlenderLLM 公开。

关键词

引用

@article{arxiv.2412.14203,
  title  = {BlenderLLM: Training Large Language Models for Computer-Aided Design with Self-improvement},
  author = {Yuhao Du and Shunian Chen and Wenbo Zan and Peizhao Li and Mingxuan Wang and Dingjie Song and Bo Li and Yan Hu and Benyou Wang},
  journal= {arXiv preprint arXiv:2412.14203},
  year   = {2024}
}