用自监督Transformer对齐边界与形状表示,提升CAD模型处理精度与效率。
Brep2Shape: Boundary and Shape Representation Alignment via Self-Supervised Transformers
- 通过几何感知任务让模型从贝塞尔控制点预测密集空间点,理解抽象系数的物理意义。
- 采用双流Transformer并行编码曲面与曲线特征,提升几何属性捕捉能力。
- 集成拓扑注意力机制,保持曲面与曲线间拓扑一致性,适合工业级三维建模场景。
边界表示(B-rep)是计算机辅助设计(CAD)领域的行业标准。尽管深度学习在处理B-rep模型方面展现出潜力,但现有方法存在表示鸿沟:连续方法虽具解析精度却视觉抽象,离散方法虽直观清晰却牺牲几何精度。为弥合这一差距,我们提出Brep2Shape,一种新颖的自监督预训练方法,旨在对齐抽象的边界表示与直观的形状表示。该方法采用几何感知任务,使模型从参数化贝塞尔控制点预测密集空间点,从而更好地理解由抽象系数导出的物理流形。为增强对齐效果,我们设计了双流Transformer主干网络,分别独立编码曲面与曲线令牌以捕捉其不同的几何特性,并引入拓扑注意力机制,建模曲面与曲线间的依赖关系,保持拓扑一致性。实验结果表明,Brep2Shape具有显著可扩展性,在多种下游任务中实现领先精度与更快收敛速度。代码已开源:https://github.com/thuml/Brep2Shape。
原文摘要 · Abstract (English)
Boundary representation (B-rep) is the industry standard for computer-aided design (CAD). While deep learning shows promise in processing B-rep models, existing methods suffer from a representation gap: continuous approaches offer analytical precision but are visually abstract, whereas discrete methods provide intuitive clarity at the expense of geometric precision. To bridge this gap, we introduce Brep2Shape, a novel self-supervised pre-training method designed to align abstract boundary representations with intuitive shape representations. Our method employs a geometry-aware task where the model learns to predict dense spatial points from parametric Bézier control points, enabling the network to better understand physical manifolds derived from abstract coefficients. To enhance this alignment, we propose a Dual Transformer backbone with parallel streams that independently encode surface and curve tokens to capture their distinct geometric properties. Moreover, the topology attention is integrated to model the interdependencies between surfaces and curves, thereby maintaining topological consistency. Experimental results demonstrate that Brep2Shape offers significant scalability, achieving state-of-the-art accuracy and faster convergence across various downstream tasks.Code is available at this repository: https://github.com/thuml/Brep2Shape.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。