爱笑的小姐姐 · 2023年03月23日 · 北京市

R2AU-Net: 基于循环残差注意力和半监督学习范式的道路裂缝分割算法

image.png

Title: A FEW-SHOT ATTENTION RECURRENT RESIDUAL U-NET FOR CRACK SEGMENTATION

Paper: https://arxiv.org/pdf/2303.01582.pdf

Code: https://github.com/ikatsamenis/CrackMap

导读

从图片中识别道路破损程度是道路养护的重要日常工作,随着人工智能领域的兴起,深度学习技术在道路基础设施的自动视觉检查中起着至关重要的作用。

image.png

为更好的动态适应用户反馈,本文主要探讨了一种关于小样本语义分割的道路裂缝自动分割算法,其基于具有循环残差和注意力模块的U-Net架构——Recurrent Residual and Attention U-Net。当一些新的校正样本被送入分类器时,采用再训练策略可以动态地微调U-Net的权重。大量实验表明,在名为CrackMap的新数据集上,所提出的小样本R2AU-Net框架在DiceIoU指标方面优于其他最先进的网络,

背景介绍

基于道路裂缝分割的工作主要可分为两类,即基于传统图像处理算法和基于现代深度学习算法。

传统方法

基于传统图像处理的方法大都是一些手工设计的模板按照一定的规律进行特征提取,如利用边缘、灰度、纹理等信息,此类方法不仅费时费力,极度依赖于专家经验,而且鲁棒性也非常差,只能适用于特定的环境下。

深度学习

基于深度学习的方法大都是基于卷积神经网络或者Transformer等架构进行端到端的特征提取。以道路裂缝分割为例,可以建模为一个二值分割网络,以原图作为输入,掩码图作为输出,通过训练自动学习出最优的参数。

image.png

方法

Framework

image.png

如图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图像中的裂缝而设计的,因此提供了对自动化和机器人驱动的维护过程至关重要的各种指标和属性的详细信息,例如几何形状、类型、方向、长度 、密度和裂纹形状。

关于语义分割这一块这里我们不会详细描述,感兴趣的小伙伴可以直接跳转到我们的语义分割大总结,强烈安利给还没看过的小伙伴,分上、下两个篇章:

image.png

image.png

Few-shot learning for segmentation refinement

image.png

如上图所示,这是一种动态整改方案,可以基于用户反馈,提供网络的整体性能。简单点讲,这其实就是增量学习的范畴,其核心思想是如何保持原有知识尽可能不被遗忘的前提下学习新的知识。

image.png

关于增量学习的更多内容解读请移步至下方链接观看。

image.png

实验

image.png

总结

今天给大家介绍的这篇工作本身没什么创新性,旨在传递给大家一个观点。就是现有的网络基础架构其实已经发展得很成熟了,无论是经典的 CNNs 还是近几年大火的 Transformer,现在新出的工作几乎都是大力出奇迹的节奏,通过堆积显卡跑消融实验发篇论文,真心不建议大多数小伙伴去卷这个方向。反而像今天这篇工作,笔者感觉真挺好,结合已有的工作探索一些新的“组合”,多去尝试解决一些实际的场景。

作者: 派派星
文章来源:CVHub

推荐阅读

更多IC设计技术干货请关注IC设计技术专栏。
迎添加极术小姐姐微信(id:aijishu20)加入技术交流群,请备注研究方向。
推荐阅读
关注数
18810
内容数
1352
嵌入式端AI,包括AI算法在推理框架Tengine,MNN,NCNN,PaddlePaddle及相关芯片上的实现。欢迎加入微信交流群,微信号:aijishu20(备注:嵌入式)
目录
极术微信服务号
关注极术微信号
实时接收点赞提醒和评论通知
安谋科技学堂公众号
关注安谋科技学堂
实时获取安谋科技及 Arm 教学资源
安谋科技招聘公众号
关注安谋科技招聘
实时获取安谋科技中国职位信息