EquiCAD: A Geometric Equivariant Neural Network for 3D Shape Classification
Yonghao Su, Yantao Gan, Junfeng Long, Caiyang Yu, Wenhao Zheng, Jiancheng Lv, Xianggen Liu
Abstract
Three-dimensional (3D) shape classification plays a central role in computer vision and computer-aided design (CAD), underpinning applications in intelligent manufacturing, automated inspection, and digital engineering. Despite recent progress with 3D CNNs and graph-based approaches, existing methods often overlook the geometric-topological regularities and symmetry principles intrinsic to CAD boundary representations (B-reps). To address this challenge, we introduce EquiCAD, a symmetry-aware learning framework that integrates equivariant representations with graph-based reasoning. By leveraging group-theoretic decomposition of curve and surface descriptors, EquiCAD preserves symmetry-consistent feature transformations while retaining rich geometric details. The model further exploits hierarchical message passing to capture interactions between local features and global structure. Experimental results across multiple datasets, including SolidLetters, Parts, the Machining Feature benchmark, and our newly constructed Features dataset, demonstrate substantial improvements over prior state-of-the-art approaches, particularly on industrially relevant shapes with fine-grained attributes. These findings highlight the value of symmetry-aware modeling for robust and generalizable 3D shape analysis.
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 on5
- Vector Neurons: A General Framework for SO(3)-Equivariant NetworksCongyue Deng, Or Litany, Yueqi Duan, Adrien Poulenard et al.ICCV 2021 · 411 citations
- DeepCAD: A Deep Generative Network for Computer-Aided Design ModelsRundi Wu, Chang Xiao, Changxi ZhengICCV 2021 · 290 citations
- Point2CAD: Reverse Engineering CAD Models from 3D Point CloudsYujia Liu, Anton Obukhov, Jan Dirk Wegner, Konrad SchindlerCVPR 2024
- BRepNet: A Topological Message Passing System for Solid ModelsJoseph G. Lambourne, Karl D. D. Willis, Pradeep Kumar Jayaraman, Aditya Sanghi et al.CVPR 2021
- UV-Net: Learning From Boundary RepresentationsPradeep Kumar Jayaraman, Aditya Sanghi, Joseph G. Lambourne, Karl D. D. Willis et al.CVPR 2021
Related papers
- Constraint-Aware Feature Learning for Parametric Point CloudXi Cheng, Ruiqi Lei, Di Huang, Zhichao Liao et al.ICCV 2025 · 1 citation
- FoV-Net: Rotation-Invariant CAD B-rep Learning via Field-of-View Ray CastingMatteo Ballegeer, Dries F. BenoitCVPR 2026 · 3 citations
- Reflection and Rotation Symmetry Detection via Equivariant LearningAhyun Seo, Byungjin Kim, Suha Kwak, Minsu ChoCVPR 2022 · 12 citations
- Beyond Canonicalization: How Tensorial Messages Improve Equivariant Message PassingPeter Lippmann, Gerrit Gerhartz, Roman Remme, Fred A. HamprechtICLR 2025
- Frame Averaging for Equivariant Shape Space LearningMatan Atzmon, Koki Nagano, Sanja Fidler, Sameh Khamis et al.CVPR 2022 · 5 citations
