Gated-SCNN: Gated Shape CNNs for Semantic Segmentation

TL;DR

提出Gated-SCNN,结合门控机制的双流网络,提升城市街景分割的边界和小物体识别,mIoU提升2%,F-score提升4%。

cs.CV 🔴 高级 2019-07-12 762 引用 38 次浏览
Towaki Takikawa David Acuna Varun Jampani Sanja Fidler
语义分割 深度学习 双流网络 门控机制 边界检测

核心发现

方法论

本文提出一种基于门控机制的双流卷积神经网络(Gated-SCNN),将形状信息作为独立的处理分支,平行于传统的语义分割主干网络。核心创新在于引入门控层(Gated Convolution Layer, GCL),利用高层次的语义特征门控低层次的边界特征,有效过滤噪声,增强边界信息。形状流采用浅层架构,操作在全图像分辨率上,配合边界监督和双任务正则化,提升边界对齐和细节表现。融合模块采用多尺度Atrous Spatial Pyramid Pooling(ASPP)整合两个分支信息,最终输出高质量的分割掩码。训练过程中,结合边界二值交叉熵损失和语义正则化,优化模型性能。实验证明,该架构在Cityscapes数据集上,显著优于DeepLabv3+,在mIoU和边界F-score上分别提升2%和4%。

关键结果

  • 在Cityscapes验证集上,Gated-SCNN在mIoU指标上达到了80.8%,比DeepLabv3+的78.8%提升2%;在边界F-score指标上,提升了4%,达到83.3%,尤其在细长和小物体(如交通灯、杆子)上表现出明显优势,提升幅度达7%。
  • 在不同基础网络(ResNet-50、ResNet-101、WideResNet)上均取得优异效果,mIoU平均提升2%,边界F-score提升3.5%。多距离评估显示,模型在远距离(即较小物体)场景中表现更优,提升幅度达6%。
  • 通过引入门控机制的边界引导,模型在边界锐利度和细节还原方面优于传统方法,验证了形状信息的有效利用和门控机制的有效性。

研究意义

该研究突破了传统深度卷积网络在处理形状与纹理信息融合上的局限,通过显式分离形状信息并引入门控机制,有效提升了边界细节和小物体识别能力。这不仅改善了城市街景等复杂场景的语义分割效果,也为自动驾驶、机器人导航等应用提供了更精确的感知基础。模型在保持高效计算的同时,显著增强了对边界和细节的捕获能力,推动了语义分割技术向更高精度和鲁棒性发展。

技术贡献

本文提出的Gated-SCNN架构引入了门控卷积层(GCL),实现高层次语义信息对低层次边界特征的门控调节,显著提升了边界锐利度和小物体识别能力。该架构将形状信息作为独立的浅层流,避免了传统深层网络中信息混杂的问题。融合模块采用多尺度ASPP,有效整合两个分支信息,增强模型的多尺度感知能力。训练中引入边界正则化和双任务损失,确保边界对齐和语义一致性。整体设计兼顾模型复杂度与性能,提供了可插拔的架构方案,适用于多种主干网络。

新颖性

本研究首次将门控机制引入语义分割的双流架构中,显式分离形状与纹理信息,利用高层特征门控低层边界特征,有效过滤噪声。相比以往仅在输出或中间层融合信息的方法,提出的门控卷积层(GCL)实现了信息流的动态调节,增强了边界细节和小物体的识别能力。这一设计突破了传统深度网络在细节还原上的瓶颈,开创了利用门控机制优化多任务信息交互的新路径。

局限性

  • 模型在极端复杂场景或遮挡严重的情况下,仍可能出现边界模糊或误识别的问题,主要由于门控机制对高层特征的依赖可能受噪声影响。
  • 引入门控机制和多任务正则化增加了模型参数和训练复杂度,可能导致训练时间延长和硬件资源需求增加,不利于实时应用。
  • 目前仅在Cityscapes数据集上验证,泛化能力和跨域适应性仍需在其他场景和数据集上验证。

未来方向

