All summaries · 2026-10-23

GraphCAD: Geometry-Aware Diffusion on Edge-Adjacency Graphs for CAD Generation

Wenhao Zheng, Xianggen Liu, Li Yue, et al.

arXiv:2610.30974 · abstract · pdf · summarized by meta/muse-spark-1.3-contributor on 2026-10-23

The one-paragraph version

GraphCAD is a new way to generate engineering CAD models that starts from the wireframe instead of from a sequence of modeling commands. It introduces the Edge-Adjacency Graph, where each graph node stands for one physical edge of the solid — a line, arc, or circle with its start, end, and center points — and each graph link means those two edges meet at a shared corner. To create new models, it runs a hybrid diffusion process that denoises discrete choices like edge type and connectivity and continuous numbers like 3D coordinates at the same time, while repeatedly rendering the noisy numbers into actual 3D curves, sampling points along them, and encoding those points with a rotation-aware network. Tested on a new benchmark called EAGBench built from DeepCAD and Fusion 360 Gallery parts, it generates complex models with more than 50 faces and beats sequence-based and graph-based baselines on geometric fidelity and distributional match, with very high topological validity.

The problem they're solving, and why it matters

In engineering, a 3D part is usually stored not as a mesh or point cloud but as a Boundary Representation, meaning a precise description of parametric faces, edges, and vertices plus how they connect. That term means the solid is defined by its skin: what the surfaces are mathematically, and which edge touches which face and vertex. Generating that format is hard because it mixes two very different jobs: getting the discrete topology right, meaning which things connect, and getting the continuous geometry right, meaning exactly where they sit in space to micron tolerance.

Most existing generators treat CAD as a sequence of operations, like draw a sketch then extrude it, in the style of DeepCAD, Text2CAD, and related command-prediction methods. The paper argues this lacks what it calls topological inductive bias, meaning a built-in preference for connected, watertight structure. A small early mistake — a line endpoint slightly off — propagates and leaves open loops, non-manifold edges where more or fewer than two faces meet, or solids that fail a geometry-kernel check. More recent graph methods use a Face-Adjacency Graph, meaning nodes are faces and links are shared edges, but faces are complex trimmed surfaces that are hard to learn, and those methods often approximate shape by sampling in the UV domain, meaning the flat 2D parameter space of a surface, without explicitly enforcing the 1D wireframe boundaries. For manufacturing, an almost-right shape that does not close into a solid is not usable, so validity and boundary precision matter as much as visual similarity.

The key idea, in plain words

Shift attention from surfaces to the wireframe skeleton. Instead of asking the model to invent complicated surfaces first, ask it to invent the network of boundary curves that must meet cleanly to form closed wires, then recover surfaces from those wires afterward.

Concretely, imagine a simple extruded L-bracket. Its solid is bounded by straight lines and perhaps a circular hole. In the Edge-Adjacency Graph, each of those lines or circles becomes a node labeled with its curve type and just three 3D points. Two nodes are linked if their curves share a corner in the real part. That is a much more compact learning target than a full surface patch, and connectivity directly encodes whether the wire can close.

The second key idea is to ground generation in physical geometry during denoising. Standard graph diffusion treats node features as abstract numbers to clean up. GraphCAD instead renders those numbers into space at every step: interpret them as a curve, sample points along it, and let the network see the resulting point cloud with a backbone that understands rotation. A line that is too short, an arc whose three points are collinear so its plane is undefined, or two supposedly connected edges whose endpoints do not coincide then look wrong geometrically, not just numerically, so the model can learn to fix them.

How it actually works, step by step

First, representation. Starting from a Boundary Representation solid, the method builds an undirected Edge-Adjacency Graph. The term Edge Nodes here means one graph node per topological edge in the CAD model, each carrying a compact feature vector of length 12: a curve-type label with three possibilities — Line, Arc, Circle — plus a start point, an end point, and a center point, each a 3D coordinate. Lines mask out the center as undefined. Circles use two points on the circumference plus a center chosen so the three points are not in a straight line, which uniquely fixes the circle's plane and curvature. Spline curves, meaning freeform curves, are not modeled directly but approximated as a chain of arc segments. Graph edges, meaning links between nodes, are added when two CAD edges share a vertex.

Second, hybrid diffusion. The term forward process means gradually adding noise to a clean graph over 500 steps; reverse process means learning to remove that noise to generate new graphs. Because the graph lives in what the authors call a product space, meaning discrete and continuous parts mixed together, they noise each part with its own rule but denoise them jointly. Type labels and topology links use categorical noise defined by transition matrices, where the chance of flipping from one category to another is prescribed. Continuous coordinates use Gaussian noise with a variance schedule, meaning progressively larger random jitters. For generation, a single shared denoising network looks at the whole noisy graph and predicts the clean types, clean links, and the noise added to the coordinates, so the three outputs stay coupled even though sampling treats them as conditionally independent given that prediction.

Third, geometry-aware denoising. For each node, the noisy coordinates are interpreted as control points for what is called a Quadratic Bezier curve, meaning a smooth curve defined by blending control points. This is a relaxation that stays differentiable even when the noisy state violates strict CAD invariants like equal radii. The method then uniformly samples 100 points along that curve to make a local point set, encodes it with a Vector Neuron encoder — a network whose vector features rotate when the input rotates — and concatenates that geometric embedding with the raw parameters. Inside the backbone, rotation-invariant information like type labels flows through standard neural layers as a scalar stream, while spatial information flows through rotation-equivariant operations as a vector stream that preserves rotational structure.

