ConvNeXt

The paper modernized a ResNet-style ConvNet using post-ViT design lessons and showed that pure convolutional backbones could remain competitive.

Metadata

FieldValue
PaperA ConvNet for the 2020s
AuthorsZhuang Liu, Hanzi Mao, Chao-Yuan Wu, Christoph Feichtenhofer, Trevor Darrell, Saining Xie
Year2022
VenueCVPR 2022
arXiv2201.03545
CVFCVPR 2022 paper
Statusverified

Question

After Vision Transformer and Swin Transformer, many strong vision results were attributed to Transformers. But Swin also reintroduced local windows, hierarchy, and dense-prediction-friendly feature maps. ConvNeXt asks a sharper question:

and how much comes from modern training recipes, stage design, large local kernels, normalization placement, and scaling choices?

The paper’s strategy is:

Main Claim

ConvNeXt shows that a pure ConvNet, modernized with design choices learned from Transformer-era vision models, can compete with hierarchical Transformers on classification, detection, and segmentation while keeping the simplicity of convolutional modules.

The durable claim is not:

It is:

ConvNeXt is therefore both a model family and a controlled architecture reading exercise.

Architecture Contract

ItemContract
Inputimage tensor
Backbone typepure convolutional hierarchical vision backbone
Starting pointResNet-style ConvNet
Main modernization targetmake ConvNet design resemble strong Transformer-era vision backbones where useful
Token mixing analoguelarge-kernel depthwise convolution
Channel mixingpointwise linear layers / convolutions
NormalizationLayerNorm-style placement in ConvNet block
Activation patternfewer activations, GELU-style nonlinearity
Output useclassification, detection, segmentation backbone

Why It Belongs in Architecture Papers

ConvNeXt is not just a performance paper. It is an architecture audit of vision backbones:

It identifies which design changes matter when comparing ConvNets and Transformers. That makes it a good paper to read after:

  1. Deep Residual Learning;
  2. Vision Transformer;
  3. Swin Transformer.

From ResNet Block to ConvNeXt Block

A simplified residual ConvNet block can be read as:

ConvNeXt keeps this residual-block view but changes the branch .

The ConvNeXt block can be summarized as:

Here is depthwise convolution. It performs local spatial mixing independently per channel, while the pointwise linear layers mix channels.

This resembles the Transformer block separation:

But token mixing is still convolutional rather than attention-based.

Depthwise Convolution as Local Token Mixing

For a feature map:

a depthwise convolution applies one spatial filter per channel:

ConvNeXt uses a larger spatial kernel than many classic CNN blocks:

The point is not only receptive field size. A large depthwise kernel behaves like cheap local token mixing:

Compared with local self-attention, depthwise convolution uses fixed learned offsets rather than input-dependent pairwise weights.

Mixing TypeWeight PatternBias
depthwise convolutionfixed offsets shared over positionslocality and translation sharing
window attentioncontent-dependent weights inside windowadaptive local interaction
global attentioncontent-dependent all-token weightslong-range interaction

Channel MLP Analogue

After spatial mixing, ConvNeXt uses pointwise channel mixing. In channel-last notation, the block resembles an MLP applied at each spatial position:

This is close to a Transformer feed-forward network:

The difference is the input layout: ConvNeXt keeps a 2D feature map and uses convolution for spatial mixing.

Macro Design

ConvNeXt updates the macro design of ResNet-like models to better match modern vision backbones.

AxisOlder ResNet HabitConvNeXt Direction
stemaggressive early convolution/poolingpatchify-like stem
stage ratioResNet-style stage depthsadjusted stage compute distribution
downsamplingResNet transition blocksseparated downsampling layers
block width/depthclassic CNN scaling habitsTransformer-era scaling comparison
dense predictionCNN feature pyramid compatibilitystill strong for detection/segmentation

The important reading point is that architecture comparisons are not only block comparisons. The training recipe and macro allocation of compute can change the conclusion.

Normalization and Activation Placement

Classic CNNs often use BatchNorm and ReLU repeatedly:

ConvNeXt moves toward a Transformer-like block style:

This matters because normalization is part of the architecture contract. A “ConvNet vs Transformer” comparison can be unfair if one side uses a more modern optimization recipe and block layout.

Relation to Swin Transformer

ConvNeXt should be read against Swin rather than against only old ResNet.

AxisSwin TransformerConvNeXt
spatial mixingshifted window attentionlarge-kernel depthwise convolution
hierarchypatch merging stagesconvolutional stages
localitylocal windowslocal convolution kernels
channel mixingMLPpointwise linear/conv layers
dense predictionstrong backbonestrong backbone
adaptive pairwise weightsyesno