未来将探索更高效的门控机制设计,减少模型复杂度,提升实时性能。还计划引入自适应门控策略,增强模型对不同场景的鲁棒性。此外,将结合弱监督和无监督学习,减少对边界标注的依赖,扩大模型的适用范围。进一步研究多尺度、多任务融合策略,提升模型在多样化场景中的表现。

AI 总览摘要

城市街景的语义分割一直是计算机视觉领域的核心挑战之一。现有的方法如DeepLabv3+在大规模场景中表现优异,但在细节还原和小物体识别方面仍存在不足。尤其是在复杂边界和微小目标的检测上,传统深度卷积网络难以兼顾全局感知与边界细节,导致预测边缘模糊、细节丢失。

为解决这一难题,本文提出了Gated-SCNN,一种结合门控机制的双流网络架构。该架构将形状信息作为独立的浅层流,与传统的语义分割主干平行处理,通过引入门控卷积层(GCL)实现高层特征对低层边界特征的动态调节。这一机制使得形状流专注于边界相关信息,过滤噪声,从而在保持全图像分辨率的同时,获得更清晰的边界细节。

核心技术在于利用高层次的语义特征作为门控信号,调节低层次边界特征的激活状态。这种设计类似于一个经验丰富的“导游”,引导浅层边界检测只关注真正重要的边缘区域。融合模块采用多尺度的Atrous Spatial Pyramid Pooling(ASPP),将两个分支的特征在不同尺度上融合,增强模型的多尺度感知能力。

在Cityscapes数据集上的大量实验证明,Gated-SCNN在多个指标上优于现有最优模型。其在mIoU指标上达到了80.8%,比DeepLabv3+提升2%;在边界F-score上也提升了4%,达到83.3%。特别是在细长和小物体(如交通灯、杆子)识别中,提升幅度高达7%。此外,该模型在远距离场景中表现更为优异,提升幅度达6%。

这一研究不仅在技术上实现了对边界细节的显著改善,也为自动驾驶、机器人导航等应用提供了更为精确的感知能力。通过显式分离和门控调节形状信息,模型在保持计算效率的同时,极大增强了对复杂场景中微小目标和细节的捕获能力。未来,研究将继续优化门控机制,提升模型的泛化能力和实时性,推动语义分割技术迈向更高的水平。

深度分析

研究背景

语义分割作为计算机视觉中的基础任务之一,经历了从传统方法到深度学习的飞跃。早期方法如FCN(Fully Convolutional Networks)实现了端到端训练,但在边界细节和小目标识别方面仍存在不足。随后,诸如DeepLab系列通过引入空洞卷积(Dilated Convolution)和多尺度特征融合(如ASP)显著提升性能。近年来,随着ResNet、DenseNet等深层网络的应用,模型在准确率上持续突破,但仍难以兼顾边界锐利和细节还原。多任务学习、边界增强和多尺度融合成为研究热点,但大多未能根本解决形状与纹理信息的有效结合问题。本文在此背景下,提出通过门控机制显式分离形状信息,突破传统深层网络的局限,推动边界细节和小物体识别的提升。

核心问题

当前深度卷积网络在语义分割中,面临边界模糊和微小目标识别不足的难题。深层网络虽然能捕获全局上下文,但在细节还原方面表现欠佳,尤其是在复杂场景中边界模糊、目标细长或远距离的小物体难以准确识别。传统方法多采用后处理(如CRF)或多尺度融合,但仍无法根本改善边界锐利度和细节还原。原因在于颜色、形状和纹理信息在网络中被混合处理,导致边界信息被噪声污染,影响识别精度。解决这一瓶颈,需在网络结构设计上实现形状信息的显式分离和高效利用,提升边界的清晰度和小目标的检测能力。

核心创新

本研究的核心创新在于引入门控卷积层(GCL),实现高层次语义特征对低层次边界特征的动态调节。具体包括:

  • �� 设计双流架构,将传统主干网络作为区域分支,新增形状流专注边界信息;
  • �� 利用门控机制(GCL)在不同层级调节两个分支的信息流,过滤噪声,增强边界细节;
  • �� 采用多尺度ASP融合两个分支的特征,提升模型对不同尺度目标的感知能力;
  • �� 引入边界正则化和双任务损失,确保边界与语义的一致性。这些创新点共同解决了传统深度网络在细节还原和小目标识别中的局限,为语义分割提供了新思路。

