论文
UniCAD:多模式多任务 CAD 的统一基准和通用模型
UniCAD: A Unified Benchmark and Universal Model for Multi-Modal Multi-Task CAD
摘要
计算机辅助设计 (CAD) 通过创建精确、可编辑的 3D 模型来支撑现代工程和制造。然而,CAD 研究通常是孤立地研究任务,并且由于缺乏统一的基准,CAD 的多模式、多任务学习受到阻碍。为了弥补这一差距,我们推出了 UniCAD,这是一个多模式 CAD 学习的综合基准,涵盖点到 CAD 重建、文本/图像到 CAD 生成以及跨不同输入模式的 CAD 问答。除了基准测试之外,我们还推出了 UniCAD-MLLM,这是一种通用的多模式 大语言模型,它可以摄取文本、图像、草图和点云,并在单个框架内以端到端的方式执行这些异构任务。对 UniCAD 和 Fusion360 基准的大量实验表明,UniCAD-MLLM 在所有任务中都实现了最先进的性能,优于现有的特定任务和多任务基准。我们将发布数据集、代码和预训练模型以加速未来的研究。