BrepGen: A B-rep Generative Diffusion Model with Structured Latent Geometry

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Xu, Xiang, Lambourne, Joseph G., Jayaraman, Pradeep Kumar, Wang, Zhengqing, Willis, Karl D. D., Furukawa, Yasutaka
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866909374820122624
author Xu, Xiang
Lambourne, Joseph G.
Jayaraman, Pradeep Kumar
Wang, Zhengqing
Willis, Karl D. D.
Furukawa, Yasutaka
author_facet Xu, Xiang
Lambourne, Joseph G.
Jayaraman, Pradeep Kumar
Wang, Zhengqing
Willis, Karl D. D.
Furukawa, Yasutaka
contents This paper presents BrepGen, a diffusion-based generative approach that directly outputs a Boundary representation (B-rep) Computer-Aided Design (CAD) model. BrepGen represents a B-rep model as a novel structured latent geometry in a hierarchical tree. With the root node representing a whole CAD solid, each element of a B-rep model (i.e., a face, an edge, or a vertex) progressively turns into a child-node from top to bottom. B-rep geometry information goes into the nodes as the global bounding box of each primitive along with a latent code describing the local geometric shape. The B-rep topology information is implicitly represented by node duplication. When two faces share an edge, the edge curve will appear twice in the tree, and a T-junction vertex with three incident edges appears six times in the tree with identical node features. Starting from the root and progressing to the leaf, BrepGen employs Transformer-based diffusion models to sequentially denoise node features while duplicated nodes are detected and merged, recovering the B-Rep topology information. Extensive experiments show that BrepGen advances the task of CAD B-rep generation, surpassing existing methods on various benchmarks. Results on our newly collected furniture dataset further showcase its exceptional capability in generating complicated geometry. While previous methods were limited to generating simple prismatic shapes, BrepGen incorporates free-form and doubly-curved surfaces for the first time. Additional applications of BrepGen include CAD autocomplete and design interpolation. The code, pretrained models, and dataset are available at https://github.com/samxuxiang/BrepGen.
format Preprint
id arxiv_https___arxiv_org_abs_2401_15563
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle BrepGen: A B-rep Generative Diffusion Model with Structured Latent Geometry
Xu, Xiang
Lambourne, Joseph G.
Jayaraman, Pradeep Kumar
Wang, Zhengqing
Willis, Karl D. D.
Furukawa, Yasutaka
Computer Vision and Pattern Recognition
Machine Learning
This paper presents BrepGen, a diffusion-based generative approach that directly outputs a Boundary representation (B-rep) Computer-Aided Design (CAD) model. BrepGen represents a B-rep model as a novel structured latent geometry in a hierarchical tree. With the root node representing a whole CAD solid, each element of a B-rep model (i.e., a face, an edge, or a vertex) progressively turns into a child-node from top to bottom. B-rep geometry information goes into the nodes as the global bounding box of each primitive along with a latent code describing the local geometric shape. The B-rep topology information is implicitly represented by node duplication. When two faces share an edge, the edge curve will appear twice in the tree, and a T-junction vertex with three incident edges appears six times in the tree with identical node features. Starting from the root and progressing to the leaf, BrepGen employs Transformer-based diffusion models to sequentially denoise node features while duplicated nodes are detected and merged, recovering the B-Rep topology information. Extensive experiments show that BrepGen advances the task of CAD B-rep generation, surpassing existing methods on various benchmarks. Results on our newly collected furniture dataset further showcase its exceptional capability in generating complicated geometry. While previous methods were limited to generating simple prismatic shapes, BrepGen incorporates free-form and doubly-curved surfaces for the first time. Additional applications of BrepGen include CAD autocomplete and design interpolation. The code, pretrained models, and dataset are available at https://github.com/samxuxiang/BrepGen.
title BrepGen: A B-rep Generative Diffusion Model with Structured Latent Geometry
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2401.15563