ConvNeXt’s claim is strong because it tests whether a modernized ConvNet can match the practical vision-backbone advantages often credited to Transformers.

Modernization Path

The paper is easiest to reproduce as a sequence of controlled changes rather than as a single final block. The exact ordering matters because it prevents the reader from attributing every gain to the last change.

StepChangeArchitectural question
1start from a strong ResNet-style baseline and modern recipeis the baseline itself competitive?
2replace the early stem with a patchify-like convolutiondoes a less aggressive stem improve the token or feature hierarchy?
3move toward a stage ratio used by hierarchical Transformerswhere should compute be spent across resolutions?
4separate downsampling from the residual blockshould resolution changes have their own normalization and projection contract?
5use depthwise convolution with a larger kernelcan cheap local mixing provide a useful receptive field?
6change normalization and activation placementare block conventions, rather than attention, responsible for part of the gain?
7reduce unnecessary nonlinearities and use an inverted bottleneckhow should channel expansion be allocated?
8add Layer Scale and stochastic depth in the final recipedoes optimization and regularization complete the modern backbone?

The table is a reading device. It does not mean that every step is independent in the original experiments. When reproducing the paper, keep the experiment identifier, baseline, and changed variables explicit.

Full Block Contract

Let the input to one block be

In a channel-last implementation, the block can be written as:

and, with Layer Scale and stochastic depth,

Here is a learned per-channel scale initialized to a small value in the final design. The exact initialization and drop-path schedule belong to the training contract; omitting them while claiming to reproduce the final model changes more than the block diagram suggests.

The block has three distinct responsibilities:

  1. DWConv mixes spatial neighbors without mixing channels.
  2. LN changes the feature statistics presented to the channel MLP.
  3. The two pointwise projections mix channels at each spatial location.

This decomposition is useful when transferring the block to another modality. A protein residue grid, voxel field, or molecular surface may support local spatial mixing, but the meaning of the neighborhood and the boundary condition must be redefined rather than copied blindly.

Tensor Shapes Through a Stage

For a stage with spatial size and width , the residual blocks preserve the shape:

The transition between stages changes both resolution and width:

The first stem similarly maps an image to a lower-resolution feature map. A patchify-like stem can be represented as a convolution with kernel and stride :

This is not identical to a ViT patch embedding. It produces a spatial feature map and continues with convolutional operations, rather than flattening patches into a sequence for global self-attention.

For the commonly discussed tiny/small/base/large family, the stage widths and depths are scaled versions of the same macro-contract. A paper note should record the exact variant used in an experiment instead of referring to “ConvNeXt” as though it were one fixed parameter count.

ComponentShape invariant or changeWhy it matters
residual blockenables identity path and repeated feature refinement
downsampling layerallocates compute to deeper semantic stages
depthwise kernelchannel count preservedspatial mixing cost scales roughly with
expansion MLPchannel mixing dominates block parameters
classifier headfinal feature map to pooled representationtask-specific, not part of the reusable backbone

Parameter and Compute Accounting

For a depthwise convolution with kernel size and channels, the parameter count is approximately

For the two pointwise projections with expansion ratio ,

ignoring biases. For a feature map with locations, the multiply-add cost has the same leading structure:

This explains two practical facts:

  • a larger depthwise kernel increases local mixing cost linearly in and quadratically in ;
  • the inverted-bottleneck MLP can dominate parameter and arithmetic cost when is large.

The absence of a quadratic token-pair term does not automatically make the model faster than attention. Runtime also depends on memory traffic, tensor layout, kernel fusion, accelerator support, batch size, and input resolution.

Why the Inverted Bottleneck Helps

The block widens the channel dimension during the pointwise transformation:

The spatial operator itself remains depthwise, so the expanded channels do not require a full convolution across all channel pairs. This separates spatial and channel costs:

The expansion is therefore not merely a larger hidden layer. It changes where representation capacity is placed. A narrow channel interface preserves the residual shape while allowing a richer intermediate channel basis.

When implementing a variant, report whether the expansion is before or after the depthwise convolution. Moving it changes activation memory, parameter cost, and the effective block family.

Downsampling Is a Separate Contract

In many older CNNs, downsampling is folded into a strided convolution inside a residual block. ConvNeXt treats resolution change as an explicit layer between stages:

The exact ordering is important because a normalization layer sees a different distribution before and after the spatial projection. It also makes feature-pyramid extraction easier to describe:

can be exposed at well-defined resolutions for detection and segmentation heads.

