用超图计算增强YOLO,提升工业缺陷检测的多尺度感知能力
HyperDefect-YOLO: Enhance YOLO with HyperGraph Computation for Industrial Defect Detection
- 引入超图机制捕捉缺陷特征的高阶关联
- 在多个数据集上优于当前最优方法,显著提升小缺陷检测率
- 适合需要高精度缺陷定位的工业质检场景
制造业中的缺陷检测是关键但具有挑战性的任务,旨在识别生产过程中产生的缺陷。尽管传统YOLO模型在缺陷检测中表现良好,但在捕捉高阶特征关系方面仍存在局限,难以应对复杂场景和多尺度缺陷。为此,本文提出将超图计算引入YOLO框架,命名为HyperDefect-YOLO(HD-YOLO),以增强表征能力和语义挖掘。HD-YOLO在主干网络中集成缺陷感知模块(DAM)与混合图网络(MGNet),分别专注于缺陷特征的感知与提取;提出超图聚合网络(HGANet),结合超图与注意力机制实现多尺度特征的有效聚合;设计跨尺度融合(CSF)模块,替代简单的拼接与卷积,实现自适应特征融合;在颈部引入语义感知模块(SAM),增强对不同尺寸缺陷在干扰背景下的定位能力。HD-YOLO在公开数据集HRIPCB和NEU-DET上进行了严格评估,性能显著优于当前先进方法。同时,在真实工业场景下采集的MINILED数据集上验证了其有效性。代码已开源。
原文摘要 · Abstract (English)
In the manufacturing industry, defect detection is an essential but challenging task aiming to detect defects generated in the process of production. Though traditional YOLO models presents a good performance in defect detection, they still have limitations in capturing high-order feature interrelationships, which hurdles defect detection in the complex scenarios and across the scales. To this end, we introduce hypergraph computation into YOLO framework, dubbed HyperDefect-YOLO (HD-YOLO), to improve representative ability and semantic exploitation. HD-YOLO consists of Defect Aware Module (DAM) and Mixed Graph Network (MGNet) in the backbone, which specialize for perception and extraction of defect features. To effectively aggregate multi-scale features, we propose HyperGraph Aggregation Network (HGANet) which combines hypergraph and attention mechanism to aggregate multi-scale features. Cross-Scale Fusion (CSF) is proposed to adaptively fuse and handle features instead of simple concatenation and convolution. Finally, we propose Semantic Aware Module (SAM) in the neck to enhance semantic exploitation for accurately localizing defects with different sizes in the disturbed background. HD-YOLO undergoes rigorous evaluation on public HRIPCB and NEU-DET datasets with significant improvements compared to state-of-the-art methods. We also evaluate HD-YOLO on self-built MINILED dataset collected in real industrial scenarios to demonstrate the effectiveness of the proposed method. The source codes are at https://github.com/Jay-zzcoder/HD-YOLO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。