atlas

Research map / Architectures and efficiency / Attention and transformers

ParameterNet: Parameters Are All You Need for Large-scale Visual Pretraining of Mobile Networks

CVPR 2024 · Architectures and efficiency › Attention and transformers · approach: Token merging and masked pretraining · 101 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Identifies a "low FLOPs pitfall" where low-FLOPs networks don't benefit from ImageNet-22K pretraining, and proposes ParameterNet, which adds parameters via dynamic convolution at near-constant FLOPs, plus a LLaMA-1B MoE-style extension for language.

Problem

Low-FLOPs mobile vision networks fail to benefit from large-scale supervised pretraining (ImageNet-22K), as their accuracy gains do not follow the pretraining data scale.

Approach

Increase the number of parameters of low-FLOPs networks without increasing FLOPs by replacing standard convolutions with dynamic convolutions (input-conditioned mixture of M expert kernels), so that large-scale pretraining helps mobile-scale models.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Pretrain GhostNet-300M on ImageNet-22K instead of ImageNet-1KGhostNet-300M · ImageNet-1K valTop-1 accuracy: 76.3 -> 75.7 (-0.6)Hurt
Add dynamic convolution with 4 experts and pretrain on ImageNet-22KGhostNet-300M · ImageNet-1K valTop-1 accuracy: 77.2 -> 79.1 (+1.9)Helped
Pretrain GhostNet-600M on ImageNet-22K instead of ImageNet-1KGhostNet-600M · ImageNet-1K valTop-1 accuracy: 78.6 -> 78.3 (-0.3)Hurt
Add dynamic convolution with 4 experts and pretrain on ImageNet-22KGhostNet-600M · ImageNet-1K valTop-1 accuracy: 79.0 -> 81.6 (+2.6)Helped
Add sparse-activated MoE experts to the LLaMA-1B up projectionLLaMA-1B · Average across language tasksAccuracy: 58.19 -> 60.56Helped
Increase dynamic experts from 1 to 8 (ablation)GhostNet-300M · ImageNet-1K val, ImageNet-1K pretrainingTop-1 accuracy: 76.3 -> 77.7Helped
Increase dynamic experts from 1 to 8 (ablation)GhostNet-300M · ImageNet-1K val, ImageNet-22K pretrainingTop-1 accuracy: 75.7 -> 79.4Helped
Apply ParameterNet to Swin-300M (ablation)Swin-300M · ImageNet-1K val, ImageNet-22K pretrainingTop-1 accuracy: 67.0 (-2.2) -> 74.5 (+2.2)Helped
Apply ParameterNet to EfficientNetV2-B0 (ablation)EfficientNetV2-B0 · ImageNet-1K val, ImageNet-22K pretrainingTop-1 accuracy: 77.6 (-1.1) -> 81.5 (+1.7)Helped
Replace dynamic convolution with re-parameterized convolution (ablation)GhostNet-300M · ImageNet-1K val, ImageNet-22K pretrainingTop-1 accuracy: 79.1 -> 76.9Hurt
Increase dynamic experts from 2 to 4 (ablation)GhostNet-300M · ImageNet-1K val, ImageNet-22K pretrainingTop-1 accuracy: 77.7 -> 79.1Helped
Increase dynamic experts from 4 to 8 (ablation)GhostNet-300M · ImageNet-1K val, ImageNet-22K pretrainingTop-1 accuracy: 79.1 -> 79.4Helped

Surprising findings

Limitations

Experimental setup

Datasets: ImageNet-22K ImageNet-1K C4 Wikipedia ArXiv

Benchmarks: ImageNet-1K ARC (easy) BoolQ SST-2 HellaSwag

Metrics: top-1 accuracy FLOPs parameters latency training loss zero-shot accuracy

Baselines: GhostNet Swin Transformer EfficientNetV2 ResNet50 (BiT) ResNet50 (Billion-scale) ResNet101 (JFT-300M) EfficientNet-B0 LLaMA-1B

Builds on: GhostNet dynamic convolution (CondConv/DyNet) Switch Transformer EfficientNetV2 Swin Transformer LLaMA

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy