0% found this document useful (0 votes)
48 views6 pages

AlexNet Architecture Overview

AlexNet was a groundbreaking convolutional neural network that significantly improved image classification performance on ImageNet. Some key aspects of AlexNet included using multiple GPUs to allow training of a larger model, introducing ReLU as the activation function which accelerated training, and employing dropout and data augmentation to reduce overfitting of the 60 million parameter model. On the 2012 ImageNet challenge, AlexNet achieved a top-5 error rate of 15.3%, vastly outperforming the previous state-of-the-art.

Uploaded by

pavithra
Copyright
© All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as DOCX, PDF, TXT or read online on Scribd
0% found this document useful (0 votes)
48 views6 pages

AlexNet Architecture Overview

AlexNet was a groundbreaking convolutional neural network that significantly improved image classification performance on ImageNet. Some key aspects of AlexNet included using multiple GPUs to allow training of a larger model, introducing ReLU as the activation function which accelerated training, and employing dropout and data augmentation to reduce overfitting of the 60 million parameter model. On the 2012 ImageNet challenge, AlexNet achieved a top-5 error rate of 15.3%, vastly outperforming the previous state-of-the-art.

Uploaded by

pavithra
Copyright
© All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as DOCX, PDF, TXT or read online on Scribd

AlexNet: The Architecture

that Challenged CNNs


Jerry Wei

Jul 3, 2019·4 min read

A few years back, we still used small datasets like CIFAR and
NORB consisting of tens of thousands of images. These datasets
were sufficient for machine learning models to learn basic
recognition tasks. However, real life is never simple and has
many more variables than are captured in these small datasets.
The recent availability of large datasets like ImageNet, which
consist of hundreds of thousands to millions of labeled images,
have pushed the need for an extremely capable deep learning
model. Then came AlexNet.
Photo by Casper Johansson on Unsplash

The Problem. Convolutional Neural Networks (CNNs) had


always been the go-to model for object recognition — they’re
strong models that are easy to control and even easier to train.
They don’t experience overfitting at any alarming scales when
being used on millions of images. Their performance is almost
identical to standard feedforward neural networks of the same
size. The only problem: they’re hard to apply to high resolution
images. At the ImageNet scale, there needed to be an innovation
that would be optimized for GPUs and cut down on training
times while improving performance.
The Dataset. ImageNet: a dataset made of more than 15
million high-resolution images labeled with 22 thousand
classes. The key: web-scraping images and crowd-sourcing
human labelers. ImageNet even has its own competition: the
ImageNet Large-Scale Visual Recognition Challenge (ILSVRC).
This competition uses a subset of ImageNet’s images and
challenges researchers to achieve the lowest top-1 and top-5
error rates (top-5 error rate would be the percent of images
where the correct label is not one of the model’s five most likely
labels). In this competition, data is not a problem; there are
about 1.2 million training images, 50 thousand validation
images, and 150 thousand testing images. The authors enforced
a fixed resolution of 256x256 pixels for their images by cropping
out the center 256x256 patch of each image.
Convolutional Neural Networks that use ReLU achieved a 25% error rate on CIFAR-10 six
times faster than those that used tanh. Image credits to Krizhevsky et al., the original authors
of the AlexNet paper.

AlexNet. The architecture consists of eight layers: five


convolutional layers and three fully-connected layers. But this
isn’t what makes AlexNet special; these are some of the features
used that are new approaches to convolutional neural networks:

 ReLU Nonlinearity. AlexNet uses Rectified


Linear Units (ReLU) instead of the tanh
function, which was standard at the time. ReLU’s
advantage is in training time; a CNN using ReLU
was able to reach a 25% error on the CIFAR-10
dataset six times faster than a CNN using tanh.

 Multiple GPUs. Back in the day, GPUs were


still rolling around with 3 gigabytes of memory
(nowadays those kinds of memory would be
rookie numbers). This was especially bad
because the training set had 1.2 million images.
AlexNet allows for multi-GPU training by
putting half of the model’s neurons on one GPU
and the other half on another GPU. Not only
does this mean that a bigger model can be
trained, but it also cuts down on the training
time.

 Overlapping Pooling. CNNs traditionally


“pool” outputs of neighboring groups of neurons
with no overlapping. However, when the authors
introduced overlap, they saw a reduction in error
by about 0.5% and found that models with
overlapping pooling generally find it harder to
overfit.

Illustration of AlexNet’s architecture. Image credits to Krizhevsky et al., the original authors
of the AlexNet paper.

The Overfitting Problem. AlexNet had 60 million


parameters, a major issue in terms of overfitting. Two methods
were employed to reduce overfitting:

 Data Augmentation. The authors used label-


preserving transformation to make their data
more varied. Specifically, they generated image
translations and horizontal reflections, which
increased the training set by a factor of 2048.
They also performed Principle Component
Analysis (PCA) on the RGB pixel values to
change the intensities of RGB channels, which
reduced the top-1 error rate by more than 1%.

 Dropout. This technique consists of “turning


off” neurons with a predetermined probability
(e.g. 50%). This means that every iteration uses a
different sample of the model’s parameters,
which forces each neuron to have more robust
features that can be used with other random
neurons. However, dropout also increases the
training time needed for the model’s
convergence.

