3 个月前

FasterSeg:搜索更快速的实时语义分割

FasterSeg:搜索更快速的实时语义分割

摘要

我们提出FasterSeg,一种通过自动设计实现的语义分割网络,不仅在性能上达到当前最优水平,同时在推理速度上也优于现有方法。FasterSeg基于神经架构搜索(NAS)技术,在一个新颖且更广阔的搜索空间中被发现,该空间融合了多分辨率分支结构——这一设计要素在人工设计的分割模型中已被证实至关重要。为更有效地平衡高精度与低延迟之间的权衡,我们提出了一种解耦且细粒度的延迟正则化方法,有效克服了在搜索过程中观察到的“坍缩”现象:即所搜索出的网络容易退化为低延迟但精度较差的模型。此外,我们无缝地将FasterSeg扩展至一种新型协同搜索(co-searching)框架,可在单次运行中同时搜索一个教师网络与一个学生网络。通过教师-学生知识蒸馏机制,进一步提升了学生模型的精度。在多个主流语义分割基准上的实验结果充分验证了FasterSeg的卓越性能。例如,在Cityscapes数据集上,FasterSeg的运行速度比最接近的人工设计竞争对手快超过30%,同时保持了相当的精度水平。

代码仓库

gaussianer/fasterseg
pytorch
GitHub 中提及
TAMU-VITA/FasterSeg
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
real-time-semantic-segmentation-on-cityscapes-1FasterSeg
Frame (fps): 163.9
mIoU: 73.1
semantic-segmentation-on-bddFasterSeg
mIoU: 55.1
semantic-segmentation-on-cityscapesFasterSeg
Mean IoU (class): 71.5%
semantic-segmentation-on-cityscapes-valFasterSeg
mIoU: 73.1%

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
FasterSeg:搜索更快速的实时语义分割 | 论文 | HyperAI超神经