For dense prediction, the backbone interface is not just the final classification vector. The relevant contract is:

OutputTypical use
early high-resolution featureedges, local texture, fine localization
middle-stage featureobject parts and medium-scale structure
late low-resolution featuresemantic context and category evidence
ordered multi-scale pyramidFPN-style detection or segmentation neck

This is one reason a hierarchical ConvNet remains attractive even when a vanilla ViT has a strong classification score.

LayerNorm, Layout, and Numerical Details

LayerNorm over channels for a feature vector is:

where

The mathematical operation is independent of whether the tensor is stored as NCHW or NHWC, but the implementation is not. A common implementation permutes NCHW to channel-last for LayerNorm and then permutes back. That can introduce layout conversions and affect measured throughput.

Checklist for a faithful block implementation:

  • confirm the normalization axes;
  • confirm whether the depthwise convolution uses groups equal to ;
  • confirm the expansion ratio and linear-layer bias;
  • confirm GELU variant and numerical precision;
  • confirm Layer Scale placement relative to DropPath;
  • confirm stage-transition normalization and stride;
  • measure both training memory and inference latency.

Training Recipe Is Part of the Comparison

ConvNeXt’s argument is specifically about a modernized ConvNet, so the training setup cannot be treated as irrelevant metadata. At minimum, record:

Training variableWhy it can change the conclusion
epoch budgetlonger optimization benefits may differ by architecture
optimizer and weight decaynormalization and residual parameterization interact with optimization
learning-rate scheduleconvergence speed and final accuracy are coupled
data augmentationstronger regularization may narrow the architecture gap
label smoothing and mixup/cutmixchanges the effective supervision signal
stochastic depthdepth-dependent regularization changes the block behavior
input resolution and crop policychanges token count, receptive-field coverage, and cost
pretraining datacan dominate the apparent backbone advantage

The fair question is therefore not “does ConvNeXt beat an old ResNet?” but:

Ablation Reading Matrix

Use the following matrix when reading the original ablation sequence or implementing a local reproduction. The right-hand column states the strongest claim that the experiment can support.

Ablation familyKeep fixedVarySupported conclusion
recipe modernizationarchitectureoptimizer, schedule, augmentationold recipe understated the baseline
stemrecipe and stage budgetstem kernel/strideearly tokenization affects the hierarchy
downsamplingstage widths and depthstransition placement and normalizationresolution changes deserve an explicit interface
kernel sizeall other block choicesdepthwise kernel sizelocal receptive-field choice matters
activation countwidth and depthReLU/GELU and placementnonlinearity placement is part of block design
bottleneck ratiostage compute budgetexpansion ratiochannel capacity and cost trade off
normalizationrecipeBatchNorm/LayerNorm and placementstatistics and optimization interact with architecture
macro stage ratiototal budgetdepth allocation by stageglobal compute allocation matters
regularizationarchitectureLayer Scale, DropPath, augmentationfinal quality is not a pure block effect

Do not read a one-variable comparison as causal if the change also alters parameter count, FLOPs, activation memory, or training stability. Add a budget-matched control when possible.

Classification Versus Dense Prediction

ConvNeXt’s architecture claim is broader than ImageNet classification, but the evidence should be separated by interface:

Classification mainly tests the quality of the final representation. Detection and segmentation additionally test spatial resolution, multi-scale alignment, and neck compatibility. A single top-1 score cannot establish the latter claims.

When comparing to Swin or ViT, record whether the downstream framework, neck, pretraining checkpoint, and fine-tuning schedule are matched. Otherwise, “backbone comparison” may actually compare an entire ecosystem.

Architecture Versus Inductive Bias

ConvNeXt does not erase the difference between convolution and attention. It narrows a practical performance gap under a particular vision regime.

PropertyConvNeXtAttention-based backbone
neighborhoodfixed local offsetscontent-dependent selected tokens
translation sharingexplicit convolutional weight sharingdepends on tokenization and positional design
interaction range per blockkernel-limitedwindow- or sequence-limited by attention pattern
locality priorstrongweaker or explicitly reintroduced
hardware primitiveconvolution/depthwise convolutionmatrix multiplication plus softmax or sparse kernels
geometry transfernatural for regular gridsrequires token/position design

The right conclusion is conditional:

The inductive bias still matters when the input is irregular, the required interaction is global, or the task benefits from content-dependent routing.

Transfer to Scientific and Biological Data

ConvNeXt is an architecture paper, so any scientific transfer should begin from the data geometry rather than from the model name.

