最新资讯

联系我们

  • 杭州路福智能科技有限公司
  • 朱亚平
  • 19012705965
  • lufuzn@163.com
  • 浙江省杭州市萧山区新街街道长鸣路500号 智慧交通科技园7幢401-1室

顶会论文背后的产线难题:路福、浙工大、中科院宁波材料所联手破解工业检测"小样本"困局

顶会论文背后的产线难题:路福、浙工大、中科院宁波材料所联手破解工业检测"小样本"困局

 

工业质检领域的"小样本"问题困扰学界和产业界多年。2024 年,这道题被一组中国团队给出简洁解法——ECML PKDD 2024 接收论文 *Cut-Stitch: A Simple and Effective Data Augmentation Method for Industrial Inspection*,由浙江工业大学计算机学院、中科院宁波材料所浙江省机器人与智能制造装备技术重点实验室,以及杭州路福智能科技有限公司联合完成,发表于 Springer LNAI 14944,pp. 403-420(DOI: 10.1007/978-3-031-70359-1_24),代码已开源:https://github.com/HuHaigen/Cut-Stitch。

 

工业检测的"小样本"到底有多难

 

深度卷积神经网络在工业缺陷检测中的应用价值早已被验证,但前提是有足够多、足够好的训练数据。现实恰恰相反:一条真实产线一年能产生的有效缺陷样本往往只有几百张,缺陷出现概率低、采集成本高、标注门槛高,是行业的结构性矛盾。

为缓解数据不足,研究者通常引入翻转、裁剪、MixUp、CutMix、CutOut、GridMask 等数据增强。但论文作者团队在电子元器件表面缺陷、陶瓷材料缺陷、Magnetic-tile 磁性材料缺陷、WM-811K 晶圆缺陷以及粉末冶金材料缺陷 通用目标检测等数据集上系统对比后发现:大多数为"自然场景物体"设计的增强方法,在工业缺陷场景下"水土不服",部分方法甚至让基线模型 mAP 掉点 1-5 个百分点。

原因在于三个差异:其一,工业缺陷几何约束弱、形态多样,类内方差远高于自然物体,传统几何增强会"过度扭曲";其二,缺陷显著性弱,图像直方图峰更少更平,MixUp 类融合方案会进一步叠加背景噪声,削弱缺陷显著性;其三,许多增强方案为分类任务设计,平移到检测任务时会出现严重的定位性能下降。

 

Cut-Stitch:把"切块"和"拼接"做到极致简洁

 

Cut-Stitch 的核心思路可浓缩为两句:先把图像按一定像素宽度切成条状或块状,再把切下来的块按"隔块拼接"的顺序重新组合。

具体地,论文提出两种处理路径。一是单向切割与拼接,含水平切割和垂直切割——原图被切成 N 个等宽(高)的块,序号为偶数的块组成一张新图,奇数块组成另一张;二是双向切割与拼接,先在一个维度上做一次单向操作,再在另一个维度做一次,最终从一张原图衍生出 4 张新图。论文把切割率定义为 ρ = 1/N,通过调节 ρ 可在"保持分布"和"提升多样性"之间做精细控制。

这种"剪纸式"操作之所以能成立,依据是图像像素之间天然存在大量冗余——MAE 等研究已证明,即使遮挡 75% 图像区域,模型依然能从剩余像素中恢复绝大多数信息。Cut-Stitch 的关键洞察是:不去"凭空造"新像素,也不"硬塞"其他图像的像素,而是用同一张图像的像素以新方式组合,既保留原始分布,又改变缺陷的细节形态。

论文还做了工业落地很看重的事:在 NEU-DET 上对增强后图像做 SSIM(结构相似度)、pHash(感知哈希)、LPIPS(学习感知图像块相似度)三类相似度度量。结果显示,Cut-Stitch 生成的图像在三项指标上分别保持在 0.6、0.76、0.75 以上——既不像传统方法那样"换了一张图",又提供了足够的形态扰动。

 

实验数据:在 4 个工业数据集上稳定跑赢基线

 

