What is: Manifold Mixup?
Source | Manifold Mixup: Better Representations by Interpolating Hidden States |
Year | 2000 |
Data Source | CC BY-SA - https://paperswithcode.com |
Manifold Mixup is a regularization method that encourages neural networks to predict less confidently on interpolations of hidden representations. It leverages semantic interpolations as an additional training signal, obtaining neural networks with smoother decision boundaries at multiple levels of representation. As a result, neural networks trained with Manifold Mixup learn class-representations with fewer directions of variance.
Consider training a deep neural network , where denotes the part of the neural network mapping the input data to the hidden representation at layer , and denotes the part mapping such hidden representation to the output . Training using Manifold Mixup is performed in five steps:
(1) Select a random layer from a set of eligible layers in the neural network. This set may include the input layer .
(2) Process two random data minibatches and as usual, until reaching layer . This provides us with two intermediate minibatches and .
(3) Perform Input Mixup on these intermediate minibatches. This produces the mixed minibatch:
where . Here, are one-hot labels, and the mixing coefficient as in mixup. For instance, is equivalent to sampling .
(4) Continue the forward pass in the network from layer until the output using the mixed minibatch .
(5) This output is used to compute the loss value and gradients that update all the parameters of the neural network.