BRepGAT: Graph neural network to segment machining feature faces in a B-rep model

Abstract

In recent years, there have been many studies using artificial intelligence to recognize machining features in 3D models in the CAD/CAM field. Most of these studies converted the original CAD data into images, point clouds, or voxels for recognition. This led to information loss during the conversion process, resulting in decreased recognition accuracy. In this paper, we propose a graph-based network called BRepGAT to segment faces in an original B-rep model containing machining features. We define descriptors that represent information about the faces and edges of the B-rep model from the perspective of feature recognition. These descriptors are extracted from the B-rep model and transformed into homogeneous graph data, which is then passed to graph networks. BRepGAT recognize machining features on a face-by-face based on the graph data input. Our experimental results using the MFCAD18++ dataset showed that BRepGAT achieved state-of-the-art recognition accuracy (99.1%). Furthermore, BRepGAT showed relatively robust performance on other datasets besides MFCAD18++.