{ "cells": [ { "cell_type": "markdown", "metadata": { "origin_pos": 0 }, "source": [ "# Network in Network (NiN)\n", ":label:`sec_nin`\n", "\n", "LeNet, AlexNet, and VGG all share a common design pattern:\n", "extract features exploiting *spatial* structure\n", "via a sequence of convolution and pooling layers\n", "and then post-process the representations via fully-connected layers.\n", "The improvements upon LeNet by AlexNet and VGG mainly lie\n", "in how these later networks widen and deepen these two modules.\n", "Alternatively, one could imagine using fully-connected layers\n", "earlier in the process.\n", "However, a careless use of dense layers might give up the\n", "spatial structure of the representation entirely,\n", "*network in network* (*NiN*) blocks offer an alternative.\n", "They were proposed based on a very simple insight:\n", "to use an MLP on the channels for each pixel separately :cite:`Lin.Chen.Yan.2013`.\n", "\n", "\n", "## (**NiN Blocks**)\n", "\n", "Recall that the inputs and outputs of convolutional layers\n", "consist of four-dimensional tensors with axes\n", "corresponding to the example, channel, height, and width.\n", "Also recall that the inputs and outputs of fully-connected layers\n", "are typically two-dimensional tensors corresponding to the example and feature.\n", "The idea behind NiN is to apply a fully-connected layer\n", "at each pixel location (for each height and width).\n", "If we tie the weights across each spatial location,\n", "we could think of this as a $1\\times 1$ convolutional layer\n", "(as described in :numref:`sec_channels`)\n", "or as a fully-connected layer acting independently on each pixel location.\n", "Another way to view this is to think of each element in the spatial dimension\n", "(height and width) as equivalent to an example\n", "and a channel as equivalent to a feature.\n", "\n", ":numref:`fig_nin` illustrates the main structural differences\n", "between VGG and NiN, and their blocks.\n", "The NiN block consists of one convolutional layer\n", "followed by two $1\\times 1$ convolutional layers that act as\n", "per-pixel fully-connected layers with ReLU activations.\n", "The convolution window shape of the first layer is typically set by the user.\n", "The subsequent window shapes are fixed to $1 \\times 1$.\n", "\n", "![Comparing architectures of VGG and NiN, and their blocks.](../img/nin.svg)\n", ":width:`600px`\n", ":label:`fig_nin`\n" ] }, { "cell_type": "code", "execution_count": 1, "metadata": { "origin_pos": 3, "tab": [ "tensorflow" ] }, "outputs": [], "source": [ "import tensorflow as tf\n", "from d2l import tensorflow as d2l\n", "\n", "\n", "def nin_block(num_channels, kernel_size, strides, padding):\n", " return tf.keras.models.Sequential([\n", " tf.keras.layers.Conv2D(num_channels, kernel_size, strides=strides,\n", " padding=padding, activation='relu'),\n", " tf.keras.layers.Conv2D(num_channels, kernel_size=1,\n", " activation='relu'),\n", " tf.keras.layers.Conv2D(num_channels, kernel_size=1,\n", " activation='relu')])" ] }, { "cell_type": "markdown", "metadata": { "origin_pos": 4 }, "source": [ "## [**NiN Model**]\n", "\n", "The original NiN network was proposed shortly after AlexNet\n", "and clearly draws some inspiration.\n", "NiN uses convolutional layers with window shapes\n", "of $11\\times 11$, $5\\times 5$, and $3\\times 3$,\n", "and the corresponding numbers of output channels are the same as in AlexNet. Each NiN block is followed by a maximum pooling layer\n", "with a stride of 2 and a window shape of $3\\times 3$.\n", "\n", "One significant difference between NiN and AlexNet\n", "is that NiN avoids fully-connected layers altogether.\n", "Instead, NiN uses an NiN block with a number of output channels equal to the number of label classes, followed by a *global* average pooling layer,\n", "yielding a vector of logits.\n", "One advantage of NiN's design is that it significantly\n", "reduces the number of required model parameters.\n", "However, in practice, this design sometimes requires\n", "increased model training time.\n" ] }, { "cell_type": "code", "execution_count": 2, "metadata": { "origin_pos": 7, "tab": [ "tensorflow" ] }, "outputs": [], "source": [ "def net():\n", " return tf.keras.models.Sequential([\n", " nin_block(96, kernel_size=11, strides=4, padding='valid'),\n", " tf.keras.layers.MaxPool2D(pool_size=3, strides=2),\n", " nin_block(256, kernel_size=5, strides=1, padding='same'),\n", " tf.keras.layers.MaxPool2D(pool_size=3, strides=2),\n", " nin_block(384, kernel_size=3, strides=1, padding='same'),\n", " tf.keras.layers.MaxPool2D(pool_size=3, strides=2),\n", " tf.keras.layers.Dropout(0.5),\n", " # There are 10 label classes\n", " nin_block(10, kernel_size=3, strides=1, padding='same'),\n", " tf.keras.layers.GlobalAveragePooling2D(),\n", " tf.keras.layers.Reshape((1, 1, 10)),\n", " # Transform the four-dimensional output into two-dimensional output\n", " # with a shape of (batch size, 10)\n", " tf.keras.layers.Flatten(),\n", " ])" ] }, { "cell_type": "markdown", "metadata": { "origin_pos": 8 }, "source": [ "We create a data example to see [**the output shape of each block**].\n" ] }, { "cell_type": "code", "execution_count": 3, "metadata": { "origin_pos": 11, "tab": [ "tensorflow" ] }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Sequential output shape:\t (1, 54, 54, 96)\n", "MaxPooling2D output shape:\t (1, 26, 26, 96)\n", "Sequential output shape:\t (1, 26, 26, 256)\n", "MaxPooling2D output shape:\t (1, 12, 12, 256)\n", "Sequential output shape:\t (1, 12, 12, 384)\n", "MaxPooling2D output shape:\t (1, 5, 5, 384)\n", "Dropout output shape:\t (1, 5, 5, 384)\n", "Sequential output shape:\t (1, 5, 5, 10)\n", "GlobalAveragePooling2D output shape:\t (1, 10)\n", "Reshape output shape:\t (1, 1, 1, 10)\n", "Flatten output shape:\t (1, 10)\n" ] } ], "source": [ "X = tf.random.uniform((1, 224, 224, 1))\n", "for layer in net().layers:\n", " X = layer(X)\n", " print(layer.__class__.__name__,'output shape:\\t', X.shape)" ] }, { "cell_type": "markdown", "metadata": { "origin_pos": 12 }, "source": [ "## [**Training**]\n", "\n", "As before we use Fashion-MNIST to train the model.\n", "NiN's training is similar to that for AlexNet and VGG.\n" ] }, { "cell_type": "code", "execution_count": 4, "metadata": { "origin_pos": 13, "tab": [ "tensorflow" ] }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "loss 0.388, train acc 0.857, test acc 0.863\n", "3319.6 examples/sec on /GPU:0\n" ] }, { "data": { "text/plain": [ "" ] }, "execution_count": 4, "metadata": {}, "output_type": "execute_result" }, { "data": { "image/svg+xml": [ "\n", "\n", "\n", " \n", " \n", " \n", " \n", " 2022-03-24T12:24:32.989331\n", " image/svg+xml\n", " \n", " \n", " Matplotlib v3.5.1, https://matplotlib.org/\n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", "\n" ], "text/plain": [ "
" ] }, "metadata": { "needs_background": "light" }, "output_type": "display_data" } ], "source": [ "lr, num_epochs, batch_size = 0.1, 10, 128\n", "train_iter, test_iter = d2l.load_data_fashion_mnist(batch_size, resize=224)\n", "d2l.train_ch6(net, train_iter, test_iter, num_epochs, lr, d2l.try_gpu())" ] }, { "cell_type": "markdown", "metadata": { "origin_pos": 14 }, "source": [ "## Summary\n", "\n", "* NiN uses blocks consisting of a convolutional layer and multiple $1\\times 1$ convolutional layers. This can be used within the convolutional stack to allow for more per-pixel nonlinearity.\n", "* NiN removes the fully-connected layers and replaces them with global average pooling (i.e., summing over all locations) after reducing the number of channels to the desired number of outputs (e.g., 10 for Fashion-MNIST).\n", "* Removing the fully-connected layers reduces overfitting. NiN has dramatically fewer parameters.\n", "* The NiN design influenced many subsequent CNN designs.\n", "\n", "## Exercises\n", "\n", "1. Tune the hyperparameters to improve the classification accuracy.\n", "1. Why are there two $1\\times 1$ convolutional layers in the NiN block? Remove one of them, and then observe and analyze the experimental phenomena.\n", "1. Calculate the resource usage for NiN.\n", " 1. What is the number of parameters?\n", " 1. What is the amount of computation?\n", " 1. What is the amount of memory needed during training?\n", " 1. What is the amount of memory needed during prediction?\n", "1. What are possible problems with reducing the $384 \\times 5 \\times 5$ representation to a $10 \\times 5 \\times 5$ representation in one step?\n" ] }, { "cell_type": "markdown", "metadata": { "origin_pos": 17, "tab": [ "tensorflow" ] }, "source": [ "[Discussions](https://discuss.d2l.ai/t/332)\n" ] } ], "metadata": { "language_info": { "name": "python" } }, "nbformat": 4, "nbformat_minor": 4 }