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

    Feature Recognition Method Based on Multi-Feature Fusion and Graph Attention

    • 摘要: 针对三维CAD模型加工特征自动识别任务中现有方法存在信息损失与泛化能力不足的问题,本文提出一种基于改进图神经网络的识别方法。该方法首先将边界表示(B-Rep)模型构建为一种能融合几何、拓扑与属性信息的几何属性邻接图,其中精确的几何特征通过UV网格化后由卷积网络(CNN)提取,丰富的属性特征则由多层感知机(MLP)编码。在此基础上,设计了一种改进的图神经网络架构,其核心是在图神经网络的基础上引入了图外部注意力机制(GEANet)。该机制通过一个可学习的外部共享记忆单元,使模型不仅能学习单个零件内部的局部特征,还能捕捉和利用跨零件(图)的共性结构模式,从而深化了对复杂和相交特征的理解,并有效提升了模型的泛化能力。实验在MFCAD和MFCAD++数据集上进行,MFCAD数据集上的准确率和平均交并比(mIoU)分别为99.73%和99.48%,MFCAD++数据集上的准确率和平均交并比分别为98.78%和97.87%,网络模型复杂度仅为0.5M,表现出较高的识别准确率的同时也体现出较高的计算效率和资源利用率。

       

      Abstract: For the task of automatic recognition of machining features in 3D CAD models, existing methods have problems of information loss and insufficient generalization ability, so this paper proposes a recognition method based on an improved graph neural network.. First, the method constructs Boundary Representation (B-Rep) models into a geometric attribute adjacency graph that integrates geometric, topological, and attribute information. Precise geometric features are extracted by Convolutional Neural Network (CNN) after UV meshing, while rich attribute features are encoded by Multi-Layer Perceptron (MLP). On this basis, an improved graph neural network architecture is designed, with the core being the introduction of Graph External Attention Network (GEANet) into the graph neural network. Through a learnable external shared memory unit, this mechanism enables the model to not only learn local features within a single part but also capture and utilize common structural patterns across parts (graphs). This deepens the understanding of complex and intersecting features and effectively improves the model's generalization ability. Experiments are conducted on the MFCAD and MFCAD++ datasets. The accuracy and mIoU on the MFCAD dataset are 99.73% and 99.48% respectively, while those on the MFCAD++ dataset are 98.78% and 97.87% respectively. The network model complexity is only 0.5M, demonstrating high recognition accuracy as well as excellent computational efficiency and resource utilization.

       

    /

    返回文章
    返回