Extending Point-Based Deep Learning Approaches for Better Semantic Segmentation in CAD

Abstract

Geometry understanding is a core concept of computer-aided design and engineering (CAD/CAE). Deep neural networks have increasingly shown success as a method of processing complex inputs to achieve abstract tasks. This work revisits a generic and relatively simple approach to 3D deep learning – a point-based graph neural network – and develops best-practices and modifications to alleviate traditional drawbacks. It is shown that these methods should not be discounted for CAD tasks; with proper implementation, they can be competitive with more specifically designed approaches. Through an additive study, this work investigates how the boundary representation data can be fully utilised by leveraging the flexibility of point-based graph networks. The final configuration significantly improves on the predictive accuracy of a standard PointNet++ network across multiple CAD model segmentation datasets and achieves state-of-the-art performance on the MFCAD++ machining features dataset. The proposed modifications leave the core neural network unchanged and results also suggest that they can be applied to other point-based approaches.