HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

UniRepLKNet: A Universal Perception Large-Kernel ConvNet for Audio, Video, Point Cloud, Time-Series and Image Recognition

Xiaohan Ding Yiyuan Zhang Yixiao Ge Sijie Zhao Lin Song Xiangyu Yue Ying Shan

UniRepLKNet: A Universal Perception Large-Kernel ConvNet for Audio, Video, Point Cloud, Time-Series and Image Recognition

Abstract

Large-kernel convolutional neural networks (ConvNets) have recently received extensive research attention, but two unresolved and critical issues demand further investigation. 1) The architectures of existing large-kernel ConvNets largely follow the design principles of conventional ConvNets or transformers, while the architectural design for large-kernel ConvNets remains under-addressed. 2) As transformers have dominated multiple modalities, it remains to be investigated whether ConvNets also have a strong universal perception ability in domains beyond vision. In this paper, we contribute from two aspects. 1) We propose four architectural guidelines for designing large-kernel ConvNets, the core of which is to exploit the essential characteristics of large kernels that distinguish them from small kernels - they can see wide without going deep. Following such guidelines, our proposed large-kernel ConvNet shows leading performance in image recognition (ImageNet accuracy of 88.0%, ADE20K mIoU of 55.6%, and COCO box AP of 56.4%), demonstrating better performance and higher speed than the recent powerful competitors. 2) We discover large kernels are the key to unlocking the exceptional performance of ConvNets in domains where they were originally not proficient. With certain modality-related preprocessing approaches, the proposed model achieves state-of-the-art performance on time-series forecasting and audio recognition tasks even without modality-specific customization to the architecture. All the code and models are publicly available on GitHub and Huggingface.

Code Repositories

ailab-cvc/unireplknet
Official
pytorch
Mentioned in GitHub
Westlake-AI/openmixup
pytorch
Mentioned in GitHub
chenller/mmseg-extension
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
image-classification-on-imagenetUniRepLKNet-B++
Top 1 Accuracy: 87.4%
image-classification-on-imagenetUniRepLKNet-S++
Top 1 Accuracy: 86.4%
image-classification-on-imagenetUniRepLKNet-T
Top 1 Accuracy: 83.2%
image-classification-on-imagenetUniRepLKNet-XL++
Top 1 Accuracy: 88%
image-classification-on-imagenetUniRepLKNet-L++
Top 1 Accuracy: 87.9%
image-classification-on-imagenetUniRepLKNet-S
Top 1 Accuracy: 83.9%
image-classification-on-imagenetUniRepLKNet-N
Top 1 Accuracy: 81.6%
image-classification-on-imagenetUniRepLKNet-P
Top 1 Accuracy: 80.2%
image-classification-on-imagenetUniRepLKNet-A
Top 1 Accuracy: 77%
image-classification-on-imagenetUniRepLKNet-F
Top 1 Accuracy: 78.6%
object-detection-on-coco-2017UniRepLKNet-S++
mAP: 54.3
object-detection-on-coco-2017UniRepLKNet-T
mAP: 51.7
object-detection-on-coco-2017UniRepLKNet-B++
mAP: 54.8
object-detection-on-coco-2017UniRepLKNet-S
mAP: 53
object-detection-on-coco-2017UniRepLKNet-XL++
mAP: 56.4
object-detection-on-coco-2017UniRepLKNet-L++
mAP: 55.8
semantic-segmentation-on-ade20kUniRepLKNet-T
Validation mIoU: 49.1
semantic-segmentation-on-ade20kUniRepLKNet-L++
Validation mIoU: 55
semantic-segmentation-on-ade20kUniRepLKNet-B++
Validation mIoU: 53.9
semantic-segmentation-on-ade20kUniRepLKNet-S++
Validation mIoU: 52.7
semantic-segmentation-on-ade20kUniRepLKNet-XL
Validation mIoU: 55.6
semantic-segmentation-on-ade20kUniRepLKNet-S
Validation mIoU: 51

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp