Draw it like Euclid: Teaching transformer models to generate CAD profiles using ruler and compass construction steps

Siyi Li, Joseph George Lambourne, Pradeep Kumar Jayaraman, Longfei Zhang, Karl D.D. Willis

Keywords: CAD, geometry, reinforcement learning, chain of thought

TL;DR: This paper introduces a new method of generating Computer Aided Design (CAD) geometry via a sequence of simple geometric constructions including curve offsetting, rotations and intersections.

Abstract: We introduce a new method of generating Computer Aided Design (CAD) profiles via a sequence of simple geometric constructions including curve offsetting, rotations and intersections. These sequences start with geometry provided by a designer and build up the points and curves of the final profile step by step. We demonstrate that adding construction steps between the designer's input geometry and the final profile improves generation quality in a similar way to the introduction of a chain of thought in language models. Similar to the constraints in a parametric CAD model, the construction sequences reduce the degrees of freedom in the modeled shape to a small set of parameter values which can be adjusted by the designer, allowing parametric editing with the constructed geometry evaluated to floating point precision. In addition we show that applying reinforcement learning to the construction sequences gives further improvements over a wide range of metrics, including some which were not explicitly optimized.

Primary Area: learning on graphs and other geometries & topologies

Submission Number: 10850