MeshPointNet: 3D Surface Classification Using Graph Neural Networks and Conformal Predictions on Mesh-Based Representations

Amin Heyrani Nobari,Justin Rey,Suhas Kodali,Matthew Jones,Faez Ahmed
DOI: https://doi.org/10.1115/1.4064673
2024-02-06
Journal of Mechanical Design
Abstract:Abstract In many design automation applications, accurate segmentation and classification of 3D surfaces and extraction of geometric insight from 3D models can be pivotal. This paper primarily introduces a machine learning-based scheme that leverages Graph Neural Networks (GNN) for handling 3D geometries, specifically for surface classification. Our model demonstrates superior performance against two state-of-the-art models, PointNet++ and PointMLP, in terms of surface classification accuracy, beating both models. Central to our contribution is the novel incorporation of conformal predictions, a method that offers robust uncertainty quantification and handling with marginal statistical guarantees. Unlike traditional approaches, conformal predictions enable our model to ensure precision, especially in challenging scenarios where mistakes can be highly costly. This robustness proves invaluable in design applications, and as a case in point, we showcase its utility in automating the Computational Fluid Dynamics (CFD) meshing process for aircraft models based on expert guidance. Our results reveal that our automatically generated mesh, guided by the proposed rules by experts enabled through the segmentation model, is not only efficient but matches the quality of expert-generated meshes, leading to accurate simulations. For the community's benefit, we have made our code and data available at https://github.com/ahnobari/AutoSurf Upon paper acceptance.
engineering, mechanical
What problem does this paper attempt to address?