Title: A FEW-SHOT ATTENTION RECURRENT RESIDUAL U-NET FOR CRACK SEGMENTATION
导读
从图片中识别道路破损程度是道路养护的重要日常工作,随着人工智能领域的兴起,深度学习技术在道路基础设施的自动视觉检查中起着至关重要的作用。
为更好的动态适应用户反馈,本文主要探讨了一种关于小样本语义分割的道路裂缝自动分割算法,其基于具有循环残差和注意力模块的U-Net
架构——Recurrent Residual and Attention U-Net
。当一些新的校正样本被送入分类器时,采用再训练策略可以动态地微调U-Net
的权重。大量实验表明,在名为CrackMap
的新数据集上,所提出的小样本R2AU-Net
框架在Dice
和IoU
指标方面优于其他最先进的网络,
背景介绍
基于道路裂缝分割的工作主要可分为两类,即基于传统图像处理算法和基于现代深度学习算法。
传统方法
基于传统图像处理的方法大都是一些手工设计的模板按照一定的规律进行特征提取,如利用边缘、灰度、纹理等信息,此类方法不仅费时费力,极度依赖于专家经验,而且鲁棒性也非常差,只能适用于特定的环境下。
深度学习
基于深度学习的方法大都是基于卷积神经网络或者Transformer
等架构进行端到端的特征提取。以道路裂缝分割为例,可以建模为一个二值分割网络,以原图作为输入,掩码图作为输出,通过训练自动学习出最优的参数。
方法
Framework
如图1(a)所示,本文基于U-Net网络并结合循环残差和注意力模块提出了一种适用于道路裂缝分割的网络,称为R2AU-Net
。与标准U-Net
相比,R2AU-Net 结合了递归残差卷积层,可确保更好地表示分割任务的特征和注意力区域,以突出显示通过跳跃连接传递的显着特征。
此外,本文针对以往方法仅能从以训练的标注数据中学习的限制,引入了一种基于 R2AU-Net 的半监督学习范式few-shot refinement
方案,它能够根据用户的反馈灵活的调整模型的行为和权重,进一步提高分割性能,如图1(b)所示.
R2AU-Net
R2AU-Net 这块没什么好讲的,拆开看就是一个Recurrent Residual U-Net
+ Attention U-Net
的超级缝合怪。作者强调该“魔改”模型是特意针对分割RGB
图像中的裂缝而设计的,因此提供了对自动化和机器人驱动的维护过程至关重要的各种指标和属性的详细信息,例如几何形状、类型、方向、长度 、密度和裂纹形状。
关于语义分割这一块这里我们不会详细描述,感兴趣的小伙伴可以直接跳转到我们的语义分割大总结,强烈安利给还没看过的小伙伴,分上、下两个篇章:
Few-shot learning for segmentation refinement
如上图所示,这是一种动态整改方案,可以基于用户反馈,提供网络的整体性能。简单点讲,这其实就是增量学习的范畴,其核心思想是如何保持原有知识尽可能不被遗忘的前提下学习新的知识。
关于增量学习的更多内容解读请移步至下方链接观看。
实验
总结
今天给大家介绍的这篇工作本身没什么创新性,旨在传递给大家一个观点。就是现有的网络基础架构其实已经发展得很成熟了,无论是经典的 CNNs 还是近几年大火的 Transformer,现在新出的工作几乎都是大力出奇迹的节奏,通过堆积显卡跑消融实验发篇论文,真心不建议大多数小伙伴去卷这个方向。反而像今天这篇工作,笔者感觉真挺好,结合已有的工作探索一些新的“组合”,多去尝试解决一些实际的场景。
作者: 派派星
文章来源:CVHub
推荐阅读
- 清华&天津大学新作 | 面向自动驾驶场景的纯视觉 3D 语义占据预测 (已开源)
- IC各细分领域公司简介以及薪资调研(GPU篇)
- 会写VIP,会提取testplan,GPT进化的太快了!!!
- 搞芯片不懂cell别乱用!看看Vt cell都是啥?
- ISSCC IC设计论文赏析:关于无线充电、汽车网关SoC
更多IC设计技术干货请关注IC设计技术专栏。
迎添加极术小姐姐微信(id:aijishu20)加入技术交流群,请备注研究方向。