华南理工大学学报(自然科学版) ›› 2025, Vol. 53 ›› Issue (5): 20-31.doi: 10.12141/j.issn.1000-565X.240329

• 机械工程 • 上一篇    

基于图神经网络的B-Rep模型加工特征识别方法

胡广华 代志刚 王清辉   

  1. 华南理工大学 机械与汽车工程学院,广东 广州 510640

  • 出版日期:2025-05-25 发布日期:2024-12-13

Machining feature recognition method of B-Rep model based on graph neural network

HU GuangHua DAI ZhiGang WANG QingHui   

  1. School of Mechanical and Automotive EngineeringSouth China University of TechnologyGuangzhou 510640, GuangdongChina
  • Online:2025-05-25 Published:2024-12-13

摘要:

动特征识别(AFR)是智能制造的关键技术之一。传统的基于规则的识别算法可扩展性较差,而基于深度卷积网络的方法以离散模型为输入,准确度不高,且识别结果难以精确映射回原始CAD模型,造成应用不便。针对上述不足,本文提出了一种基于图神经网络的特征识别方法,能够直接处理B-Rep模型。该方法首先从B-Rep结构中提取有效的属性和几何信息,形成特征描述符;接着根据CAD模型拓扑结构建立具有高级语义信息的邻接图;进而以邻接图为输入,构建了高效的图神经网络模型,通过引入可微的广义消息聚合函数和残差连接机制,提升模型的信息聚合及多层级特征捕捉能力,同时采用消息归一化策略确保训练稳定性并加速收敛;训练完成后,网络能对B-Rep模型中的所有面进行分类标注,实现特征识别。在公共数据集MFCAD++上测试,本文方法取得99.53%的准确率和99.15%的平均交并比,优于同类研究。采用更复杂的测试用例和工程应用中的典型真实CAD案例作进一步检验,结果均表明本文方法具有更好的泛化能力及适应性。

关键词: 加工特征识别, 图神经网络, 深度学习, 计算机辅助设计

Abstract:

Automatic feature recognition (AFR) is one of the key technologies of intelligent manufacturing. Traditional rule-based recognition algorithms have poor scalability, while methods based on deep convolutional networks use discrete models as input, have low accuracy, and the recognition results are difficult to accurately map back to the original CAD model, causing inconvenience in application. In view of the above shortcomings, we proposed a feature recognition method based on graph neural network, which can directly analyze B-Rep models. Our method extracts effective characteristic information and geometric information from the B-Rep structures to form a feature descriptor, and then establishes an adjacency graph consist of high-level semantic information based on the topological structure of the CAD model. Taking the adjacency graph as input, an efficient graph neural network model is constructed. By introducing a differentiable generalized message aggregation function and a residual connection mechanism, the model has stronger information aggregation performance and multi-level feature capture capabilities. What is more, the message normalization strategy is used to ensure the stability of the training process and to accelerate the convergence of the model. After training, the network can directly classify and annotate all faces in the B-Rep model, thereby realizing feature recognition. Experimental results based on the public dataset MFCAD++ demonstrates that the proposed method achieved an accuracy of 99.53% and an average intersection-over-union ratio of 99.15%, which outperforms other similar studies. Further evaluations using more complex testing cases and typical CAD cases from real engineering applications show that the proposed method has better generalization ability and adaptability.

Key words: machining feature recognition, graph neural network, deep learning, computer aided design