SketchGraphs: A Large-Scale Dataset for Modeling Relational Geometry in Computer-Aided Design

TL;DR

SketchGraphs is a large dataset of 15 million CAD sketches with geometric constraint graphs, enabling research in generative modeling and constraint prediction for parametric CAD design automation.

Contribution

The paper introduces SketchGraphs, a comprehensive dataset and processing pipeline for modeling relational geometry in CAD, supporting machine learning research in design synthesis.

Findings

Established benchmarks for generative modeling of sketches.

Demonstrated conditional generation of constraints from geometry.

Provided a large-scale, real-world CAD sketch dataset.

Abstract

Parametric computer-aided design (CAD) is the dominant paradigm in mechanical engineering for physical design. Distinguished by relational geometry, parametric CAD models begin as two-dimensional sketches consisting of geometric primitives (e.g., line segments, arcs) and explicit constraints between them (e.g., coincidence, perpendicularity) that form the basis for three-dimensional construction operations. Training machine learning models to reason about and synthesize parametric CAD designs has the potential to reduce design time and enable new design workflows. Additionally, parametric CAD designs can be viewed as instances of constraint programming and they offer a well-scoped test bed for exploring ideas in program synthesis and induction. To facilitate this research, we introduce SketchGraphs, a collection of 15 million sketches extracted from real-world CAD models coupled with an open-source data processing pipeline. Each sketch is represented as a geometric constraint graph where edges denote designer-imposed geometric relationships between primitives, the nodes of the graph. We demonstrate and establish benchmarks for two use cases of the dataset: generative modeling of sketches and conditional generation of likely constraints given unconstrained geometry.