论文以工业界使用最广的一阶段实时检测模型 YOLOv9 为基线,并在分类任务上对比 EfficientNet-V2-S、Mobilenet-V3、ResNet-18/34、Mobile-ViT、GhostNet-V2 等轻量骨干。在分类任务上,Cut-Stitch 在 Magnetic-tile 数据集上将 EfficientNet-V2-S 的分类准确率从 79.95% 提升到 83.46%(+3.51%),其他模型获得 0.75%–1.5% 的稳定提升;在 WM-811K 晶圆缺陷数据集上,各模型获得 0.53%–0.91% 的提升。在检测任务上,NEU-DET 数据集 mAP50 从 75.48% 提升到 78.37%(+2.89%),GC-10-DET 数据集 mAP50 从 73.59% 提升到 75.45%(+1.86%),多项类别 AP 同步提升。作为对比,MixUp、GridMask、AdvancedBlur 等方法在同样基线下 mAP 普遍掉点 1–4 个百分点。最具说服力的是通用目标检测实验。在 MS COCO 这类数据量充足、目标密集、难度极高的基准上,Cut-Stitch 依然在 mAP50(56.15% → 56.88%,+0.73%)和 mAP50:95(36.28% → 36.63%,+0.35%)上取得最优结果,说明该方法是一个跨场景、跨任务的通用增强框架,而不是"为工业数据集定制的小玩具"。

论文还通过 XGradCAM 热力图做了可视化分析:传统区域丢弃类方法(CutOut、GridMask)会让模型"看不到关键特征";MixUp 类融合方法会让目标信息"更模糊";而 Cut-Stitch 通过局部块拼接,显著提升模型对目标局部细节的感知,同时保留全局轮廓信息——这与工业现场"既要看得准、又要框得稳"的需求高度一致。

 

落地的另一头:路福的产线与开源

 

论文的第三单位杭州路福智能科技有限公司,主营电子元器件、陶瓷材料、粉末冶金材料人工智能在线检测设备、机器视觉等智能检测装备的研发、生产与销售,是把 Cut-Stitch 率先在产线上验证的研究方。论文中重点讨论的 Magnetic-tile 数据集,正是路福深耕的应用场景之一。对路福而言,参与这种顶会论文的产学合作有三层价值:技术上,把论文成果反哺到产线模型迭代,提升复杂背景下的缺陷检出率;学术上,借助浙工大、中科院宁波材料所的科研力量,在国际同行评审体系中验证自家方法的有效性;品牌上,作为一家工业视觉公司,能在 ECML PKDD 2024 这类 CCF-B 类国际会议联合署名,本身就是对"我们做的是硬科技、不是集成商"的最好背书。值得一提的是,Cut-Stitch 的代码已以 MIT 协议开源在 GitHub,研究者与工程师可直接在自有数据集上复现实验、调整切割率 ρ、组合水平/垂直切割策略。这也让这套产线经验,第一次以可复用的形式走出了杭州。

 

当工业 AI 走向"以方法对抗数据稀缺"

 

过去十年,工业视觉领域长期被一种隐性的"数据焦虑"笼罩——谁的客户多、谁的数据多,谁的模型就更好。但数据终究是稀缺资源,缺陷种类、产线工况、客户隐私、数据合规,每一环都在限制数据的可获得性。Cut-Stitch 给出的信号是:在数据之外,方法论的进步依然有巨大空间。"切+拼"这种朴素到几乎像儿童手工的操作,反而在多个工业基准上稳定超越了 MixUp、CutMix、CutOut、GridMask、CLAHE、Defocus、AdvancedBlur、RandomGamma 等一长串主流方案。它的成功或许印证了那句工程老话:真正能在产线上跑得久的方法,往往是那些看上去最简单、最容易被解释、最容易在工程师之间传开的方法。而对路福、浙工大、中科院宁波材料所这一组合作团队来说,更深一层的意义在于——中国工业 AI 的研究范式,正在从"把开源论文落地"转向"从产线长出问题、再把解法写回论文"。这或许才是 Cut-Stitch 这篇论文最值得被记住的地方。

联系我们

无论您身在何处,杭州路福智能科技有限公司始终在这里等您

地址

浙江省杭州市萧山区新街街道长鸣路500号 智慧交通科技园7幢401-1室

电话

19012705965

微信扫码联系

扫码联系

验证码
联系我们

电话

19012705965

扫码咨询

扫码咨询