Command Palette
Search for a command to run...
Apprentissage résiduel profond pour la reconnaissance d'images
Apprentissage résiduel profond pour la reconnaissance d'images
Kaiming He Xiangyu Zhang Shaoqing Ren Jian Sun
Résumé
Les réseaux de neurones profonds sont plus difficiles à entraîner. Nous présentons un cadre d'apprentissage résiduel destiné à faciliter l'entraînement de réseaux considérablement plus profonds que ceux utilisés auparavant. Nous reformulons explicitement les couches de manière à apprendre des fonctions résiduelles par rapport aux entrées de la couche, au lieu d'apprendre des fonctions sans référence. Nous fournissons des preuves empiriques exhaustives montrant que ces réseaux résiduels sont plus faciles à optimiser et permettent d'obtenir une meilleure précision grâce à une profondeur considérablement accrue. Sur le jeu de données ImageNet, nous évaluons des réseaux résiduels d'une profondeur pouvant atteindre 152 couches — 8 fois plus profonds que les réseaux VGG tout en conservant une complexité inférieure. Un ensemble de ces réseaux résiduels atteint un taux d'erreur de 3,57 % sur l'ensemble de test ImageNet. Ce résultat a remporté la première place de la tâche de classification de l'ILSVRC 2015. Nous présentons également une analyse sur CIFAR-10 avec 100 et 1000 couches. La profondeur des représentations est d'une importance capitale pour de nombreuses tâches de reconnaissance visuelle. En raison uniquement de nos représentations extrêmement profondes, nous obtenons une amélioration relative de 28 % sur le jeu de données de détection d'objets COCO. Les réseaux résiduels profonds constituent la base de nos soumissions aux concours ILSVRC et COCO 2015, où nous avons également remporté la première place sur les tâches de détection ImageNet, de localisation ImageNet, de détection COCO et de segmentation COCO.
One-sentence Summary
A residual learning framework reformulates layers as learning residual functions with reference to their inputs, enabling the training of networks up to 152 layers deep, 8× deeper than VGG nets with lower complexity, achieving 3.57% error on the ImageNet test set and winning first place on ILSVRC and COCO 2015 tasks.
Key Contributions
- A residual learning framework reformulates layers as learning residual functions relative to their inputs, enabling substantially deeper networks to be trained end-to-end with standard SGD and backpropagation without modifying the solvers.
- Experiments on ImageNet show that residual networks are easier to optimize and can gain accuracy with increased depth, with models up to 152 layers having lower complexity than VGG nets and an ensemble reaching 3.57% top-5 error and winning first place in the ILSVRC 2015 classification competition.
- Further validation on CIFAR-10 includes successfully trained models above 100 layers and exploration of models above 1000 layers, while deep residual representations produce a 28% relative improvement on COCO object detection and underpin first-place submissions on ImageNet detection, ImageNet localization, COCO detection, and COCO segmentation.
Introduction
Deep convolutional neural networks transformed image classification and visual recognition because depth allows networks to integrate low, mid, and high level features, making very deep models important for strong performance. Although normalized initialization and intermediate normalization reduced vanishing and exploding gradient issues, the authors identify a degradation problem: deeper plain networks can show higher training error even without overfitting, indicating that current solvers struggle to optimize very deep networks directly. The authors address this by introducing residual learning, where stacked layers fit a residual mapping instead of an unreferenced mapping, using parameter free identity shortcut connections to make extremely deep networks easier to train and more accurate.
Method
Method
The authors leverage residual learning to make very deep convolutional networks easier to optimize. A key issue is that deeper plain networks can suffer from degradation: once the network is deep enough to converge, adding more layers may increase both training and test error even though the deeper model can represent the shallower one by adding identity mappings. The degradation is not simply a consequence of overfitting. It suggests that optimization solvers struggle to approximate identity mappings when they must be learned through multiple stacked nonlinear layers.
As shown in the figure below:
To address this, they reformulate the learning target. Let H(x) denote the desired underlying mapping over a few stacked layers, with x being the input to those layers. Instead of forcing the layers to learn H(x) directly, they let the layers learn a residual function
F(x):=H(x)−x,so that the original mapping is reconstructed as
H(x)=F(x)+x.This reformulation makes identity mappings easier to approach: if the optimal mapping is close to identity, the residual function can simply be close to zero, and the solver can drive the weights toward small values. In practice, learned residual functions often have small magnitude, which supports the idea that identity mappings provide useful preconditioning for deep optimization.
As shown in the figure below:
The residual formulation is implemented through shortcut connections. A basic building block is defined as
y=F(x,{Wi})+x,where x and y are the input and output of the block, and F(x,{Wi}) is the residual mapping learned by a few weighted layers. In a common two layer example, F can be written as W2σ(W1x), where σ is a ReLU nonlinearity. The addition of x is performed through an identity shortcut, followed by another nonlinearity. This design does not add extra parameters or significant computation, allowing direct comparisons between plain and residual networks with the same depth, width, and computational budget.
The building block is shown in the figure below:
When the shortcut input and residual output have different dimensions, the identity mapping is no longer directly valid. In that case, the authors use a linear projection shortcut,
y=F(x,{Wi})+Wsx,typically implemented with 1×1 convolutions when convolutional feature maps are used. They also show that the identity shortcut is generally sufficient for addressing degradation and is more economical, so projection shortcuts are mainly used for dimension matching. The form of F is flexible and may contain two or more convolutional layers. For deeper networks, a bottleneck design can compress the representation before a 3×3 convolution and expand it again afterward, preserving the shortcut formulation while controlling computational cost.
A common residual function and bottleneck variant are shown in the figure below:
The overall architecture follows a plain deep convolutional design as the baseline, inspired by the philosophy of VGG style networks. It mainly uses 3×3 convolutions and keeps the number of filters constant for a fixed feature map size. When the feature map resolution is halved, the number of filters is doubled to preserve per layer complexity. Downsampling is performed directly with strided convolutions, and the network ends with global average pooling followed by a fully connected softmax layer. The residual version keeps the same main layer stack and inserts shortcut connections around groups of layers. Solid line shortcuts denote identity mappings for equal dimensions, while dotted line shortcuts denote dimension changing paths, such as projection shortcuts or identity mappings with zero padding for increased channels.
Refer to the framework diagram below:
During training, images are randomly resized for scale augmentation, then cropped for training augmentation, and standard color augmentation is applied. Batch normalization is placed after each convolution and before activation. The networks are trained from scratch with stochastic gradient descent, using a mini batch size of 256. The initial learning rate is set to 0.1 and divided by 10 when the error plateaus. A weight decay of 0.0001 and momentum of 0.9 are used, while dropout is not applied. For testing, standard multi crop evaluation can be used, and for best performance the network is evaluated in a fully convolutional form and averaged over multiple input scales.
Experiment
Experiments on ImageNet, CIFAR-10, and detection benchmarks evaluate plain and residual networks to test whether residual learning alleviates optimization degradation in deep models. The results show that ordinary deeper networks are harder to train, whereas residual connections enable deeper models to converge and generalize better, with analyses indicating that residual functions typically have smaller responses. The learned representations also transfer strongly to object detection, demonstrating broad recognition improvements beyond classification.
The table presents ImageNet architectures that share a common downsampling path and increase depth by stacking repeated building blocks. The shallower plain variants use paired 3x3 blocks, while the deeper variants use bottleneck blocks and place more repetitions in the high-dimensional stages. The cited experiments use these models for ImageNet classification and show that the deeper plain network suffers higher validation error than the shallower plain baseline. Model depth is expanded by repeating blocks while keeping the stage resolution pattern fixed. Plain architectures use stacked 3x3 block pairs, while deeper architectures use bottleneck block stacks. The deeper plain network performs worse than the shallower plain network despite having more capacity.
For ImageNet top-1 validation error, residual networks match or improve upon plain networks without adding parameters. Increasing depth from 18 to 34 layers hurts the plain network, but the residual network becomes markedly better. The result points to residual connections reducing degradation in deeper models. The 34 layer plain network performs worse than the 18 layer plain network. The residual network greatly improves the 34 layer model and matches the 18 layer model closely.
Residual networks improve over the 34-layer plain network on ImageNet validation in both top-1 and top-5 errors. Among the ResNet-34 variants, option C achieves the best errors, with option B slightly ahead of option A. The plain network trails the residual variants and remains less accurate than several reported baselines. Residual connections reduce validation error compared with the plain 34-layer network. Option C is the strongest of the three ResNet-34 variants. The evaluated models achieve competitive top-5 performance against prior ImageNet models.
On the ImageNet results shown, ResNet-34 models achieve lower top-1 error than the compared VGG, PReLU, and BN inception baselines. The ResNet-34 C configuration has the best reported top-1 and top-5 errors among these rows, while VGG with top-1 accuracy is clearly less competitive. One VGG result is noted as being reported on the test set, unlike the other listed validation results. ResNet-34 improves ImageNet classification accuracy relative to the listed single-model baselines. The ResNet-34 C variant gives the strongest top-1 and top-5 performance among the reported rows.
The comparison reports top-5 error rates for ensemble results on ImageNet, showing that earlier ILSVRC 2014 models have higher error than more recent approaches. Among the listed ensembles, ResNet shows the strongest test performance, while PReLU-net and BN-inception also reduce error relative to early VGG and GoogLeNet baselines. ResNet ensembles achieve the lowest reported top-5 error among the compared methods. More recent ensemble methods outperform the earlier ILSVRC 2014 VGG and GoogLeNet baselines.
Experiments on ImageNet classification compare plain and residual architectures that share similar downsampling paths and grow depth through repeated blocks. The results show that simply deepening plain networks causes degradation, whereas residual connections allow deeper models to train more effectively and achieve better validation performance. Additional comparisons to prior baselines and ensembles indicate that the residual designs are more accurate than the plain network and competitive or stronger than representative deep-network methods.