Fourth, prediction and reconstruction. Separate heads predict discrete types and links with classification losses and continuous geometry with a masked squared-error loss. The mask, meaning a binary switch derived from the true type, zeros out supervision for meaningless entries like the center of a line so the model is not forced to hallucinate them. After sampling, lines, arcs, and circles are instantiated with Open CASCADE builder calls, assembled into closed wires, skinned into surfaces and shells, and checked for watertightness to make solids. The paper is explicit that this post-processing recovers full surface definitions rather than generating them directly, which becomes a source of error on highly curved regions.

What they showed — results, honestly, with limitations

On unconditional graph generation, GraphCAD outperformed flow-based GraphNVP and diffusion-based GDSS, ConGress, and DiGress on distributional distances for degree, clustering, and orbit counts, and on Validity, Uniqueness, and Novelty. For example, on the harder Fusion 360-derived split, it kept 96.58% validity versus 77.45% for the strongest baseline DiGress, while uniqueness and novelty stayed above 95% and 92%, suggesting diversity without simple copying. Qualitative examples show varied parts with complex topology.

On conditional CAD generation, where a model must match an input constraint and is scored after full solid reconstruction, GraphCAD achieved the best Chamfer Distance, meaning average nearest-point error between sampled surfaces, Hausdorff Distance, meaning worst-case surface deviation, Jensen-Shannon Divergence and Minimum Matching Distance, meaning distributional similarity and diversity, on both splits. On the DeepCAD-derived split it cut Chamfer Distance from 2.34 to 1.73 and Jensen-Shannon Divergence from 65.92 to 33.22 compared to TGBFN, with distances scaled by 100 in a normalized cube, though TGBFN had slightly lower Invalidity Ratio, meaning fraction of failures, at 0.78% versus 1.67%. On the Fusion 360-derived split it was best on all five metrics including invalidity at 5.42%.

Ablations confirm each piece matters, but geometry grounding matters most. Removing geometric rendering dropped validity from about 98.3% to 66.5% and sharply worsened fidelity. Removing loss masking dropped validity to about 73.2%. Removing rotation equivariance caused a smaller but consistent decline. Even sparse rendering helps: with zero sampled points the invalidity ratio was 27.59%, with 10 points it fell to 2.52%, and with 100 points to around 1.8% at a modest extra cost of about 0.66 GB memory.

Limitations are stated plainly. Surfaces including NURBS control points, meaning the control net for freeform surfaces, are recovered afterward, not generated, so complex curvature can fail. Diffusion needs 500 steps, so inference is slower than single-pass methods. Experiments are component-level parts; scaling to assemblies with thousands of edges with a much larger discrete state space remains unsolved. The paper does not claim text-conditioned generation in its main results, so text-to-CAD controllability is not demonstrated here.

Why this might matter to me

For spatial reasoning in language and vision models, the lesson is that explicit topology plus rendered geometry beats command-history imitation. An LLM or VLM that predicts CAD tokens can produce plausible-looking sequences that fail to close; an edge-graph with shared-vertex links gives a checkable scaffold for whether endpoints coincide and wires close, which could serve as a verifier or intermediate representation for language-driven design.

For world models, which need to predict physically valid 3D structure rather than just pixels, the hybrid discrete-continuous diffusion and rotation-equivariant encoding are directly relevant. The result that validity collapses without rendering but recovers with even 10 sampled points suggests world models should decode latent vectors into point sets and test physical consistency during rollout, not only at the end, and that rotation robustness should be architectural rather than purely augmented.

For text-to-CAD and parametric geometry, GraphCAD does not itself solve language conditioning, but it offers a cleaner target for it: generate typed primitives with analytic parameters and adjacency, then solve surfaces. That is more editable and kernel-friendly than meshes, and the semantic loss mask is a useful pattern for handling type-dependent parameters like radii that only exist for certain primitives. Future text-to-CAD could condition this diffusion on prompts while keeping the watertightness check.

For 3D generation and scene representation, the wireframe-first view complements surface, volumetric, and neural-field representations. For engineered scenes with many prismatic and drilled parts, starting from 1D boundaries that enforce closure may yield more manufacturable assets than starting from faces or implicit fields, at the cost of handling freeform surfaces and large assemblies separately.

Terms worth knowing

Boundary Representation, often shortened to B-rep, is the standard CAD format that defines a solid by parametric geometry plus topology of faces, edges, and vertices. Edge-Adjacency Graph, or EAG, is this paper's representation where nodes are CAD edges and links mean shared vertices. Face-Adjacency Graph, or FAG, is the prior alternative where nodes are faces. Hybrid diffusion here means joint denoising of discrete categories via categorical transitions and continuous coordinates via Gaussian noise. SE(3)-equivariance means that if the input is rigidly rotated and translated, the network prediction transforms the same way; translation is handled by canonicalization and rotation by the backbone. Vector Neurons are a building block for networks whose 3D vector features rotate consistently with the input. Invalidity Ratio measures how often reconstruction into a solid fails; Chamfer and Hausdorff distances measure average and worst-case surface error; Jensen-Shannon Divergence and Minimum Matching Distance measure distributional match and diversity.