方法详解

  • �� 输入:城市街景图像,经过预处理后输入到双流网络;
  • �� 正常流(区域分支):采用ResNet-101或WideResNet,提取全局语义特征;
  • �� 形状流(边界分支):由浅层卷积和残差块组成,专注边界检测,配合门控卷积层(GCL)调节信息流;
  • �� 门控机制:在不同层级,将高层特征(来自正常流)与低层边界特征拼接,经过1×1卷积和sigmoid激活,生成注意力图;
  • �� GCL:利用注意力图对边界特征进行加权,过滤噪声,残差连接增强信息流;
  • �� 融合模块:采用多尺度ASPP,将两个分支的特征在不同尺度上融合,生成细节丰富的特征表示;
  • �� 训练:结合边界二值交叉熵和语义交叉熵损失,进行端到端优化,确保边界对齐和语义准确;
  • �� 输出:最终融合特征经过分类层,生成每个像素的语义类别概率。

实验设计

  • �� 数据集:Cityscapes,包含2975个训练图像、500验证图像,主要用于城市街景场景;
  • �� 基线模型:DeepLabv3+,使用ResNet-50、ResNet-101和WideResNet作为主干网络;
  • �� 训练细节:输入分辨率为800×800,采用多GPU训练、学习率1e-2的多项式衰减,训练100轮,最终在230轮取得最佳性能;
  • �� 评价指标:mIoU、边界F-score、距离变化的IoU,特别关注细长和远距离小目标的识别效果;
  • �� Ablation研究:逐步引入门控卷积层、边界正则化、多尺度融合,验证每个模块的贡献;
  • �� 超参数:门控层数3个,边界正则化系数λ1-λ4设置合理,确保多任务平衡。

结果分析

  • �� 在Cityscapes验证集上,Gated-SCNN的mIoU达到80.8%,比DeepLabv3+的78.8%提升2%;边界F-score提升4%,达到83.3%;在细长、小物体(如交通灯、杆子)识别中,提升幅度达7%;
  • �� 多基础网络实验显示,ResNet-50、ResNet-101和WideResNet上均实现性能提升,平均mIoU提升2%,边界F-score提升3.5%;
  • �� 远距离场景中,模型表现更优,提升幅度达6%,验证了边界引导机制在微小目标中的优势;
  • �� Ablation结果显示,门控卷积层和多尺度融合是性能提升的关键因素,模型对边界锐利度和细节还原具有显著改善。

应用场景

  • �� 自动驾驶:提供更精确的道路边界和障碍物检测,提升行车安全;
  • �� 机器人导航:增强环境感知能力,尤其在复杂场景中识别微小目标;
  • �� 城市规划与监控:实现高精度的街景分析,为城市管理提供支持;
  • �� 未来智能交通系统:结合实时处理,优化交通流和安全管理。

局限与展望

  • �� 目前模型在极端复杂或遮挡严重的场景中,仍存在边界模糊和误识别的问题,主要由于门控机制对高层特征的依赖受噪声影响;
  • �� 增加的门控层和多任务损失带来模型参数增长和训练时间延长,不利于实时应用;
  • �� 仅在Cityscapes数据集验证,泛化能力和跨域适应性有待进一步测试和提升。

通俗解读 非专业人士也能看懂

想象你在画一幅复杂的城市街景画。传统的方法就像用一支笔同时画出所有细节——道路、建筑、行人、交通灯,颜色、边界、纹理都在一起,虽然可以画得很丰富,但有时候会把边界画得模糊,特别是那些细长的小物体,比如交通灯或杆子。

现在,Gated-SCNN就像是请了两个画家合作:一个专门画大块区域,比如道路和建筑,另一个专注于画边界线和细节。两个画家通过一套特殊的“调节器”——门控机制——互相沟通。这个调节器会告诉边界画家,哪些地方需要特别注意,哪些可以忽略,从而让边界线变得更清晰、锐利。

