InternImage:利用可变形卷积探索大规模视觉基础模型

InternImage: Exploring Large-Scale Vision Foundation Models with Deformable Convolutions

王文海 Wenhai Wang · Shanghai AI Lab · 2022-11-10 · arXiv:2211.05778 ↗ · 被引 1127

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

与近年来大规模视觉 Transformer(ViT)的巨大进步相比,基于卷积神经网络(CNN)的大规模模型仍处于早期阶段。本文提出了一种新的大规模 CNN 基础模型,称为 InternImage,它可以像 ViT 一样从增加参数和训练数据中获益。与最近关注大型密集核的 CNN 不同,InternImage 以可变形卷积为核心算子,因此我们的模型不仅具有检测和分割等下游任务所需的大有效感受野,还具有由输入和任务信息调节的自适应空间聚合能力。因此,所提出的 InternImage 减少了传统 CNN 的严格归纳偏置,使得从海量数据中学习更强、更鲁棒的具有大规模参数的模型成为可能,就像 ViT 一样。我们的模型在包括 ImageNet、COCO 和 ADE20K 在内的具有挑战性的基准测试中证明了其有效性。值得一提的是,InternImage-H 在 COCO test-dev 上达到了 65.4 mAP 的新纪录,在 ADE20K 上达到了 62.9 mIoU,超越了当前领先的 CNN 和 ViT。代码将在 https://github.com/OpenGVLab/InternImage 发布。

Compared to the great progress of large-scale vision transformers (ViTs) in recent years, large-scale models based on convolutional neural networks (CNNs) are still in an early state. This work presents a new large-scale CNN-based foundation model, termed InternImage, which can obtain the gain from increasing parameters and training data like ViTs. Different from the recent CNNs that focus on large dense kernels, InternImage takes deformable convolution as the core operator, so that our model not only has the large effective receptive field required for downstream tasks such as detection and segmentation, but also has the adaptive spatial aggregation conditioned by input and task information. As a result, the proposed InternImage reduces the strict inductive bias of traditional CNNs and makes it possible to learn stronger and more robust patterns with large-scale parameters from massive data like ViTs. The effectiveness of our model is proven on challenging benchmarks including ImageNet, COCO, and ADE20K. It is worth mentioning that InternImage-H achieved a new record 65.4 mAP on COCO test-dev and 62.9 mIoU on ADE20K, outperforming current leading CNNs and ViTs. The code will be released at https://github.com/OpenGVLab/InternImage.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 27)

阅读逐段中英对照全文 →