Input representationPossible ConvNeXt-like useMain caveat
2D microscopy imageimage encoder or dense segmentation backbonebiological scale and acquisition shift
voxelized molecular fieldlocal 3D convolutional extensionresolution and rotational symmetry
protein contact or distance map2D relational map encodermap is not the full 3D object
molecular surface rasterlocal surface-feature encoderdiscretization and orientation choices
sequence or graphgenerally not a direct ConvNeXt inputregular-grid locality may be artificial

For protein or molecular coordinates, a regular-grid ConvNet can violate the desired transformation behavior. If rotations or translations should preserve the prediction, use an explicitly invariant or equivariant architecture such as EGNN or a suitable geometric model. ConvNeXt can still serve as a baseline for a rasterized representation, but it should not be presented as a geometry-preserving model by default.

Reproduction Specification

Before running a reproduction, write the specification below in a machine-readable experiment record:

FieldRequired value
variantTiny, Small, Base, Large, or custom stage configuration
inputresolution, channels, crop policy
stemkernel, stride, output width
stagesdepth and width of each stage
blockdepthwise kernel, expansion ratio, activation, normalization
transitiondownsampling kernel, stride, normalization placement
residualLayer Scale initialization and DropPath schedule
recipeoptimizer, schedule, epochs, augmentation, weight decay
precisionfp32, bf16, fp16, or mixed policy
evaluationdataset split, metric, crop/test protocol
systemaccelerator, batch size, layout, compiler/kernel settings

Minimum acceptance checks:

  1. A single block preserves .
  2. A stage transition halves spatial resolution and applies the documented width change.
  3. Depthwise convolution has groups == channels.
  4. The residual branch and identity branch have compatible layouts.
  5. Parameter and FLOP counts agree with the declared variant within the tool’s counting convention.
  6. Classification and dense-prediction feature maps are taken from the documented stages.
  7. Accuracy and throughput are reported with the same evaluation and system conditions as the baseline.

Relation to ResNet

ResNet introduced deep residual learning:

ConvNeXt keeps the residual paradigm but changes the branch design:

This makes ConvNeXt a later chapter in the ResNet family:

The paper is useful because it does not treat “CNN” as a frozen 2015 design.

Evidence to Read

ConvNeXt’s evidence should be read along three axes:

EvidenceWhat It Tests
ImageNet classificationwhether modernized ConvNet scales as a classifier
COCO detectionwhether it works as a general dense-prediction backbone
ADE20K segmentationwhether spatial hierarchy and feature maps transfer
ablation path from ResNetwhich design changes account for the gain
comparison to Swinwhether attention is necessary for the observed frontier

The ablation path is especially important. Without it, ConvNeXt would be just another high-performing backbone. With it, the paper becomes a structured architecture comparison.

What to Watch in the Ablations

Read each step as a change in one part of the architecture/training contract:

Change TypePossible Confound
training recipelonger schedule, augmentation, regularization
macro designstage ratio and compute allocation
block designkernel size, depthwise conv, expansion ratio
normalizationBatchNorm vs LayerNorm behavior
activationReLU vs GELU and activation count
scalingparameter/FLOP regime

The paper is valuable because it exposes these confounds explicitly.

Why This Matters for AI Architecture Reading

ConvNeXt teaches a general lesson:

When a new architecture claims superiority, check:

  • training recipe;
  • data scale;
  • augmentation and regularization;
  • resolution and token count;
  • macro depth/width distribution;
  • kernel or attention implementation;
  • hardware and throughput.

This applies beyond vision. The same issue appears when comparing Transformers, state-space models, GNNs, and CNNs.

Failure Modes and Caveats

  • ConvNeXt does not remove the value of attention for long-range, content-dependent interaction.
  • The paper’s conclusions are strongest for vision backbone settings, not arbitrary sequence modeling.
  • A modernized ConvNet can be competitive, but deployment speed still depends on kernels, memory format, and hardware.
  • The result depends on a modern training recipe; old CNN training baselines are not enough.
  • Large-kernel depthwise convolution is local and shared, so it has a different bias from attention even when performance is close.

Reading Checks

QuestionWhy It Matters
Is the comparison against a modernized CNN baseline?avoids strawman CNN comparisons
Are training recipes matched?separates architecture from optimization
Is the task classification or dense prediction?backbone quality may transfer differently
Is the gain from block design, stage design, or scaling?identifies the actual contribution
Does the claim require adaptive global interaction?ConvNeXt remains local and convolutional

One-Line Memory

ConvNeXt is the paper that asks whether ConvNets were obsolete or merely outdated, then modernizes ResNet into a Transformer-era convolutional backbone.