用Transformer学习CAD的边界表示,让模型理解复杂几何结构。
Bringing Attention to CAD: Boundary Representation Learning via Transformer
- 将曲面转为贝塞尔三角形,连续编码避免离散化失真。
- 通过拓扑感知嵌入,让模型识别零件特征与几何关系。
- 在零件分类和特征识别上达当前最佳效果,适合工业设计场景。
生成式人工智能(如Transformer)在自然语言处理、计算机视觉和图形学中取得显著进展,但在计算机辅助设计(CAD)领域,尤其是边界表示(B-rep)模型处理方面仍处于空白。为填补这一空白,本文提出一种名为边界表示Transformer(BRT)的新方法,将Transformer应用于B-rep学习。B-rep模型具有不规则拓扑和连续几何定义,与Transformer擅长的离散结构数据差异显著。为此,BRT采用连续几何嵌入法,将修剪与非修剪曲面编码为贝塞尔三角形,保持形状与连续性而不需离散化;同时引入拓扑感知嵌入,将几何嵌入组织为适合Transformer处理的离散序列,同时捕捉几何与拓扑特征。这使Transformer注意力机制能有效学习边界元素间的形状模式与上下文语义。大量实验表明,BRT在零件分类与特征识别任务中达到当前最优性能。
原文摘要 · Abstract (English)
The recent rise of generative artificial intelligence (AI), powered by Transformer networks, has achieved remarkable success in natural language processing, computer vision, and graphics. However, the application of Transformers in computer-aided design (CAD), particularly for processing boundary representation (B-rep) models, remains largely unexplored. To bridge this gap, we propose a novel approach for adapting Transformers to B-rep learning, called the Boundary Representation Transformer (BRT). B-rep models pose unique challenges due to their irregular topology and continuous geometric definitions, which are fundamentally different from the structured and discrete data Transformers are designed for. To address this, BRT proposes a continuous geometric embedding method that encodes B-rep surfaces (trimmed and untrimmed) into Bezier triangles, preserving their shape and continuity without discretization. Additionally, BRT employs a topology-aware embedding method that organizes these geometric embeddings into a sequence of discrete tokens suitable for Transformers, capturing both geometric and topological characteristics within B-rep models. This enables the Transformer's attention mechanism to effectively learn shape patterns and contextual semantics of boundary elements in a B-rep model. Extensive experiments demonstrate that BRT achieves state-of-the-art performance in part classification and feature recognition tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。