这种合作让画面变得更细腻,尤其是在那些细长、微小的目标上,比如交通灯、杆子,甚至远处的小物体。最终,合成的画面既有大场景的整体感,又有细节的清晰度,就像用高精度的相机拍摄一样。这个方法的好处是,既保持了画面的整体感,又能捕捉到最细微的细节,极大地提升了城市街景的识别能力。

简单解释 像给14岁少年讲一样

想象你在玩一个超级复杂的拼图游戏。普通的拼图就像用一只手把所有碎片都拼在一起,虽然可以拼出完整的图像,但有时候边缘不够清楚,特别是那些细长的小碎片,比如交通信号灯或者电线杆子。

现在,假设你有两个朋友帮你拼图,一个专门拼大块的部分,比如天空和道路,另一个专门拼边缘和细节,比如边界线和小物体。你们用一台神奇的机器——门控机制——让边缘拼图的朋友只专注于重要的边界线,不被其他碎片干扰。这个机器会告诉他们:这里的边界很重要,要特别拼好;那里可以忽略一些细节。

这样一来,拼出来的图像就会非常清晰,边界锐利,特别是那些细长的小物体,比如交通灯、杆子,甚至远处的小车都能一清二楚。这个方法就像让两个擅长不同事情的朋友合作,用神奇的机器帮忙调节,最后拼出的城市街景既完整又细腻,远比单纯用一只手拼要厉害得多!

原文摘要

Current state-of-the-art methods for image segmentation form a dense image representation where the color, shape and texture information are all processed together inside a deep CNN. This however may not be ideal as they contain very different type of information relevant for recognition. Here, we propose a new two-stream CNN architecture for semantic segmentation that explicitly wires shape information as a separate processing branch, i.e. shape stream, that processes information in parallel to the classical stream. Key to this architecture is a new type of gates that connect the intermediate layers of the two streams. Specifically, we use the higher-level activations in the classical stream to gate the lower-level activations in the shape stream, effectively removing noise and helping the shape stream to only focus on processing the relevant boundary-related information. This enables us to use a very shallow architecture for the shape stream that operates on the image-level resolution. Our experiments show that this leads to a highly effective architecture that produces sharper predictions around object boundaries and significantly boosts performance on thinner and smaller objects. Our method achieves state-of-the-art performance on the Cityscapes benchmark, in terms of both mask (mIoU) and boundary (F-score) quality, improving by 2% and 4% over strong baselines.

cs.CV cs.LG

参考文献 (20)

Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation

Liang-Chieh Chen, Yukun Zhu, G. Papandreou 等

2018 17651 引用 ⭐ 高影响力 查看解读 →

Deep Residual Learning for Image Recognition

Kaiming He, X. Zhang, Shaoqing Ren 等

2015 238124 引用 ⭐ 高影响力 查看解读 →

Instance-Level Segmentation for Autonomous Driving with Deep Densely Connected MRFs

Ziyu Zhang, S. Fidler, R. Urtasun

2015 222 引用 查看解读 →

Very Deep Convolutional Networks for Large-Scale Image Recognition

K. Simonyan, Andrew Zisserman

2014 113881 引用 查看解读 →

Efficient Inference in Fully Connected CRFs with Gaussian Edge Potentials

Philipp Krähenbühl, V. Koltun

2011 3749 引用 查看解读 →

Are we ready for autonomous driving? The KITTI vision benchmark suite

Andreas Geiger, Philip Lenz, R. Urtasun

2012 15606 引用

An Exemplar-Based CRF for Multi-instance Object Segmentation

Xuming He, Stephen Gould

2014 54 引用

Semantic Segmentation with Boundary Neural Fields

Gedas Bertasius, Jianbo Shi, L. Torresani

2015 211 引用 查看解读 →

Semantic Image Segmentation with Task-Specific Edge Detection Using CNNs and a Discriminatively Trained Domain Transform

Liang-Chieh Chen, J. Barron, G. Papandreou 等

2015 381 引用 查看解读 →

Superpixel Convolutional Networks Using Bilateral Inceptions

