MechaFormer: Sequence Learning for Kinematic Mechanism Design Automation
Diana Bolanos, Mohammadmehdi Ataei, Pradeep Kumar Jayaraman
Abstract
Designing mechanical mechanisms to trace specific paths is a classic yet notoriously difficult engineering problem, characterized by a vast and complex search space of discrete topologies and continuous parameters. We introduce MechaFormer, a Transformer-based model that tackles this challenge by treating mechanism design as a conditional sequence generation task. Our model learns to translate a target curve into a domain-specific language (DSL) string, simultaneously determining the mechanism's topology and geometric parameters in a single, unified process. MechaFormer significantly outperforms existing baselines, achieving state-of-the-art path-matching accuracy and generating a wide diversity of novel and valid designs. We demonstrate a suite of sampling strategies that can dramatically improve solution quality and offer designers valuable flexibility. Furthermore, we show that the high-quality outputs from MechaFormer serve as excellent starting points for traditional optimizers, creating a hybrid approach that finds superior solutions with remarkable efficiency.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on2
- CodeT5: Identifier-aware Unified Pre-trained Encoder-Decoder Models for Code Understanding and GenerationYue Wang, Weishi Wang, Shafiq R. Joty, Steven C. H. HoiEMNLP 2021 · 1,224 citations
- Deep Generative Model for Mechanical System Configuration DesignYasaman Etesam, Hyunmin Cheong, Mohammadmehdi Ataei, Pradeep Kumar JayaramanAAAI 2025 · 6 citations
Related papers
- SketchGen: Generating Constrained CAD SketchesWamiq Reyaz Para, Shariq Farooq Bhat, Paul Guerrero, Tom Kelly et al.NeurIPS 2021 · 114 citations
- MatFormer: a generative model for procedural materialsPaul Guerrero, Milos Hasan, Kalyan Sunkavalli, Radomír Mech et al.SIGGRAPH 2022 · 44 citations
- ACT: A Unified Framework for Rigging and Animating Characters with Arbitrary TopologiesPengyu Long, Weirui Wang, Qingcheng Zhao, Xiaoyang Guo et al.SIGGRAPH 2026
- 3DMolFormer: A Dual-channel Framework for Structure-based Drug DiscoveryXiuyuan Hu, Guoqing Liu, Can Chen, Yang Zhao et al.ICLR 2025
- Freehand Sketch Generation from Mechanical ComponentsZhichao Liao, Fengyuan Piao, Di Huang, Xinghui Li et al.ACM MM 2024 · 12 citations
