4 个月前

同时增强:通过IBN-Net提升学习和泛化能力

同时增强:通过IBN-Net提升学习和泛化能力

摘要

卷积神经网络(CNNs)在许多计算机视觉问题中取得了显著的成功。与现有工作不同的是,这些工作设计的CNN架构仅针对单一领域的单一任务进行优化且不具备泛化能力,我们提出了一种新颖的卷积架构——IBN-Net,该网络不仅显著增强了CNN在一个领域(例如Cityscapes)上的建模能力,同时在另一个领域(例如GTA5)上也具有出色的泛化能力,而无需进行微调。IBN-Net巧妙地将实例归一化(Instance Normalization, IN)和批归一化(Batch Normalization, BN)作为构建模块集成在一起,并可以嵌入多种先进的深度网络以提升其性能。本研究有三个关键贡献:(1) 通过深入研究IN和BN,我们揭示了IN学习到的特征对颜色、风格和虚拟/现实等外观变化具有不变性,而BN对于保留与内容相关的信息至关重要。(2) IBN-Net可以应用于多种先进的深度架构,如DenseNet、ResNet、ResNeXt和SENet,并且在不增加计算成本的情况下持续提升它们的性能。(3) 当将训练好的网络应用于新领域时,例如从GTA5到Cityscapes,IBN-Net能够实现与领域适应方法相当的改进效果,即使没有使用目标领域的数据。凭借IBN-Net,我们在WAD 2018挑战赛中的可行驶区域赛道上获得了第一名,mIoU达到86.18%。

代码仓库

alibaba/cluster-contrast
pytorch
GitHub 中提及
yxgeee/MMT
pytorch
GitHub 中提及
leeBooMla/ICSR
pytorch
GitHub 中提及
WangWenhao0716/DomainMix
pytorch
GitHub 中提及
theziqi/dccc
pytorch
GitHub 中提及
Mind23-2/MindCode-52
mindspore
GitHub 中提及
alibaba/cluster-contrast-reid
pytorch
GitHub 中提及
XingangPan/IBN-Net
官方
pytorch
GitHub 中提及
osmr/imgclsmob
mxnet
GitHub 中提及
bupt-ai-cz/hhcl-reid
pytorch
GitHub 中提及
wangguangyuan/ClusterContrast
pytorch
GitHub 中提及
yxgeee/SpCL
pytorch
GitHub 中提及
wangyuan249/Mymmt767
pytorch
GitHub 中提及
leeboomla/calr
pytorch
GitHub 中提及
jihaoxuanye/MetaPRD
pytorch
GitHub 中提及

基准测试

基准方法指标
all-day-semantic-segmentation-on-all-dayIB-Net
mIoU: 64.5
domain-generalization-on-gta-to-avgIBN
mIoU: 34.63
robust-object-detection-on-dwdIBN-Net
mPC [AP50]: 25.5

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
同时增强:通过IBN-Net提升学习和泛化能力 | 论文 | HyperAI超神经