高级检索

BRepGeom: 一种面向B-rep模型的轻量双流几何特征增强网络

BRepGeom: A Lightweight Dual-Stream Network for Geometric Enhancement of B-rep Models

  • 摘要: 边界表示(B-rep)是三维CAD模型的一种核心表示方法, 能无歧义地表达模型的几何与拓扑信息. 然而, 现有基于B-rep的模型分类与分割网络常面临几何特征提取不充分与参数规模过大的挑战. 为此, 本文提出一种轻量双流几何特征增强网络BRepGeom, 通过多层次几何特征增强机制, 有效服务于分类、分割等下游任务. 该网络核心包括: 设计多维度注意力编码器MDA-Encoder, 利用通道分组与跨维度注意力在线性复杂度下提取全局特征; 构建轻量细节增强编码器LDE-Encoder, 通过轻量卷积捕获局部几何细节; 最终动态融合全局特征与局部几何细节实现多层次特征的优化集成. 在多个公开数据集上的实验表明, BRepGeom在分类与分割任务上均达到最优性能, 验证了其高效性与有效性. 此外, 该网络几何特征提取仅需基本UV-grid输入, 为工业场景提供了一种高精度、低成本的CAD模型处理方案.

     

    Abstract: Boundary Representation (B-rep) serves as a fundamental schema for 3D CAD models, enabling precise and unambiguous encoding of both geometric and topological information. Nevertheless, existing B-rep-based net-works for model classification and segmentation often encounter challenges such as inadequate extraction of geometric features and excessive parameter sizes. To overcome these limitations, this paper introduces BRep-Geom, a lightweight dual-stream network designed for geometric feature enhancement in B-rep models. The proposed approach incorporates a multi-level geometric feature enhancement strategy, effectively supporting downstream tasks including classification and segmentation. The core contributions of BRepGeom include: a Multi-Dimensional Attention Encoder (MDA-Encoder), which leverages channel grouping and cross-dimensional attention mechanisms to efficiently extract global features under linear time complexity; a Lightweight Detail Enhancement Encoder (LDE-Encoder), which employs lightweight convolutions to capture fine-grained local geometric details; and a dynamic fusion mechanism that optimally integrates global features and local details. Comprehensive experiments on multiple public datasets demonstrate that BRepGeom achieves state-of-the-art performance in both classification and segmentation tasks, validating its efficiency and effectiveness. Furthermore, the network's geometric feature extraction relies solely on basic UV-grid input, offering a high-precision and cost-efficient solution suitable for industrial applications in CAD model processing.

     

/

返回文章
返回