Self-Supervised Representation Learning for CAD
Not on the map. Why?
Abstract (arXiv)
The design of man-made objects is dominated by computer aided design (CAD) tools. Assisting design with data-driven machine learning methods is hampered by lack of labeled data in CAD's native format; the parametric boundary representation (B-Rep). Several data sets of mechanical parts in B-Rep format have recently been released for machine learning research. However, large scale databases are largely unlabeled, and labeled datasets are small. Additionally, task specific label sets are rare, and costly to annotate. This work proposes to leverage unlabeled CAD geometry on supervised learning tasks. We learn a novel, hybrid implicit/explicit surface representation for B-Rep geometry, and show that this pre-training significantly improves few-shot learning performance and also achieves state-of-the-art performance on several existing B-Rep benchmarks.
Key facts. Cited by 22 papers in the map (data through 7 Oct 2026). Builds on 34 of its 47 references that are in the map. First posted on arXiv 19 Oct 2022.
Builds on
34 of 47 references.
References · 34
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2017Graph Attention Networkscited by 8,854
- 2013Auto-Encoding Variational Bayescited by 3,980
- 2019DeepSDF: Learning Continuous Signed Distance Functions for Shape Representationcited by 2,682
- 2018Occupancy Networks: Learning 3D Reconstruction in Function Spacecited by 2,037
- 2018Unsupervised Representation Learning by Predicting Image Rotationscited by 1,586
- 2016Context Encoders: Feature Learning by Inpaintingcited by 1,424
- 2016Unsupervised Learning of Visual Representations by Solving Jigsaw Puzzlescited by 1,379
- 2016Colorful Image Colorizationcited by 1,113
- 2015Unsupervised Visual Representation Learning by Context Predictioncited by 1,062
- 2018Learning Implicit Fields for Generative Shape Modelingcited by 1,060
- 2016A Point Set Generation Network for 3D Object Reconstruction from a Single Imagecited by 972
22 more
- 2018AtlasNet: A Papier-Mach'e Approach to Learning 3D Surface Generationcited by 539
- 2014Semi-Supervised Learning with Deep Generative Modelscited by 494
- 2019Generalizing from a Few Examples: A Survey on Few-Shot Learningcited by 462
- 2018ABC: A Big CAD Model Dataset For Geometric Deep Learningcited by 339
- 2021DeepCAD: A Deep Generative Network for Computer-Aided Design Modelscited by 266
- 2016Learning Representations for Automatic Colorizationcited by 251
- 2020Design Space for Graph Neural Networkscited by 182
- 2016Generative and Discriminative Voxel Modeling with Convolutional Neural Networkscited by 128
- 2022SkexGen: Autoregressive Generation of CAD Construction Sequences with Disentangled Codebookscited by 121
- 2021BRepNet: A topological message passing system for solid modelscited by 92
- 2021Computer-Aided Design as Languagecited by 82
- 2022SolidGen: An Autoregressive Model for Direct B-rep Synthesiscited by 78
- 2021JoinABLe: Learning Bottom-up Assembly of Parametric CAD Jointscited by 77
- 2020SketchGraphs: A Large-Scale Dataset for Modeling Relational Geometry in Computer-Aided Designcited by 72
- 2017Variational Autoencoders for Deforming 3D Mesh Modelscited by 69
- 2021SketchGen: Generating Constrained CAD Sketchescited by 68
- 2021Vitruvion: A Generative Model of Parametric CAD Sketchescited by 66
- 2017Interactive 3D Modeling with a Generative Adversarial Networkcited by 18
- 2022ComplexGen: CAD Reconstruction by B-Rep Chain Complex Generationcited by 17
- 2020Sketch2CAD: Sequential CAD Modeling by Sketching in Contextcited by 14
- 2021AutoMate: A Dataset and Learning Approach for Automatic Mating of CAD Assembliescited by 13
- 2021DeepCurrents: Learning Implicit Representations of Shapes with Boundariescited by 12
Built on by
Citing papers per year: 2023: 2; 2024: 6; 2025: 5; 2026: 9.
22 papers.
Sort:
By area:
- 2024BrepGen: A B-rep Generative Diffusion Model with Structured Latent Geometrycited by 101
- 2024CAD-MLLM: Unifying Multimodality-Conditioned CAD Generation With MLLMcited by 74
- 2025HoLa: B-Rep Generation using a Holistic Latent Representationcited by 29
- 2024CadVLM: Bridging Language and Vision in the Generation of Parametric CAD Sketchescited by 28
- 2025Seek-CAD: A Self-refined Generative Modeling for 3D Parametric CAD Using Local Inference via DeepSeekcited by 22
- 2024Geometric Deep Learning for Computer-Aided Design: A Surveycited by 14
- 2025BrepGPT: Autoregressive B-rep Generation with Voronoi Half-Patchcited by 12
- 2024VIRL: Volume-Informed Representation Learning towards Few-shot Manufacturability Estimationcited by 4
- 2026FoV-Net: Rotation-Invariant CAD B-rep Learning via Field-of-View Ray Castingcited by 3
- 2026Mamba-CAD: State Space Model For 3D Computer-Aided Design Generative Modelingcited by 3
- 2026Masked BRep Autoencoder via Hierarchical Graph Transformercited by 3
- 2023Representation Learning for Sequential Volumetric Design Taskscited by 2
10 more
- 2025A Survey of AI Methods for Geometry Preparation and Mesh Generation in Engineering Simulationcited by 2
- 2026SceMoS: Scene-Aware 3D Human Motion Synthesis by Planning with Geometry-Grounded Tokenscited by 2
- 2023P2CADNet: An End-to-End Reconstruction Network for Parametric 3D CAD Model from Point Cloudscited by 1
- 2024Self-supervised Graph Neural Network for Mechanical CAD Retrievalcited by 1
- 2026BRepCLIP: Contrastive Multimodal Pretraining on BRep Primitives for CAD Understandingcited by 1
- 2025Attention to Detail: Fine-Scale Feature Preservation-Oriented Geometric Pre-training for AI-Driven Surrogate Modelingcited by 0
- 2026Kolmogorov-Arnold Networks-Based Tolerance-Aware Manufacturability Assessment Integrating Design-for-Manufacturing Principlescited by 0
- 2026Brep2Shape: Boundary and Shape Representation Alignment via Self-Supervised Transformerscited by 0
- 2026HiFi-BRep: High-Fidelity Latent Representation for Robust B-Rep Generationcited by 0
- 2026DepthBenchCAD: When Does Deeper Auditing Yield More Reliable Conclusions?cited by 0
- 2026DepthBenchCAD: When Does Deeper Auditing Yield More Reliable Conclusions?14 Sep 2026
- 2026HiFi-BRep: High-Fidelity Latent Representation for Robust B-Rep Generation17 Aug 2026
- 2026BRepCLIP: Contrastive Multimodal Pretraining on BRep Primitives for CAD Understanding3 Jun 2026
- 2026Masked BRep Autoencoder via Hierarchical Graph Transformer16 Mar 2026
- 2026Mamba-CAD: State Space Model For 3D Computer-Aided Design Generative Modeling28 Feb 2026
- 2026FoV-Net: Rotation-Invariant CAD B-rep Learning via Field-of-View Ray Casting27 Feb 2026
- 2026SceMoS: Scene-Aware 3D Human Motion Synthesis by Planning with Geometry-Grounded Tokens24 Feb 2026
- 2026Brep2Shape: Boundary and Shape Representation Alignment via Self-Supervised Transformers7 Feb 2026
- 2026Kolmogorov-Arnold Networks-Based Tolerance-Aware Manufacturability Assessment Integrating Design-for-Manufacturing Principles9 Jan 2026
- 2025A Survey of AI Methods for Geometry Preparation and Mesh Generation in Engineering Simulation16 Dec 2025
- 2025BrepGPT: Autoregressive B-rep Generation with Voronoi Half-Patch27 Nov 2025
- 2025Seek-CAD: A Self-refined Generative Modeling for 3D Parametric CAD Using Local Inference via DeepSeek23 May 2025
10 more
- 2025Attention to Detail: Fine-Scale Feature Preservation-Oriented Geometric Pre-training for AI-Driven Surrogate Modeling27 Apr 2025
- 2025HoLa: B-Rep Generation using a Holistic Latent Representation19 Apr 2025
- 2024CAD-MLLM: Unifying Multimodality-Conditioned CAD Generation With MLLM7 Nov 2024
- 2024CadVLM: Bridging Language and Vision in the Generation of Parametric CAD Sketches26 Sep 2024
- 2024VIRL: Volume-Informed Representation Learning towards Few-shot Manufacturability Estimation18 Jun 2024
- 2024Self-supervised Graph Neural Network for Mechanical CAD Retrieval13 Jun 2024
- 2024Geometric Deep Learning for Computer-Aided Design: A Survey27 Feb 2024
- 2024BrepGen: A B-rep Generative Diffusion Model with Structured Latent Geometry28 Jan 2024
- 2023P2CADNet: An End-to-End Reconstruction Network for Parametric 3D CAD Model from Point Clouds4 Oct 2023
- 2023Representation Learning for Sequential Volumetric Design Tasks5 Sep 2023
- 2024CAD-MLLM: Unifying Multimodality-Conditioned CAD Generation With MLLMcited by 74
- 2025HoLa: B-Rep Generation using a Holistic Latent Representationcited by 29
- 2024Geometric Deep Learning for Computer-Aided Design: A Surveycited by 14
- 2025BrepGPT: Autoregressive B-rep Generation with Voronoi Half-Patchcited by 12
- 2024VIRL: Volume-Informed Representation Learning towards Few-shot Manufacturability Estimationcited by 4
- 2026FoV-Net: Rotation-Invariant CAD B-rep Learning via Field-of-View Ray Castingcited by 3
- 2026Masked BRep Autoencoder via Hierarchical Graph Transformercited by 3
- 2023Representation Learning for Sequential Volumetric Design Taskscited by 2
- 2025A Survey of AI Methods for Geometry Preparation and Mesh Generation in Engineering Simulationcited by 2
- 2026SceMoS: Scene-Aware 3D Human Motion Synthesis by Planning with Geometry-Grounded Tokenscited by 2
- 2023P2CADNet: An End-to-End Reconstruction Network for Parametric 3D CAD Model from Point Cloudscited by 1
- 2024Self-supervised Graph Neural Network for Mechanical CAD Retrievalcited by 1
12 most cited of 18 All 18 (JSON)
- 2026DepthBenchCAD: When Does Deeper Auditing Yield More Reliable Conclusions?14 Sep 2026
- 2026HiFi-BRep: High-Fidelity Latent Representation for Robust B-Rep Generation17 Aug 2026
- 2026BRepCLIP: Contrastive Multimodal Pretraining on BRep Primitives for CAD Understanding3 Jun 2026
- 2026Masked BRep Autoencoder via Hierarchical Graph Transformer16 Mar 2026
- 2026FoV-Net: Rotation-Invariant CAD B-rep Learning via Field-of-View Ray Casting27 Feb 2026
- 2026SceMoS: Scene-Aware 3D Human Motion Synthesis by Planning with Geometry-Grounded Tokens24 Feb 2026
- 2026Brep2Shape: Boundary and Shape Representation Alignment via Self-Supervised Transformers7 Feb 2026
- 2026Kolmogorov-Arnold Networks-Based Tolerance-Aware Manufacturability Assessment Integrating Design-for-Manufacturing Principles9 Jan 2026
- 2025A Survey of AI Methods for Geometry Preparation and Mesh Generation in Engineering Simulation16 Dec 2025
- 2025BrepGPT: Autoregressive B-rep Generation with Voronoi Half-Patch27 Nov 2025
- 2025Attention to Detail: Fine-Scale Feature Preservation-Oriented Geometric Pre-training for AI-Driven Surrogate Modeling27 Apr 2025
- 2025HoLa: B-Rep Generation using a Holistic Latent Representation19 Apr 2025
12 newest of 18 All 18 (JSON)