BrepGen: A B-rep Generative Diffusion Model with Structured Latent Geometry
BrepGen: A B-rep Generative Diffusion Model with Structured Latent Geometry(BrepGen:一个具有结构化潜在几何的B-rep生成扩散模型)
摘要
BrepGen是一种基于扩散的生成方法,可直接输出边界表示(B-rep)计算机辅助设计(CAD)模型。BrepGen将B-rep模型表示为层次树中的新型结构化潜在几何结构。根节点表示整个CAD实体,B-rep模型的每个元素(即,面,边或顶点)从上到下逐渐变为子节点。B-rep几何信息作为每个原语的全局边界框与描述局部几何形状的潜在代码一起进入节点。B-rep拓扑信息由节点复制隐式表示。当两个面共用一条边时,该边曲线将在树中出现两次,具有三条关联边的t结点顶点在具有相同节点特征的树中出现六次。BrepGen采用基于transformer的扩散模型对节点特征进行降噪,同时对重复节点进行检测和合并,恢复B-Rep拓扑信息。大量实验表明,BrepGen在CAD B-rep生成方面树立了新的里程碑,在各种基准测试中超越了现有方法。我们新收集的家具数据集的结果进一步展示了它在生成复杂几何图形方面的卓越能力。虽然以前的方法仅限于生成简单的棱柱形,但BrepGen首次结合了自由曲面和双曲面。BrepGen的其他应用包括CAD自动补全和设计插补。代码、预训练模型和数据集将被发布。
贡献:
- 一种结构化的潜在几何表示,其层次树具有节点复制,以统一的格式编码Brep几何和拓扑。
- 潜在扩散模块,能够生成自由曲面和修剪曲线。
- 新收集的家具B-rep数据集,包含10个类别的室内物体的高质量B-rep。
- 直接B-rep生成SOTA性能。
框架
局限性和未来工作
BrepGen可以在边界表示中生成3D模型,但有几个限制,可以保证未来的工作。BrepGen只支持水密的单体固体;具有多个装配体的更复杂的CAD模型留给未来的工作。如果边缘或面之间的距离太近,则在重复数据删除后将它们合并并识别为一个。在实践中,这意味着一个最小阈值,决定两个相似的表面或边缘可以有多接近。在将CAD几何归一化到[−3,3]范围后,我们只对最小阈值为0.05的数据进行训练。这种限制类似于其他素描和挤压生成方法,例如[42],将几何量化到一个固定的位范围,从而导致顶点合并。所选择的阈值0.05大致相当于7位量化后的1位差。最后,虽然用于生成最终B-rep的启发式后处理简单、快速,并且可以处理复杂的数据,但为了获得更好的结果,基于学习的后处理模块的未来工作可能会提供对无效形状的更健壮的处理。