Attention Is All You Need
Ashish Vaswani, Noam Shazeer et al.
130k
Citations
13k
Influential Citations
Communications of the ACM
Venue
2017
Year
We trained a large, deep convolutional neural network to classify the 1.2 million high-resolution images in the ImageNet LSVRC-2010 contest into the 1000 different classes. On the test data, we achieved top-1 and top-5 error rates of 37.5% and 17.0%, respectively, which is considerably better than the previous state-of-the-art. The neural network, which has 60 million parameters and 650,000 neurons, consists of five convolutional layers, some of which are followed by max-pooling layers, and three fully connected layers with a final 1000-way softmax. To make training faster, we used non-saturating neurons and a very efficient GPU implementation of the convolution operation. To reduce overfitting in the fully connected layers we employed a recently developed regularization method called "dropout" that proved to be very effective. We also entered a variant of this model in the ILSVRC-2012 competition and achieved a winning top-5 test error rate of 15.3%, compared to 26.2% achieved by the second-best entry.
AlexNet is arguably the most influential paper in modern computer vision. Before 2012, image classification relied heavily on hand-crafted features like SIFT and HOG combined with shallow classifiers. The paper demonstrated that a deep convolutional neural network trained end-to-end on a massive labeled dataset (ImageNet) could achieve a dramatic leap in accuracy, reducing the top-5 error rate from 26.2% to 15.3% in the ILSVRC-2012 competition. This result was so decisive that it triggered the rapid adoption of deep learning across the entire field of AI.
The paper also showcased the importance of engineering: using ReLU activations to accelerate training, implementing efficient GPU convolutions, and employing dropout to combat overfitting. These practical innovations made deep networks feasible for large-scale problems and set a template for future architectures.
On the LSVRC-2010 test set, AlexNet achieved top-1 and top-5 error rates of 37.5% and 17.0%, respectively, compared to previous best results of 45.7% and 25.7%. In the ILSVRC-2012 competition, the model's top-5 error of 15.3% was nearly 11 percentage points better than the second-place entry (26.2%). These results were unprecedented and demonstrated that deep CNNs could generalize well to a 1000-class problem with 1.2 million training images.
AlexNet's success catalyzed the deep learning revolution. It proved that large neural networks trained on big data with powerful hardware could outperform all traditional computer vision methods. The architecture and training techniques (ReLU, dropout, GPU parallelism) became standard in subsequent models like VGGNet, GoogLeNet, and ResNet. Beyond image classification, the principles of deep CNNs were quickly applied to object detection, segmentation, video analysis, and even non-vision domains like speech recognition. The paper also highlighted the importance of computational resources and large datasets, driving investment in GPUs and data collection. Today, AlexNet is a foundational reference in AI education and research, and its impact continues to be felt across the entire field of machine learning.
Ashish Vaswani, Noam Shazeer et al.
Douglas M. Bates, Martin Mächler et al.
Diederik P. Kingma, Jimmy Ba
Jakubův, Jan, Chvalovský, Karel et al.