Raghudeep Gadde, V. Jampani, Martin Kiefel 等

2015 136 引用 查看解读 →

Multi-Scale Context Aggregation by Dilated Convolutions

F. Yu, V. Koltun

2015 9609 引用 查看解读 →

Holistically-Nested Edge Detection

Saining Xie, Z. Tu

2015 3958 引用 查看解读 →

Fast, Exact and Multi-scale Inference for Semantic Image Segmentation with Deep Gaussian CRFs

Siddhartha Chandra, Iasonas Kokkinos

2016 191 引用 查看解读 →

The Cityscapes Dataset for Semantic Urban Scene Understanding

Marius Cordts, Mohamed Omran, Sebastian Ramos 等

2016 14010 引用 查看解读 →

Cross-Stitch Networks for Multi-task Learning

Ishan Misra, Abhinav Shrivastava, A. Gupta 等

2016 1581 引用 查看解读 →

Laplacian Pyramid Reconstruction and Refinement for Semantic Segmentation

Golnaz Ghiasi, Charless C. Fowlkes

2016 466 引用

Wide Residual Networks

Sergey Zagoruyko, N. Komodakis

2016 9066 引用 查看解读 →

A Benchmark Dataset and Evaluation Methodology for Video Object Segmentation

Federico Perazzi, J. Pont-Tuset, B. McWilliams 等

2016 2417 引用

DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFs

Liang-Chieh Chen, G. Papandreou, Iasonas Kokkinos 等

2016 21567 引用 查看解读 →

Conditional Image Generation with PixelCNN Decoders

Aäron van den Oord, Nal Kalchbrenner, O. Vinyals 等

2016 2826 引用 查看解读 →

被引用 (20)

ECAB-SegFormer: A Boundary-Aware and Efficient Channel Attention Network for Ulva prolifera Semantic Segmentation in Remote Sensing Imagery

2026 1 引用 ⭐ 高影响力

Fusion‐Guided and Distillation‐Optimized Framework for Freespace Detection in Off‐Road Environments

2026 ⭐ 高影响力

Make It Up: Fake Images, Real Gains in Generalized Few-shot Semantic Segmentation

2026 1 引用 查看解读 →

Enhancing medical image segmentation with the modification of U-shaped network

2026

A Cross-Scale Decoder with Token Refinement for Off-Road Semantic Segmentation

A dynamic pseudo-label driven dual-branch transformer framework for multi-scenario lake water body mapping

2026

BFNet: A real-time edge-deployable dual-stream boundary-aware network for defect detection of aquatic photovoltaic systems

2026

AutoMamba: Efficient Autonomous Driving Segmentation Model with Mamba

2026 2 引用

Bridging the Geometry Mismatch: Frequency-Aware Anisotropic Serialization for Thin-Structure SSMs

DPCANet: detail-preserving cross-scale alignment for real-time small-object detection in unmanned aerial vehicle imagery

2026

DBBGNet: a dual-branch network with boundary information guidance for real-time semantic segmentation

2026 1 引用

TSNet: Three-branch scale-aware network for real-time semantic segmentation

2026

SynCAFG-Net: Synergistic Cross-Attention Feature Generation for High-Accuracy Lightweight Image Classification

2026

FMSNet: frequency-domain aware and multi-scale edge enhancement network for remote sensing land cover image segmentation

2026

YOLO-GBPA: Enhanced YOLO11 with Gated Bilinear Pyramid Aggregation for Dust Detection

2026

Enhanced remote sensing image segmentation via dual-branch uncertainty-aware haar-enhanced network

2026

MPFT-UNet: A Boundary-Refined and Multi-Scale Dynamic Fusion Network for UAV-Based Port Ship Segmentation

2026 1 引用

OmniISR: A Unified Framework for Centralized and Federated Learning via Intermediate Supervision and Regularization

Coarse-to-Fine Domain Incremental Learning with Attentive Distillation for Mining Footprint Segmentation in Multispectral Imagery

GABI: Geometry-Aware Boundary Integration for Spacecraft Segmentation

2026 1 引用 查看解读 →