Efficient Deformable ConvNets: Rethinking Dynamic and Sparse Operator for Vision Applications
来源: 上海人工智能实验室|2024-06-13
Efficient Deformable ConvNets: Rethinking Dynamic and Sparse
Operator for Vision Applications
Yuwen Xiong∗1,2 Zhiqi Li∗3,2 Yuntao Chen∗4 Feng Wang∗5
Xizhou Zhu5,6 Jiapeng Luo6 Wenhai Wang7,2 Tong Lu3
Hongsheng Li7 Yu Qiao2 Lewei Lu6 Jie Zhou5 Jifeng Dai5,2
1University of Toronto 2OpenGVLab, Shanghai AI Laboratory
3Nanjing University 4CAIR, HKISI, CAS 5Tsinghua University
6SenseTime Research 7The Chinese University of Hong Kong
Abstract
We introduce Deformable Convolution v4 (DCNv4), a highly efficient and effective operator designed for a broad spectrum of vision applications. DCNv4 addresses the limitations of its predecessor, DCNv3, with two key enhancements: 1. removing softmax normalization in spatial aggregation to enhance its dynamic property and expressive power and 2.optimizing memory access to minimize redundant operations for speedup. These improvements result in a significantly faster convergence compared to DCNv3 and a substantial increase in processing speed, with DCNv4 achieving more than three times the forward speed. DCNv4 demonstrates exceptional performance across various tasks, including image classification, instance and semantic segmentation, and notably, image generation. When integrated into generative models like U-Net in the latent diffusion model, DCNv4 outperforms its baseline, underscoring its possibility to enhance generative models. In practical applications, replacing DCNv3 with DCNv4 in the InternImage model to create FlashInternImage results in up to 80% speed increase and further performance improvement without further modifications. The advancements in speed and efficiency of DCNv4, combined with its robust performance across diverse vision tasks, show its potential as a foundational building block for future vision models.

