核心概念
BrepGen 是一種新穎的生成式 AI 方法,可以直接輸出邊界表示 (B-rep) 的電腦輔助設計 (CAD) 模型,超越了以往基於草圖和擠出的方法,能夠生成包含自由曲面和雙曲面的複雜幾何形狀。
要約
論文資訊
- 標題:BrepGen:一種基於結構化潛在幾何的 B-rep 生成式擴散模型
- 作者:Xiang Xu, Joseph G. Lambourne, Pradeep Kumar Jayaraman, Zhengqing Wang, Karl D.D. Willis, and Yasutaka Furukawa
- 期刊:ACM Transactions on Graphics
- 時間:2024 年 7 月
研究目標
本研究旨在開發一種可以直接生成邊界表示 (B-rep) 電腦輔助設計 (CAD) 模型的生成式 AI 方法,以克服現有方法的局限性,例如僅限於草圖和擠出建模操作,無法生成包含自由曲面和雙曲面的複雜幾何形狀。
方法
BrepGen 使用一種新穎的結構化潛在幾何表示法,將任何 B-rep 模型轉換為樹狀資料結構。樹的根節點表示整個 CAD 實體,而子節點則定義每個面、邊和頂點的全局位置和局部幾何形狀。局部幾何形狀使用沿 UV 域均勻採樣的點網格來表示,並使用變分自動編碼器 (VAE) 壓縮成潛在代碼。拓撲關係則透過節點複製來編碼,包括「配對複製」和「關聯複製」。配對複製確保共享的邊和頂點在樹中有多個節點,而關聯複製則將每個父節點的子節點數量填充到預定的最大分支因子。
BrepGen 採用基於 Transformer 的擴散模型,從根節點到葉節點依序對節點特徵進行去噪。在去噪過程中,具有相似特徵的節點會被檢測並合併,以恢復 B-rep 拓撲資訊。最後,透過將修剪後的曲面拼接在一起,直接輸出 B-rep 格式的 CAD 模型。
主要發現
- BrepGen 能夠生成包含自由曲面和雙曲面的複雜 B-rep 模型,超越了以往基於草圖和擠出的方法。
- BrepGen 在多個基準測試中均優於現有方法,包括 DeepCAD 和 SolidGen。
- BrepGen 在新收集的家具資料集上展現出生成複雜幾何形狀的卓越能力。
主要結論
BrepGen 是一種新穎且有效的 B-rep 生成式 AI 方法,為 CAD 設計工作流程帶來了革新。它能夠生成具有複雜拓撲和幾何形狀的逼真 CAD 模型,並為 CAD 自動完成和設計插值等應用提供了新的可能性。
貢獻
- 一種結構化潛在幾何表示法,其樹狀結構和節點複製機制能夠以統一的格式編碼 B-rep 幾何和拓撲資訊。
- 一個能夠生成自由曲面和修剪曲線的潛在擴散模組。
- 一個新收集的家具 B-rep 資料集,其中包含跨越 10 個不同類別的室內物件高品質 B-rep 模型。
- 直接 B-rep 生成,並具有最先進的效能。
局限與未來研究方向
- BrepGen 的生成速度相對較慢,未來可以探索更快的採樣方法。
- BrepGen 的訓練需要大量的計算資源,未來可以研究更有效率的訓練策略。
- BrepGen 目前僅限於生成單一零件的 CAD 模型,未來可以擴展到生成包含多個零件的組件。
統計
BrepGen 在 DeepCAD 資料集上的覆蓋率 (COV) 達到 73.87%,優於 DeepCAD 的 65.46% 和 SolidGen 的 71.03%。
BrepGen 在 DeepCAD 資料集上的最小匹配距離 (MMD) 為 1.04,優於 DeepCAD 的 1.29 和 SolidGen 的 1.08。
BrepGen 在 DeepCAD 資料集上的 Jensen-Shannon Divergence (JSD) 為 1.28,優於 DeepCAD 的 1.67 和 SolidGen 的 1.31。
BrepGen 在 DeepCAD 資料集上的新穎性 (Novel) 達到 99.8%,獨特性 (Unique) 達到 99.7%,有效性 (Valid) 達到 62.9%。
引用
"A system capable of directly generating B-reps would revolutionize the CAD design workflow, reducing the extensive manual labor required from skilled designers and the reliance on professional CAD software."
"BrepGen incorporates free-form and doubly-curved surfaces for the first time."
"Additional applications of BrepGen include CAD autocomplete and design interpolation."