The stem of the AnyNet model can be implemented as a sequential block containing a $$3 \times 3$$ convolution with a stride of $$2$$ and padding, followed by batch normalization and a ReLU activation function. This sequence halves the input resolution and generates the initial channels.

PyTorch Implementation:
```python
class AnyNet(d2l.Classifier):
    def stem(self, num_channels):
        return nn.Sequential(
            nn.LazyConv2d(num_channels, kernel_size=3, stride=2, padding=1),
            nn.LazyBatchNorm2d(), nn.ReLU())
```

MXNet Implementation:
```python
class AnyNet(d2l.Classifier):
    def stem(self, num_channels):
        net = nn.Sequential()
        net.add(nn.Conv2D(num_channels, kernel_size=3, padding=1, strides=2),
                nn.BatchNorm(), nn.Activation('relu'))
        return net
```

JAX Implementation:
```python
class AnyNet(d2l.Classifier):
    arch: tuple
    stem_channels: int
    lr: float = 0.1
    num_classes: int = 10
    training: bool = True

    def setup(self):
        self.net = self.create_net()

    def stem(self, num_channels):
        return nn.Sequential([
            nn.Conv(num_channels, kernel_size=(3, 3), strides=(2, 2),
                    padding=(1, 1)),
            nn.BatchNorm(not self.training),
            nn.relu
        ])
```

TensorFlow Implementation:
```python
class AnyNet(d2l.Classifier):
    def stem(self, num_channels):
        return tf.keras.models.Sequential([
            tf.keras.layers.Conv2D(num_channels, kernel_size=3, strides=2,
                                   padding='same'),
            tf.keras.layers.BatchNormalization(),
            tf.keras.layers.Activation('relu')])
```

AnyNet Stem Code Implementation

The stem of an AnyNet architecture serves as the initial processing stage for RGB input images ($$3$$ channels). It typically applies a $$3 	imes 3$$ convolution with a stride of $$2$$, followed by batch normalization. This sequence halves the spatial resolution of the image (from $$r 	imes r$$ to $$r/2 	imes r/2$$) and generates an initial set of $$c_0$$ channels that serve as the input for the network's body.

Claude

The AnyNet design space provides a generic architectural template for constructing and exploring families of convolutional neural networks. This macro-structure is composed of three primary components: a stem that performs the initial image processing, a body that carries out the bulk of the transformations to build object representations across multiple stages, and a head that converts these representations into the final desired outputs (e.g., via a softmax regressor). Within this generic framework, design choices such as stage depth, channel counts, and block structures (often using ResNeXt blocks) parameterize a vast space of potential network configurations.

AnyNet Design Space

Dive into Deep Learning

AnyNet Stem

The body of an AnyNet architecture conducts the majority of feature extraction by operating on images at progressively decreasing resolutions. Typically designed to process ImageNet inputs (e.g., reducing a $$224 \times 224 \times 3$$ image to a $$7 \times 7 \times c_4$$ representation), the body is divided into multiple stages (e.g., $$4$$ stages). Each stage quarters the total spatial resolution (by halving both height and width) and is constructed from a sequence of ResNeXt blocks. The initial block in a stage applies a stride of $$2$$ to downsample the feature map, which necessitates a $$1 \times 1$$ convolution on the residual branch to match dimensions, while all subsequent blocks in the stage maintain constant resolution and channel counts.

AnyNet Body

The head of an AnyNet architecture converts the feature representations extracted by the network body into the final desired predictions. It employs an entirely standard design consisting of global average pooling to collapse spatial dimensions, followed by a fully connected layer to emit an $$n$$-dimensional vector for $$n$$-class classification.

Learn Before

Related

Learn After