Skip to main navigation Skip to search Skip to main content

Are Large Kernels Better Teachers than Transformers for ConvNets?

Research output: Chapter in Book/Report/Conference proceeding › Conference contribution › Academic › peer-review

164 Downloads (Pure)

Abstract

This paper reveals a new appeal of the recently emerged large-kernel Convolutional Neural Networks (ConvNets): as the teacher in Knowledge Distillation (KD) for small-kernel ConvNets. While Transformers have led state-of-the-art (SOTA) performance in various fields with ever-larger models and labeled data, small-kernel ConvNets are considered more suitable for resource-limited applications due to the efficient convolution operation and compact weight sharing. KD is widely used to boost the performance of small-kernel ConvNets. However, previous research shows that it is not quite effective to distill knowledge (e.g., global information) from Transformers to small-kernel ConvNets, presumably due to their disparate architectures. We hereby carry out a first-of-its-kind study unveiling that modern large-kernel ConvNets, a compelling competitor to Vision Transformers, are remarkably more effective teachers for small-kernel ConvNets, due to more similar architectures. Our findings are backed up by extensive experiments on both logit-level and feature-level KD ``out of the box", with no dedicated architectural nor training recipe modifications. Notably, we obtain the \textbf{best-ever pure ConvNet} under 30M parameters with \textbf{83.1\%} top-1 accuracy on ImageNet, outperforming current SOTA methods including ConvNeXt V2 and Swin V2. We also find that beneficial characteristics of large-kernel ConvNets, e.g., larger effective receptive fields, can be seamlessly transferred to students through this large-to-small kernel distillation. Code is available at: \url{https://github.com/VITA-Group/SLaK}.
Original languageEnglish
Title of host publication40th International Conference on Machine Learning, ICML 2023
EditorsAndreas Krause, Emma Brunskill, Kyunghyun Cho, Barbara Engelhardt, Sivan Sabato, Jonathan Scarlett
PublisherPMLR
Pages14023-14038
Number of pages16
Publication statusPublished - 2023
Event40th International Conference on Machine Learning, ICML 2023 - Hawaii Convention Center , Hawaii, United States
Duration: 23 Jul 2023 → 29 Jul 2023
Conference number: 40

Publication series

NameProceedings of Machine Learning Research (PMLR)
Volume202
ISSN (Electronic)2640-3498

Conference

Conference40th International Conference on Machine Learning, ICML 2023
Abbreviated titleICML
Country/TerritoryUnited States
CityHawaii
Period23/07/23 → 29/07/23

Bibliographical note

Accepted by ICML 2023

Fingerprint

Dive into the research topics of 'Are Large Kernels Better Teachers than Transformers for ConvNets?'. Together they form a unique fingerprint.

Cite this