The Results. On the 2010 version of the ImageNet


competition, the best model achieved 47.1% top-1 error and
28.2% top-5 error. AlexNet vastly outpaced this with a 37.5%
top-1 error and a 17.0% top-5 error. AlexNet is able to recognize
off-center objects and most of its top five classes for each image
are reasonable. AlexNet won the 2012 ImageNet competition
with a top-5 error rate of 15.3%, compared to the second place
top-5 error rate of 26.2%.

Common questions

Powered by AI

AlexNet distinguished its approach to handling high-resolution images from earlier CNNs through several architectural innovations. These included using the ReLU activation function, which avoided the vanishing gradient issue and sped up training; adopting multi-GPU training to manage memory constraints effectively and improve computational efficiency; and introducing overlapping pooling, which helped generalize feature extraction across larger resolution images. These innovations significantly impacted AlexNet's ability to reduce errors in large datasets like ImageNet by improving learning efficiency and effectively exploiting the vast and diverse dataset .

ReLU activation function contributes to performance improvements in AlexNet by substantially reducing training time compared to the traditional tanh activation function. Specifically, a CNN using ReLU achieved a 25% error rate on the CIFAR-10 dataset six times faster than a CNN using tanh. This efficiency is due to ReLU's ability to introduce non-linearity without suffering from the vanishing gradient problem, which is prevalent in functions like tanh .

AlexNet managed to reduce the top-1 error rates by implementing several key innovations that enhanced model performance over existing models. These included the use of the ReLU activation function, which sped up training and improved model convergence; multi-GPU training, which allowed for handling a larger model more efficiently; overlapping pooling, which slightly reduced error rates by improving the model's ability to generalize; and robust data preprocessing through augmentation techniques. These features collectively contributed to achieving a lower top-1 error rate compared to others, as evidenced by its substantial performance in the ImageNet competition .

AlexNet's performance in the 2012 ImageNet competition was significant because it demonstrated a substantial improvement in both top-1 and top-5 error rates compared to previous models. Specifically, AlexNet achieved a top-5 error rate of 15.3%, dramatically surpassing the second-place model with a top-5 error rate of 26.2%. This performance highlighted the effectiveness of AlexNet's novel features, such as ReLU activation, multi-GPU training, and innovative regularization techniques, setting a new benchmark for future deep learning models and proving the potential of CNNs in handling large-scale image recognition tasks .

Multi-GPU training played a crucial role in AlexNet by efficiently leveraging the limited GPU resources available at the time. Given the hardware constraints of GPUs having only 3 gigabytes of memory, AlexNet utilized multi-GPU setups to split the model's neurons between GPUs, effectively doubling the usable memory and computing resources. This enabled the training of larger CNN architectures without running into memory bottlenecks, which was particularly important given the 1.2 million image training set of ImageNet. This approach not only facilitated larger model training but also significantly reduced training times, demonstrating an innovative optimization technique for deep learning at the time .

Overlapping pooling is effective in reducing overfitting because it introduces redundancy and robustness in the pooling process, which helps CNNs like AlexNet generalize better to unseen data. By pooling outputs of adjacent neuron groups with overlap, it minimizes the error rate by capturing more nuanced features while maintaining the hierarchical structure of the data. This reduced overfitting tendency makes the model more robust to variations in the input .

The introduction of large datasets like ImageNet had a significant impact on the development of deep learning models such as AlexNet by necessitating the creation of more capable architectures that could handle vast amounts of diverse data. ImageNet's vast scale, with over 15 million high-resolution images across 22 thousand classes, provided the challenge and opportunity to innovate models that could not only process this volume but also improve performance metrics like error rates. For AlexNet, this meant innovations such as using ReLU, multi-GPU training, and techniques to combat overfitting, which ultimately won them the 2012 ImageNet competition .

Data augmentation and dropout techniques were crucial in addressing the overfitting problem in AlexNet. Data augmentation increased the diversity of the training dataset using label-preserving transformations such as image translations, horizontal reflections, and PCA on RGB channels, thereby mitigating overfitting by effectively increasing the dataset size by a factor of 2048. Dropout, on the other hand, involved randomly 'turning off' neurons during training, forcing the network to learn more robust features that wouldn't rely on specific neurons. This stochastic practice reduced overfitting by ensuring the network didn't become overly reliant on particular paths during learning .

AlexNet introduced several innovative features to overcome GPU memory limitations and reduce training time. It employed multi-GPU training by allocating half of the model's neurons to one GPU and the other half to another, allowing bigger models to be trained faster. Overlapping pooling was another innovation that reduced errors by about 0.5% and lessened the tendency to overfit. These features collectively allowed AlexNet to manage its large number of parameters and extensive dataset efficiently .

Data augmentation through PCA on RGB pixel values was significant in reducing top-1 error rates in AlexNet as it introduced variability in the color composition of images, enhancing the model's robustness and generalization ability. By altering the intensities of the RGB channels, this method artificially expanded the training dataset's diversity, thereby improving the model's ability to handle variations in real-world imagery and reducing the likelihood of overfitting to specific color patterns in the data. This technique alone reduced the top-1 error rate by more than 1% .

You might also like