统一模型实现多类别3D异常检测,兼顾局部与全局几何信息。
MC3D-AD: A Unified Geometry-aware Reconstruction Model for Multi-category 3D Anomaly Detection
- 通过自适应几何掩码注意力捕捉不同类别的几何变化特征。
- 在真实数据集上实现3.1%和9.3%的物体级AUROC提升。
- 适合需要高效泛化能力的工业质检场景使用。
3D异常检测是保障产品制造质量的重要手段。然而,现有方法通常需为每类单独训练专用模型,成本高、效率低且泛化能力差。为此,本文提出一种统一的多类别3D异常检测模型(MC3D-AD),旨在利用局部与全局几何感知信息重建各类别正常样本表示。首先,设计自适应几何感知掩码注意力模块,提取几何变化信息以引导掩码注意力;其次,引入改进掩码注意力增强的局部几何感知编码器,编码组级别特征令牌;最后,构建基于点云位置嵌入的全局查询解码器,提升解码与重建能力。该模型生成兼具局部与全局几何感知的重构特征。在公开数据集Real3D-AD与Anomaly-ShapeNet上评估,显著优于当前最先进单类别方法,在物体级AUROC上分别提升3.1%与9.3%。代码已开源。
原文摘要 · Abstract (English)
3D Anomaly Detection (AD) is a promising means of controlling the quality of manufactured products. However, existing methods typically require carefully training a task-specific model for each category independently, leading to high cost, low efficiency, and weak generalization. Therefore, this paper presents a novel unified model for Multi-Category 3D Anomaly Detection (MC3D-AD) that aims to utilize both local and global geometry-aware information to reconstruct normal representations of all categories. First, to learn robust and generalized features of different categories, we propose an adaptive geometry-aware masked attention module that extracts geometry variation information to guide mask attention. Then, we introduce a local geometry-aware encoder reinforced by the improved mask attention to encode group-level feature tokens. Finally, we design a global query decoder that utilizes point cloud position embeddings to improve the decoding process and reconstruction ability. This leads to local and global geometry-aware reconstructed feature tokens for the AD task. MC3D-AD is evaluated on two publicly available Real3D-AD and Anomaly-ShapeNet datasets, and exhibits significant superiority over current state-of-the-art single-category methods, achieving 3.1\% and 9.3\% improvement in object-level AUROC over Real3D-AD and Anomaly-ShapeNet, respectively. The code is available at https://github.com/iCAN-SZU/MC3D-AD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。