Journal of Computing and Information Science in Engineering 2024

HG-CAD

Hierarchical Graph Learning for Material Prediction and Recommendation in Computer-Aided Design

Fig. 1 Proposed method for predicting materials of assembly bodies using hierarchical graph representation and learning

Abstract

To support intelligent computer-aided design (CAD), we introduce a machine learning architecture, namely HG-CAD, that recommends assembly body material through joint learning of body and assembly-level features using a hierarchical graph representation. Specifically, we formulate the material prediction and recommendation process as a node-level classification task over a novel hierarchical graph representation of CAD models, with a low-level graph capturing the body geometry, a high-level graph representing the assembly topology, and a batch-level mask randomization enabling contextual awareness. This enables our network to aggregate geometric and topological features from both the body and assembly levels, leading to competitive performance. Qualitative and quantitative evaluation of the proposed architecture on the Fusion 360 Gallery Assembly Dataset demonstrates the feasibility of our approach, outperforming selected computer vision and human baselines while showing promise in application scenarios. The proposed HG-CAD architecture that unifies the processing, encoding, and joint learning of multi-modal CAD features indicates the potential to serve as a recommendation system for design automation and a baseline for future work.

Download publication

Associated Researchers

Shijie Bian

California State University Northridge

Tianyang Liu

California State University Northridge

Elliot Sadler

California State University, Northridge

Bodia Borijin

California State University, Northridge

Thomas Lu

California Institute of Technology

Richard Otis

California Institute of Technology

Nhut Ho

California State University, Northridge

Bingbing Li

California State University, Northridge

View all researchers

Other Resources

Related Publications

Publication

2023

CAD-LLM: Large Language Model for CAD Generation

This research presents generating Computer Aided Designs (CAD) using…

Publication

2023

Conceptual Design Generation Using Large Language Models

Generating design concepts in product design using Large Language…

Publication

2023

What’s In A Name? Evaluating Assembly-Part Semantic Knowledge in Language Models through User-Provided Names in CAD Files

The natural language names designers use in CAD software are a…

Publication

2023

SolidGen: An Autoregressive Model for Direct B-rep Synthesis

A generative model that can synthesize 3D CAD models in the boundary…

Get in touch

Something pique your interest? Get in touch if you’d like to learn more about Autodesk Research, our projects, people, and potential collaboration opportunities.

Contact us