GeoFusion-CAD: Structure-Aware Diffusion with Geometric State Space for Parametric 3D Design

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhou, Xiaolei, Fang, Chuangjie, Wu, Jie, Yang, Jingyi, Lin, Boyi, Zheng, Jianwei
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915883456135168
author Zhou, Xiaolei
Fang, Chuangjie
Wu, Jie
Yang, Jingyi
Lin, Boyi
Zheng, Jianwei
author_facet Zhou, Xiaolei
Fang, Chuangjie
Wu, Jie
Yang, Jingyi
Lin, Boyi
Zheng, Jianwei
contents Parametric Computer-Aided Design (CAD) is fundamental to modern 3D modeling, yet existing methods struggle to generate long command sequences, especially under complex geometric and topological dependencies. Transformer-based architectures dominate CAD sequence generation due to their strong dependency modeling, but their quadratic attention cost and limited context windowing hinder scalability to long programs. We propose GeoFusion-CAD, an end-to-end diffusion framework for scalable and structure-aware generation. Our proposal encodes CAD programs as hierarchical trees, jointly capturing geometry and topology within a state-space diffusion process. Specifically, a lightweight C-Mamba block models long-range structural dependencies through selective state transitions, enabling coherent generation across extended command sequences. To support long-sequence evaluation, we introduce DeepCAD-240, an extended benchmark that increases the sequence length ranging from 40 to 240 while preserving sketch-extrusion semantics from the ABC dataset. Extensive experiments demonstrate that GeoFusion-CAD achieves superior performance on both short and long command ranges, maintaining high geometric fidelity and topological consistency where Transformer-based models degrade. Our approach sets new state-of-the-art scores for long-sequence parametric CAD generation, establishing a scalable foundation for next-generation CAD modeling systems. Code and datasets are available at GitHub.
format Preprint
id arxiv_https___arxiv_org_abs_2603_21978
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle GeoFusion-CAD: Structure-Aware Diffusion with Geometric State Space for Parametric 3D Design
Zhou, Xiaolei
Fang, Chuangjie
Wu, Jie
Yang, Jingyi
Lin, Boyi
Zheng, Jianwei
Computer Vision and Pattern Recognition
Graphics
Parametric Computer-Aided Design (CAD) is fundamental to modern 3D modeling, yet existing methods struggle to generate long command sequences, especially under complex geometric and topological dependencies. Transformer-based architectures dominate CAD sequence generation due to their strong dependency modeling, but their quadratic attention cost and limited context windowing hinder scalability to long programs. We propose GeoFusion-CAD, an end-to-end diffusion framework for scalable and structure-aware generation. Our proposal encodes CAD programs as hierarchical trees, jointly capturing geometry and topology within a state-space diffusion process. Specifically, a lightweight C-Mamba block models long-range structural dependencies through selective state transitions, enabling coherent generation across extended command sequences. To support long-sequence evaluation, we introduce DeepCAD-240, an extended benchmark that increases the sequence length ranging from 40 to 240 while preserving sketch-extrusion semantics from the ABC dataset. Extensive experiments demonstrate that GeoFusion-CAD achieves superior performance on both short and long command ranges, maintaining high geometric fidelity and topological consistency where Transformer-based models degrade. Our approach sets new state-of-the-art scores for long-sequence parametric CAD generation, establishing a scalable foundation for next-generation CAD modeling systems. Code and datasets are available at GitHub.
title GeoFusion-CAD: Structure-Aware Diffusion with Geometric State Space for Parametric 3D Design
topic Computer Vision and Pattern Recognition
Graphics
url https://arxiv.org/abs/2603.21978