URL: https://proceedings.mlr.press/v28/sutskever13.pdf
π― Pitch
Carefully tuned SGD with momentum can train deep and recurrent nets from scratch, matching or beating complex second-order methodsβbut only when a very specific initialization and a slowly increasing momentum schedule are used together; miss either, and performance collapses.
1. Executive Summary
This paper analyzes how stochastic gradient descent with momentum can train deep neural networks and recurrent neural networks from random initializations β a feat previously thought to require sophisticated second-order methods like Hessian-Free optimization β by establishing that two mechanisms are jointly essential: a well-designed random initialization (the sparse initialization of Martens (2010), which constrains each unit to receive connections from only 15 randomly chosen preceding units) and a carefully tuned momentum schedule (a slowly increasing schedule for the momentum coefficient ΞΌ, coupled with Nesterov's accelerated gradient). On the three deep autoencoder benchmarks from Hinton & Salakhutdinov (2006), Nesterov's accelerated gradient with the proposed momentum schedule and sparse initialization achieves the lowest published training errors β surpassing even previously reported Hessian-Free results β while for recurrent neural networks on long-term dependency problems (addition, multiplication, and memorization tasks with temporal spans up to 200 steps), the same combination solves tasks that standard SGD fails on entirely, establishing that first-order momentum methods suffice for deep and recurrent network training only when both the initialization and the momentum schedule are jointly optimized.
2. Context and Motivation
The Core Problem: Training Deep Networks Was Considered Nearly Impossible with First-Order Methods
The fundamental problem this paper addresses is one that defined the deep learning field through the mid-2000s: deep and recurrent neural networks could not be reliably trained from random initializations using simple first-order optimization methods like stochastic gradient descent. Despite their theoretical representational power β DNNs can express highly complex, hierarchical feature detectors, and RNNs can model arbitrarily long sequences through temporal unfolding β these architectures had remained largely impractical because the optimization algorithms available at the time simply could not find good parameter configurations.
This difficulty was not incidental or minor; it was the primary barrier to adoption. The paper opens by stating that DNNs were "considered to be almost impossible to train using stochastic gradient descent with momentum" and that this difficulty "prevented their widespread use until fairly recently" (Section 1). The severity of this barrier is difficult to overstate from a modern perspective. Before this problem was solved, deep learning was a niche approach, and most practitioners used shallow models (single hidden layer) or hand-engineered features because attempting to train deeper architectures reliably resulted in failure β either the optimization would get stuck in poor local minima, diverge, or simply fail to make meaningful progress.
For recurrent neural networks, the situation was even more dire. RNNs can be viewed as "very deep neural networks that have a 'layer' for each time-step with parameter sharing across the layers" (Section 1), meaning that a sequence of length 200 effectively creates a 200-layer network. The well-known vanishing gradient problem (Bengio et al., 1994) meant that gradient-based learning signals decayed exponentially as they propagated backward through time, making it "impossibly difficult for first-order optimization methods" to learn dependencies spanning more than a handful of time steps. Special architectures like Long Short-Term Memory (LSTM; Hochreiter & Schmidhuber, 1997) were developed specifically to circumvent this problem, but the paper focuses on the harder case: training standard RNNs with conventional neurons (no gating mechanisms, no specialized memory units) on tasks exhibiting extreme long-range temporal dependencies β exactly the regime where first-order methods were believed to be helpless.
Why This Problem Matters: Practical and Theoretical Significance
The importance of solving this optimization problem spans practical deployment, theoretical understanding, and the trajectory of the entire field.
Practical significance: unlocking deep architectures for real tasks. By 2013, when this paper was published, deep networks had already demonstrated breakthrough performance on speech recognition (Dahl et al., 2012; Hinton et al., 2012), image classification (Krizhevsky et al., 2012), and language modeling (Graves, 2012). But the training recipes used to achieve these results relied on either greedy layerwise pre-training (Hinton et al., 2006; Bengio et al., 2007) β a complex, multi-phase procedure that trains each layer sequentially using auxiliary objectives before fine-tuning β or on highly specialized second-order optimization methods like Hessian-Free optimization (Martens, 2010). Both approaches carry substantial costs: pre-training adds engineering complexity and constrains the types of architectures that can be used (since each layer must be trainable with an auxiliary objective), while Hessian-Free optimization requires solving large linear systems at each update, making it computationally expensive and difficult to implement correctly.
A demonstration that plain SGD with momentum could achieve comparable or better results would radically simplify the training pipeline, removing the need for pre-training stages or specialized optimizers, and thereby making deep learning accessible to a wider range of practitioners and applications. The paper explicitly positions itself in this practical tradition, stating that it aims to "provide a simple to understand and easy to use framework for deep learning that is surprisingly effective" (Section 1).
Theoretical significance: understanding why training fails. Beyond the practical recipe, the paper addresses a deeper conceptual question: what actually causes first-order methods to fail on deep networks? Prior work had documented the symptoms β slow progress, vanishing gradients, poor local minima β but the root causes remained poorly understood. The paper's investigation reveals that the failure is not inherent to first-order methods per se, but rather stems from two specific, fixable deficiencies: poor initialization schemes that place the network in regions of parameter space from which recovery is impossible, and inadequate momentum tuning that fails to accelerate progress along the low-curvature directions that dominate deep network optimization landscapes. This shifts the theoretical narrative from "SGD cannot train deep networks" to "SGD fails when initialization and momentum are mishandled" β a much more precise and actionable diagnosis.
Field trajectory: enabling the shift from pre-training to end-to-end training. The paper occupies a pivotal position in the history of deep learning. It was published in 2013, during the transition from the pre-training era (roughly 2006β2011) to the end-to-end training era that would follow. While Krizhevsky et al. (2012) had already shown that a deep convolutional network could be trained from scratch (with careful architecture design, ReLU activations, and aggressive regularization), the question of whether this success generalized to other architectures and problem domains remained open. This paper provides evidence that the answer is yes β that with proper initialization and momentum, end-to-end training of deep architectures is broadly feasible β helping to solidify the emerging consensus that pre-training, while useful, was not strictly necessary.
Prior Approaches and Where They Fell Short
The paper identifies four distinct lines of prior work, each of which partially addressed the problem but left significant gaps.
Greedy layerwise pre-training (Hinton et al., 2006; Bengio et al., 2007). This was the dominant paradigm for training deep networks in the mid-to-late 2000s. The approach trains one layer at a time using an unsupervised auxiliary objective (such as reconstructing the input or modeling the distribution of the previous layer's activations), and only after all layers are pre-trained is the full network fine-tuned with standard SGD. While this achieved impressive results and arguably launched the deep learning renaissance, it suffers from fundamental limitations: (a) it restricts architectures to those where each layer can meaningfully be trained in isolation, (b) it adds substantial engineering complexity and hyperparameter burden (each pre-training stage has its own learning rate, stopping criterion, and so on), and (c) it does not address the underlying optimization problem β fine-tuning still uses first-order methods, and the pre-training merely provides a better starting point. The paper cites this work to establish that prior solutions were workarounds rather than solutions to the core optimization challenge.
Hessian-Free optimization (Martens, 2010; Martens & Sutskever, 2011). This approach demonstrated that a sophisticated truncated-Newton method could train deep autoencoders and RNNs from random initializations without pre-training, achieving lower errors than pre-training-based methods on standard benchmarks. This was a significant breakthrough because it showed that the optimization problem was not fundamentally intractable β there existed algorithms that could solve it. However, HF optimization came with its own limitations: (a) it is substantially more complex to implement than SGD, requiring the solution of large linear systems via conjugate gradient at each update; (b) it carries higher computational cost per iteration; (c) specialized damping techniques were required to make it work on RNNs with long-term dependencies (Martens & Sutskever, 2011); and (d) it remained unclear whether its success was due to the second-order curvature information it exploited, or whether similar results could be achieved with simpler methods if they were properly configured. The paper explicitly positions its work relative to HF, asking whether the performance gap between SGD and HF can be "eliminated or nearly eliminated" through better initialization and momentum (Section 1).
Well-designed random initializations (Glorot & Bengio, 2010; Chapelle & Erhan, 2011). By the time this paper was written, several works had begun to challenge the narrative that first-order methods were fundamentally incapable. Glorot & Bengio (2010) introduced a normalized initialization scheme designed to preserve variance across layers, and reported training networks up to 8 layers deep. Chapelle & Erhan (2011) used this initialization with plain SGD to train the 11-layer autoencoder from Hinton & Salakhutdinov (2006), surpassing the original pre-training results. However β and this is the critical gap the paper identifies β these results "still fell short of those reported by Martens (2010) for the same tasks" (Section 1). The performance gap between well-initialized SGD and HF remained unexplained. Had Glorot & Bengio (2010) and Chapelle & Erhan (2011) already solved the problem? The paper's answer is: partially. They solved the initialization piece but not the optimization piece. Their SGD configuration lacked the carefully tuned momentum that this paper shows is essential to close the remaining gap.
Prior work on momentum in neural networks (Orr, 1996; LeCun et al., 1998). Momentum had been studied extensively in the neural network literature before 2013, but the paper argues that this prior work systematically underestimated its importance because it focused on the wrong regime. Specifically, prior theoretical analyses and benchmarks (Orr, 1996; Wiegerinck et al., 1999) examined momentum in the stochastic asymptotic regime, where the optimization problem reduces to estimation β averaging out noise in the gradient to converge to a precise minimum. In this regime, the theory predicts that momentum provides no asymptotic benefit (the convergence rate remains O(1/T) with or without momentum when gradients are stochastic), and experiments confirmed this. As a result, "interest in momentum methods diminished after they had received substantial attention in the 90's" and "some authors even discourage using momentum or downplay its potential advantages" (Section 2).
The paper's key insight is that this asymptotic analysis misses the point entirely for deep learning. The dominant computational cost in deep network training occurs not during the final fine-convergence phase, but during the transient phase (Darken & Moody, 1993) β the initial period where the optimizer must move from a random initialization through complex, highly non-convex terrain to reach a good region of parameter space. During this transient phase, gradient signals exhibit persistent directional structure (they are not purely noise), and momentum's ability to accelerate along low-curvature directions becomes critical. The paper argues that the field's prior negative conclusions about momentum arose from a misalignment between the analysis framework (asymptotic stochastic convergence) and the actual computational bottleneck (transient optimization).
Additionally, the paper identifies a second reason prior work underestimated momentum: the use of "poorly designed standard random initializations" and "suboptimal meta-parameter schedules (for the momentum constant in particular)" (Section 1). If the network is initialized in a region from which no reasonable optimizer can escape, or if the momentum schedule is too conservative, then momentum will appear ineffective β but the failure is in the initialization and scheduling, not in the method itself. The paper hypothesizes that previous negative results conflated these factors, and that isolating them reveals momentum's true effectiveness.
How This Paper Positions Itself
The paper positions itself as a resolution of the tension between two bodies of evidence that appeared contradictory at the time: on one side, the success of HF optimization from random initializations (Martens, 2010; Martens & Sutskever, 2011), which suggested that deep networks require sophisticated second-order methods; on the other, the partial success of well-initialized first-order methods (Glorot & Bengio, 2010; Chapelle & Erhan, 2011), which suggested that the problem was not as hard as previously believed.
The paper's thesis is that both perspectives contain partial truth, and the full picture requires both pieces: a well-designed initialization gets you into a trainable regime, but only carefully tuned momentum (specifically, Nesterov's accelerated gradient with a gradually increasing schedule for ΞΌ) can navigate the optimization landscape efficiently enough to match or exceed second-order methods. Neither initialization nor momentum alone is sufficient; both are necessary.
This is stated most clearly in the paper's abstract: "both the initialization and the momentum are crucial since poorly initialized networks cannot be trained with momentum and well-initialized networks perform markedly worse when the momentum is absent or poorly tuned." This formulation is deliberately symmetric β it argues against the implicit assumption in prior work that one factor could compensate for deficiencies in the other.
The paper also positions itself relative to the convex optimization theory of Nesterov (1983, 2003). While Nesterov's accelerated gradient had been studied extensively in the convex optimization community and was known to provide provably faster convergence on smooth convex functions (O(1/TΒ²) vs. O(1/T) for gradient descent), its relevance to non-convex deep network training was unclear. The paper bridges this gap by arguing that the transient phase of deep network optimization β where persistent gradient structure exists and the objective, while non-convex, exhibits enough local regularity for acceleration to help β is precisely the regime where Nesterov-style momentum should be most beneficial. The paper's empirical results on deep autoencoders and RNNs provide evidence for this connection that the optimization theory alone could not justify (since the convergence guarantees require convexity).
Finally, the paper positions its contribution as complementary to, rather than competing with, Hessian-Free optimization. Section 5 develops a conceptual connection between HF and momentum methods, arguing that HF can be viewed as a type of momentum method that uses special CG initializations to persist information across updates. The paper even experiments with making HF "behave even more like NAG" and reports improved results (Table 1, column HFβ ). This framing is notable: rather than declaring first-order methods superior, the paper argues that the boundary between first-order and second-order methods is blurrier than commonly assumed, and that understanding the role of momentum-like mechanisms in both classes of methods can lead to improvements across the board.
In summary, the paper addresses a specific and well-defined gap: the unexplained performance difference between HF optimization (which worked well but was complex) and SGD with good initializations (which was simpler but performed worse). The resolution it proposes β that momentum, properly configured, accounts for the difference β transforms the narrative around deep network optimization and provides a practical recipe that, by the paper's own evidence, achieves state-of-the-art results on challenging benchmarks.
3. Technical Approach
3.1 Reader Orientation
This paper is an empirical investigation into why stochastic gradient descent (SGD) with momentum succeeds or fails at training deep networks, not a proposal for a new optimization algorithm. The system being analyzed is the training pipeline itself β the combination of network initialization, optimization algorithm, and hyperparameter schedule that takes a randomly initialized deep or recurrent neural network and produces a trained model. The problem it solves is the previously unexplained performance gap between simple first-order methods (SGD with momentum) and sophisticated second-order methods (Hessian-Free optimization) on deep architectures. The shape of the solution is a two-part diagnosis: a specific random initialization scheme (sparse initialization) that places the network in a trainable region of parameter space, combined with a specific momentum configuration (Nesterov's accelerated gradient with a carefully designed, gradually increasing schedule for the momentum coefficient ΞΌ) that efficiently navigates the optimization landscape during the critical transient phase of learning.
3.2 Big-Picture Architecture (Diagram in Words)
The training system has four major components connected in a pipeline:
-
Random Initialization Module β takes a network architecture specification (layer sizes, connectivity pattern) and produces a set of initial parameters. For feed-forward networks, this is the sparse initialization (SI) scheme: each unit receives connections from exactly 15 randomly chosen units in the previous layer, with weights drawn from a unit Gaussian, and biases set to zero. For recurrent networks, this is the Echo-State Network (ESN)-inspired initialization: a hidden-to-hidden matrix with spectral radius 1.1, sparse 15-fan-in connectivity, and carefully scaled input-to-hidden weights (standard deviation 0.001 for tasks with distractors, 0.1 otherwise).
-
Momentum Optimizer β the core iterative update procedure. At each step, it computes a gradient estimate on a mini-batch, combines it with a decaying velocity vector (the "momentum"), and produces a parameter update. The system studies two variants: classical momentum (CM), which computes the gradient at the current parameter position, and Nesterov's accelerated gradient (NAG), which computes the gradient at a "lookahead" position (current parameters plus the decayed velocity). NAG is the primary variant that achieves the best results.
-
Momentum Schedule Controller β determines the value of the momentum coefficient ΞΌ at each training iteration. This is not a fixed hyperparameter but a function of the iteration count, following a formula that gradually increases ΞΌ from a low value to a high ceiling ΞΌβββ. The schedule is: ΞΌβ = min(1 β 2^(-1 β logβ(βt/250β + 1)), ΞΌβββ). The schedule is identical in structure for both feed-forward and recurrent experiments, differing only in the ceiling value and the iteration at which it is engaged.
-
Training Loop β integrates the components above: initialize parameters, then for a fixed number of iterations (750,000 for autoencoders, 50,000 for RNNs), sample a mini-batch, compute the gradient, apply the momentum update with the current ΞΌ from the schedule, and repeat. For autoencoders, a final "fine-tuning" phase of 1,000 iterations reduces ΞΌ to 0.9 (or 0 if ΞΌ was already 0) to allow finer convergence.
Information flows sequentially: network architecture β sparse initialization β initial parameters β training loop (which queries the schedule controller for ΞΌ at each step, computes batch gradients, and passes them to the momentum optimizer) β trained parameters.
3.3 Roadmap for the Deep Dive
- First, the initialization schemes β sparse initialization for feed-forward networks and ESN-inspired initialization for recurrent networks β because initialization determines whether the network starts in a region from which optimization is even possible, and the paper's central claim is that prior failures were due to poor initialization rather than fundamental limitations of first-order methods.
- Second, the momentum mechanisms β classical momentum and Nesterov's accelerated gradient β including their update equations, the geometric intuition for why NAG is more stable, and a formal theorem characterizing their difference on quadratic objectives, because the paper's second central claim is that momentum, not second-order curvature information, is the missing ingredient that closes the gap with Hessian-Free optimization.
- Third, the momentum schedule β the specific formula for ΞΌβ and its theoretical motivation from Nesterov's convergence theory β because the schedule is what makes momentum work in practice, and prior work's failure to use such schedules explains why momentum was previously underestimated.
- Fourth, the learning rate selection and fine-tuning protocol β how the learning rate is chosen per configuration and how ΞΌ is reduced at the end of training β because these practical details are essential for reproducibility and reveal the paper's model of optimization as a two-phase process (transient acceleration followed by fine convergence).
- Fifth, the connection between momentum and Hessian-Free optimization β the conceptual argument that HF can be viewed as a momentum method with CG-based initializations β because this connection unifies the paper's findings with prior work and explains why momentum can substitute for second-order methods.
3.4 Detailed, Sentence-Based Technical Breakdown
This is primarily an empirical analysis paper whose core idea is that the failure of first-order methods to train deep networks was caused by poor initialization and inadequate momentum scheduling, not by any fundamental limitation of gradient-based optimization, and that correcting both factors allows SGD with Nesterov momentum to match or exceed Hessian-Free optimization on challenging benchmarks.
Sparse Initialization for Feed-Forward Networks
The initialization scheme used for all deep autoencoder experiments is the sparse initialization (SI) from Martens (2010). The core idea is to constrain each unit to receive input from only a small, fixed number of units in the previous layer, rather than from all units. This addresses two failure modes of standard dense random initializations in deep sigmoidal networks: saturation (where the weighted sum of inputs to a unit is so large in magnitude that the unit's output is always near 0 or 1, producing vanishing gradients) and homogenization (where each unit in a layer computes roughly the same function because all receive similar blends of the previous layer's outputs).
The specific procedure for sigmoid networks is as follows. For each unit in each hidden layer, 15 units from the previous layer are selected uniformly at random (without replacement). The weights connecting these 15 chosen units to the current unit are drawn independently from a standard Gaussian distribution β a normal distribution with mean 0 and variance 1, denoted N(0, 1). All other incoming weights to the current unit (from the remaining unselected units in the previous layer) are set to zero and remain zero throughout training. The bias of each unit is initialized to zero.
This procedure has several immediate consequences. Because each unit receives input from exactly 15 sources regardless of the previous layer's size, the expected total input to each unit is independent of the layer width. The sum of 15 independent N(0, 1) draws has mean 0 and variance 15, so the expected squared magnitude of the pre-activation input is 15. This stays constant as layer sizes increase, unlike dense initialization with independent weights, where the variance of the total input grows with the fan-in. Keeping the input variance controlled prevents the sigmoid from saturating at initialization β the unit starts in its approximately linear regime where gradients are non-trivial.
Additionally, because different units sample different subsets of 15 input connections, each unit sees a qualitatively different projection of the previous layer's activity. The receptive fields are sparse and non-overlapping in different patterns for different units. This produces diverse representations from the start: units in the same layer compute different functions of their inputs rather than near-identical functions, which would happen with dense initialization (by the law of large numbers, all units with full connectivity would receive roughly the same total input if the previous layer's activity is broadly distributed).
The paper also describes the transformation used when networks use tanh activation instead of sigmoid. Tanh is a sigmoidal function scaled to output in the range [-1, 1] rather than [0, 1], and is related to the standard sigmoid Ο(x) by tanh(x) = 2Ο(2x) β 1. To simulate sigmoid-like behavior with tanh units (maintaining the same effective initialization), the paper states: "we transform the weights to simulate sigmoid units by setting the biases to 0.5 and rescaling the weights by 0.25." This means: take the weights sampled from N(0, 1) for sigmoid initialization, multiply them by 0.25, and set the biases to 0.5 instead of 0. The factor 0.25 accounts for the steeper slope of tanh near zero compared to sigmoid, and the bias shift of 0.5 accounts for the different output range.
Sensitivity analysis. The paper reports a brief sensitivity study on the scaling of the sparse initialization (Table 3). Using the standard SI with a scale multiplier of 1 (weights from N(0, 1)) achieves a final training squared error of 0.074 on the Curves dataset. Multiplying all weights by 0.5 (making them half as large, reducing the initial input variance by a factor of 4) achieves an error of 16. Multiplying by 0.25 achieves an error of 16. Multiplying by 2 achieves 0.083 β slightly worse than the default but still reasonable. Multiplying by 4 achieves 0.35 β substantially worse. This shows that the initialization is modestly robust to the scale parameter (factors of 2 still work) but fails hard when the scale is too small (0.5 or 0.25), likely because the initial signals become too weak to propagate through many layers, or too large (5, not directly listed but implied by "factor 5 we did not achieve sensible results").
ESN-Inspired Initialization for Recurrent Neural Networks
For recurrent neural networks, the paper develops an initialization scheme inspired by Echo-State Networks (ESNs; Jaeger & Haas, 2004). ESNs are a family of RNNs where the recurrent weights are fixed and randomly drawn, and only the output layer is trained. While the paper's goal is to train all weights (recurrent, input, and output), it borrows the ESN insight that the spectral radius of the hidden-to-hidden weight matrix strongly controls the network's temporal dynamics.
The spectral radius of a matrix is the maximum absolute value of its eigenvalues. For a recurrent weight matrix Wββ, the spectral radius Ο(Wββ) determines whether the dynamics of the hidden state (with a tanh nonlinearity) are contractive (Ο < 1: inputs are quickly forgotten, the state converges to a fixed point), expansive (Ο β« 1: the state diverges or becomes chaotic, producing varied responses to different input histories but also potentially exploding gradients), or at the edge of chaos (Ο β 1).
The paper's key insight is that a spectral radius slightly greater than 1 β specifically, 1.1 β is the right regime for gradient-based learning. When Ο is exactly 1 or below, the RNN cannot retain information over long time spans because the dynamics are too strongly contractive: any difference in hidden state caused by a distant input is exponentially squeezed toward zero. When Ο is much larger than 1, the dynamics become chaotic and the gradients explode, making learning impossible. But when Ο is only slightly above 1, the dynamics are oscillatory enough to retain information over long distances while the gradients, even if they grow, grow slowly enough that learning remains stable. The paper states: "when the spectral radius is only slightly greater than 1, the dynamics remain oscillatory and chaotic while the gradient are no longer exploding (and if they do explode, then only 'slightly so'), so learning may be possible."
The full ESN-inspired initialization for the RNN experiments (summarized in Table 4) has the following components:
- Hidden-to-hidden connections: 15 fan-in sparse connectivity (each hidden unit receives from 15 randomly chosen hidden units), with weights drawn such that the resulting matrix has a spectral radius of 1.1. The paper does not specify the exact procedure for achieving a target spectral radius, but the standard approach (used in ESN literature) is to sample random weights, compute the spectral radius, and rescale all weights uniformly to achieve the desired value. The 15 fan-in sparsity is analogous to the sparse initialization for feed-forward networks and serves the same purpose: controlling the variance of the recurrent input and ensuring diverse receptive fields.
- Input-to-hidden connections: These are dense (each hidden unit receives from all input units), with weights drawn from a Gaussian distribution. The scale of these weights is task-dependent and crucial. For tasks with many irrelevant distractor inputs (the addition and multiplication problems, described in Appendix A.3), the standard deviation is 0.001 β a very small value. For tasks without distractors, a larger standard deviation of 0.1 is used. The paper distinguishes between two types of input connections in the addition and multiplication tasks: the "add" and "mul" connections (which carry the actual operands) and the "mem" connections (which carry a marker signal indicating when to output the result). The small scale of 0.001 applies to the operand connections; the marker connections use the larger 0.1 scale.
- Hidden biases: Initialized to zero.
- Output biases: Initialized to the average of the target outputs. This is a centering trick: by initializing the output bias to the mean output value, the output layer starts already roughly calibrated, and only needs to learn the deviations from the mean.
- Centering of inputs and outputs: The paper emphasizes that "centering (mean subtraction) of both the inputs and the outputs" was "important to reliably solve all of the training problems." This subtracts the mean of each input and output dimension across the training set, centering the data distribution. The paper notes that Martens & Sutskever (2011) solved these problems without centering, while the current approach required centering for the multiplication problem specifically.
The scale tradeoff for input-to-hidden weights. The paper provides a detailed explanation for why the input-to-hidden weight scale matters. If the scale is too large, the many irrelevant distractor inputs collectively generate a strong signal that "overwrites" the hidden state's memory of relevant past information before it can be used. The optimizer, seeing no correlation between distant inputs and outputs because the information is destroyed in transit, converges to a poor local minimum where the RNN ignores long-range dependencies entirely. If the scale is too small, learning is simply too slow β the gradient signals from the inputs are attenuated, and progress along input-to-hidden weight directions is minimal. The value 0.001 was found to balance these concerns for the distractor-heavy tasks. For tasks without distractors, the issue of irrelevant inputs overwriting the state does not arise, so a larger scale (0.1) accelerates learning without penalty.
Classical Momentum (CM)
Classical momentum (Polyak, 1964) is a technique for accelerating gradient descent that accumulates a velocity vector β an exponentially decaying moving average of past gradients β and uses this vector, rather than the raw gradient, to update the parameters. The paper provides the standard update equations:
where is the velocity vector at iteration (initialized to zero), is the momentum coefficient controlling how much of the past velocity is retained, is the learning rate, and is the gradient of the objective function evaluated at the current parameter vector .
What these equations compute: At each iteration, the velocity update (Equation 1) takes the previous velocity , scales it down by factor (forgetting a fraction of the past), and adds the negative gradient scaled by the learning rate (the direction of steepest local descent). The parameter update (Equation 2) simply adds the resulting velocity to the current parameters. The velocity acts as a momentum in the physical sense: parameters continue moving in directions that have consistently received gradient pushes, and resist rapid changes in direction.
Why this form: momentum exploits the fact that in ill-conditioned optimization problems (including deep network training objectives), the objective function changes much more slowly along some directions in parameter space than others. Directions of low curvature β where the second derivative is small β correspond to "valleys" in the loss landscape where gradient descent makes slow progress because each step is small. Along high-curvature directions, gradient descent oscillates or diverges unless the learning rate is small. Momentum addresses both issues simultaneously: in low-curvature directions, consecutive gradients point in roughly the same direction, so they accumulate in the velocity, producing larger effective steps than gradient descent could take (the velocity grows roughly as in steady state). In high-curvature directions, consecutive gradients oscillate in sign, so they cancel out in the velocity rather than accumulating, allowing the use of a larger learning rate than would be stable without momentum. The paper states that CM can require "-times fewer iterations than steepest descent to reach the same level of accuracy" when , where is the condition number (ratio of largest to smallest curvature) at the local minimum. This is a classic result from Polyak (1964) for deterministic quadratic objectives.
Nesterov's Accelerated Gradient (NAG)
Nesterov's accelerated gradient (Nesterov, 1983) is a modification of classical momentum that was developed from a different mathematical framework (the "estimate sequence" technique in convex optimization) but can be expressed as a small change to the CM update. The paper provides a reformulation of NAG that makes the relationship to CM explicit:
where all symbols have the same meaning as in CM. The parameters are updated identically to CM (), and the velocity decay term is identical. The sole difference is where the gradient is evaluated: CM uses β the gradient at the current position β while NAG uses β the gradient at a "lookahead" position that is the current position plus the decayed velocity from the previous step. This position is approximately where the parameters would move if the velocity were applied unchanged (it differs from only by the yet-unknown gradient correction ).
What this computes differently from CM: NAG performs a "tentative" update using the old velocity alone, evaluates the gradient at this tentative position, and then uses that gradient to correct the velocity before applying the final update. The consequence is that if the velocity is pointing in a poor direction β if would increase the objective β the gradient at will point back toward more strongly than the gradient at would. This earlier, stronger correction prevents the velocity from committing fully to a bad direction.
Geometric intuition (Figure 1). The paper illustrates the difference with a diagram: for CM, the gradient correction (the term) is computed at the current position and applied to a velocity that is already pointing partially in a bad direction, so the correction is applied "after the fact" and may be insufficient if is large and the velocity is large. For NAG, the gradient is evaluated further along the trajectory, "peeking ahead" to see the consequences of continuing with the current velocity, and applies a correction that is larger and more timely. This "lookahead" makes NAG more stable β it tolerates larger values of without oscillating or diverging. The paper demonstrates this with a two-dimensional quadratic objective in the appendix (Figure 2): CM oscillates severely along the high-curvature direction, repeatedly overshooting and correcting, while NAG damps these oscillations much more effectively with the same and .
Why this form: the standard presentation of Nesterov's method in the optimization literature uses an auxiliary sequence and a different parameterization. The reformulation in Equations 3-4 (derived in the appendix of the paper) shows that the method previously analyzed as a sophisticated acceleration scheme for convex optimization is, operationally, just classical momentum with a one-line change to where the gradient is evaluated. This connection is non-obvious and important: it means that practitioners who already use momentum can switch to NAG by changing a single line of code, and it reframes NAG as an improved momentum variant rather than a fundamentally different algorithm.
Local Convergence Analysis on Quadratic Objectives (Theorem 2.1)
To formalize the difference between CM and NAG and quantify precisely how NAG achieves greater stability, the paper analyzes both methods on a positive-definite quadratic objective. The objective is , where is a symmetric positive-definite matrix (the curvature/Hessian) and is a vector. Since is symmetric, it can be diagonalized: , where is orthonormal (a rotation/reflection) and is diagonal with positive entries β the eigenvalues of , representing the curvature along each eigenvector direction.
The analysis uses a reparameterization: instead of optimizing directly, optimize in the rotated coordinate system . Because is orthonormal (), this is simply a rotation β it preserves all distances and angles. In the rotated coordinates, the quadratic separates into a sum of independent one-dimensional quadratics: , where . Each dimension evolves independently with its own curvature .
The paper then proves that both CM and NAG are invariant under orthonormal transformations (Proposition 6.1 in the appendix) β applying CM or NAG to and then rotating back to -coordinates via produces exactly the same iterates as applying them directly to . This means we can analyze their behavior one dimension at a time in the rotated coordinates and then reassemble the results.
Theorem 2.1 states that under this decomposition, NAG along any eigen-direction with eigenvalue is equivalent to CM along that same direction, but with an effective momentum coefficient of instead of . Formally, if we apply one step of NAG with momentum to the full , the resulting parameter vector and velocity vector decompose across dimensions exactly as if we had applied CM independently to each one-dimensional quadratic , using momentum coefficient for dimension . CM, by contrast, uses the same momentum for all dimensions.
What this means operationally: for high-curvature directions (large ), the effective momentum in NAG is , which is smaller than β possibly much smaller, and potentially negative if . For low-curvature directions (small ), the effective momentum is approximately , since . NAG thus automatically reduces momentum more aggressively in high-curvature directions, preventing the oscillations that plague CM, while maintaining full momentum in low-curvature directions where acceleration is needed. CM applies the same momentum everywhere, so if is set high enough to accelerate low-curvature directions adequately, it over-accelerates high-curvature directions and oscillates (or diverges).
Why this form: The theorem makes precise the intuition from Figure 1. Because NAG evaluates the gradient at the lookahead position , the gradient already reflects (to first order) the effect of the velocity on the objective. In a quadratic, the gradient at is , so . The extra term is a curvature-dependent correction: it is large when has components along high-curvature eigendirections (where has large eigenvalues), acting as an additional damping force that slows the velocity precisely in those directions. The effective momentum reduction emerges directly from this correction term.
Practical implication: NAG tolerates larger values of than CM for a given learning rate . The paper exploits this directly: the best results use or with NAG, values at which CM would oscillate severely or diverge (as confirmed in Tables 1 and 5, where CM performance degrades at these high values while NAG continues to improve).
The Momentum Schedule
The momentum coefficient is not held constant throughout training. Instead, the paper uses a schedule that gradually increases from a low initial value to a maximum ceiling , following this formula:
where is the iteration number (starting at 1 for the first update), is the integer division of by 250 (the floor operation), and is the ceiling chosen from .
What this schedule computes: The inner term produces a sequence of values that increase over time. Let's trace through the first several values. At , , so the term is . At , , so the term is . At , , so the term is . As grows, the inner term approaches 1 from below β it starts at 0.5, then 0.75, then approximately 0.833, 0.875, 0.9, 0.917, 0.933, 0.941, 0.95, 0.957, 0.962, and converges toward 1 β but never exceeds because the outer operation caps it. The parameter 250 in the denominator controls how quickly increases: it changes value every 250 iterations (since the floor operation only changes when crosses a multiple of 250).
Why this schedule: The paper states that this schedule is "motivated by Nesterov (1983) who advocates using what amounts to after some manipulation, and by Nesterov (2003) who advocates a constant that depends on (essentially) the condition number." The two Nesterov references correspond to two different theoretical settings. For general smooth convex functions (not strongly convex), the optimal schedule is , which decays to 1 at a rate of O(1/t) β the momentum grows slowly over time, never reaching 1, because the function is not curved enough to benefit from a constant high momentum. For strongly convex functions (with a finite condition number ), the optimal choice is a constant , set based on the condition number, which achieves exponential convergence. The paper's schedule blends these: it increases toward a ceiling (like the constant schedule for strongly convex functions), but does so gradually (like the increasing schedule for general convex functions). The form produces a schedule that increases in discrete steps, with the step frequency controlled by 250. The specific constants (base 2, offset -1, division by 250) appear to be chosen empirically to produce a range of schedules by sweeping only the single hyperparameter .
Operational meaning: At the start of training, is low (0.5), meaning the optimizer behaves more like plain SGD with a small velocity β it can quickly respond to the gradient and doesn't commit strongly to any particular direction. As training progresses and the optimizer encounters persistent gradient structure (directions of consistent reduction), increases, allowing the velocity to accumulate and accelerate along these low-curvature directions. By capping at , the schedule prevents the velocity from growing unboundedly. The schedule thus implements a natural progression: exploration early (low momentum, responsive to gradients), exploitation later (high momentum, committed to persistent descent directions).
For recurrent networks, the schedule is simpler: for the first 1000 parameter updates, after which , where is a constant selected from . This is a two-phase schedule: a lower momentum warmup (0.9 for 1000 steps), then a constant high momentum thereafter. The paper does not use the gradually increasing schedule for RNNs, presumably because the RNN problems require fewer total updates (50,000) and the dynamic evolution of over many repeated doublings would be harder to tune.
Learning Rate Selection and Fine-Tuning Protocol
For each combination of momentum type (NAG or CM) and (or for RNNs), the learning rate is selected to minimize the final training error after the fixed number of updates. The paper sweeps over a discrete set of candidate learning rates and selects the best.
For deep autoencoders, the candidate learning rates are . Each candidate is run for 750,000 parameter updates on mini-batches of size 200, and the learning rate achieving the lowest final training error is reported. The paper does not use learning rate decay β remains constant throughout the 750,000 updates. The mini-batch size of 200 is fixed across all autoencoder experiments.
For recurrent neural networks, the candidate learning rates are , substantially smaller than those used for feed-forward networks. Each candidate is run for 50,000 parameter updates on mini-batches of 100 sequences. The learning rate is also constant throughout training. The paper notes that RNNs require "a particularly small learning rate (as compared with feedforward networks)" (Section 4.2), which is consistent with the greater instability of gradient estimates in recurrent architectures with long temporal dependencies.
Reducing momentum at the end of training (fine-tuning phase). For the autoencoder experiments, the paper introduces an additional phase after the main 750,000 updates: "during the final 1000 parameter updates of the optimization," is reduced "to 0.9 (unless is 0, in which case it is unchanged) without reducing the learning rate" (Section 3). This means that for all runs where (which is all non-zero momentum runs), the final 1000 iterations use regardless of the original , and the learning rate remains at its originally selected value. For runs with (no momentum), stays at 0.
Why this phase shift: The paper justifies this with the two-phase model of optimization from Darken & Moody (1993). The first phase β the transient phase β is where momentum is most beneficial: the optimizer is far from a good minimum, gradient signals exhibit persistent directional structure, and high momentum accelerates movement along low-curvature valleys. The second phase β the fine convergence phase β is where the optimizer is near a local minimum and the problem becomes more like estimation (averaging out gradient noise to settle precisely into the minimum). In this second phase, high momentum becomes counterproductive: the large velocity overshoots the minimum, and the oscillatory behavior in high-curvature directions prevents precise convergence. Reducing to 0.9 (or 0) allows the optimizer to "settle" and achieve a lower final error.
The paper explicitly warns against reducing too early: "it may be tempting then to use lower values of from the outset, or to reduce it immediately when progress in reducing the error appears to slow down. However, in our experiments we found that doing this was detrimental in terms of the final errors we could achieve" (Section 3). The reasoning is that even when the error appears to plateau or become non-monotonic at high , the optimizer is making useful progress along low-curvature directions that will pay off later, and reducing momentum prematurely prevents this long-range progress. The fine-tuning phase should only be engaged after the transient phase has fully run its course.
For RNNs, no analogous fine-tuning phase is described. The momentum schedule for RNNs is already two-phase (0.9 for 1000 steps, then constant ), and there is no final reduction at the end of the 50,000 updates. This may be because the RNN experiments terminate after fewer total updates (50,000 vs. 750,000) and the optimizer may not have entered the fine convergence phase by completion, or because the RNN problems exhibit a different optimization structure where the transient phase dominates even more completely.
The Connection Between Momentum and Hessian-Free Optimization
Section 5 of the paper develops a conceptual bridge between momentum methods and Hessian-Free (HF) optimization. This connection is not mathematical β there is no theorem showing equivalence β but rather structural: the paper argues that HF already derives some of its benefits from a momentum-like mechanism, and that making this mechanism more explicit can improve HF.
How HF works (condensed). HF (Martens, 2010) is a truncated-Newton method. At each parameter update, it constructs a local quadratic approximation to the objective using the gradient and a matrix-vector-product-based approximation to the Hessian, and approximately minimizes this quadratic using the linear conjugate gradient (CG) algorithm. CG is itself a first-order method β it iteratively computes search directions and step sizes within the quadratic subproblem. Critically, the number of CG iterations per HF update is typically small (a few dozen to a few hundred); CG is terminated before full convergence, hence "truncated" Newton.
The momentum-like mechanism in HF. The paper identifies two aspects of HF that resemble momentum:
-
CG initializations ("hot-starting"): Standard truncated-Newton methods initialize CG from zero at each outer iteration. HF instead initializes CG from the solution found at the previous CG call β the previous update direction. This means that information computed during one HF update persists into the next. The persisted solution may be well-converged along some directions (typically low-curvature ones where CG converges slowly) and poorly converged along others, but to the extent that the new quadratic resembles the old one, the old solution provides a head start. This persistence of information across updates is directly analogous to momentum: the velocity vector in CM/NAG is essentially the accumulated gradient history, providing a starting direction for the next update. The CG initialization in HF serves a similar role, accumulating information about persistent descent directions across updates.
-
Single-step CG as NAG: If CG is terminated after exactly one iteration per HF update, HF reduces to a gradient step from the current point plus the previous update reapplied β which (with appropriate choices of the damping parameter) is essentially NAG with a curvature-dependent learning rate rather than a fixed . The paper states: "if CG terminated after just 1 step, HF becomes equivalent to NAG, except that it uses a special formula based on the curvature matrix for the learning rate instead of a fixed constant." The number of CG iterations thus acts as a "dial": more CG iterations make HF behave more like a full second-order method (converging the quadratic subproblem more completely), while fewer iterations make it behave more like momentum.
-
Decay constant in HF: The most effective implementations of HF employ a structural damping parameter (Tikhonov regularization added to the quadratic model) that is decayed over time. This damping parameter plays a role analogous to in momentum: higher damping makes the quadratic model more conservative and the steps smaller, mimicking lower momentum; decaying the damping over time allows larger steps and faster progress, mimicking increasing momentum.
The modified HF experiment. Inspired by the success of NAG, the paper experiments with making HF "behave even more like NAG than it already does" (Section 5). The details are in Appendix A.6, but the key modification is to strengthen the momentum-like persistence in HF by changing the CG initialization strategy. The resulting variant, labeled HFβ in Table 1, achieves training squared errors of 0.058 on Curves, 0.69 on MNIST, and 7.5 on Faces β all substantially lower than the standard HF results from Martens (2010) (labeled HF*: 0.11, 1.40, 12.0 respectively) and competitive with or better than the best NAG results (0.074, 0.73, 7.7). This supports the paper's thesis that momentum-like mechanisms are a common thread connecting first-order and second-order methods in deep learning.
The CG vs. NAG comparison on a real quadratic (Appendix Figure 5). To directly compare the convergence properties of CG and NAG on the kinds of quadratics that arise during deep network training, the paper extracts a quadratic model from the middle of an HF training run on the Curves dataset and runs both CG (initialized from zero) and NAG (initialized from zero) on this quadratic. The results show that CG converges faster β which is expected, since CG is provably optimal for quadratics β but the margin is not enormous. The paper interprets this as evidence that "the quadratics which arise during the optimization of neural networks by HF" may not be of the worst-case variety where CG dramatically outperforms NAG. Combined with the fact that the quadratic approximation itself degrades as CG iterates (since the parameters move away from the point where the quadratic was constructed), the practical advantage of full CG over NAG on these problems may be modest β explaining why a well-tuned first-order method can match or exceed HF.
Summary of Design Choices and Their Justifications
-
Sparse initialization (15 fan-in) over dense initialization: prevents saturation and homogenization in deep sigmoidal networks by controlling the variance of the total input to each unit (independent of layer size) and ensuring diverse receptive fields across units. Dense initialization causes all units in a layer to compute near-identical functions and saturates sigmoid units when layers are wide.
-
Spectral radius 1.1 for RNN hidden-to-hidden matrix over values β€ 1 or β« 1: keeps the dynamics at the edge of chaos β oscillatory enough to retain information over long time spans, but not so chaotic that gradients explode. Values β€ 1 cause forgetting; values β« 1 cause exploding gradients.
-
Task-dependent input-to-hidden scale (0.001 vs. 0.1) for RNNs: a small scale prevents irrelevant distractor inputs from overwriting useful hidden state information, but slows learning when distractors are absent. A larger scale accelerates learning when all inputs are relevant.
-
Nesterov's accelerated gradient over classical momentum: NAG is strictly more stable for the same ΞΌ and Ξ΅ because the lookahead gradient evaluation provides curvature-dependent damping (effective momentum is ), automatically reducing oscillations in high-curvature directions while preserving acceleration in low-curvature directions. CM applies uniform momentum and requires smaller ΞΌ to avoid instability, sacrificing acceleration.
-
Gradually increasing momentum schedule over constant ΞΌ: matches the two-phase structure of deep network optimization β low momentum initially allows responsive gradient-following and exploration, while high momentum later accelerates along persistent low-curvature directions. Constant high momentum from the start would be unstable; constant low momentum would fail to accelerate adequately in the transient phase. The schedule is inspired by Nesterov's theoretical schedules for convex optimization.
-
Momentum reduction (ΞΌ β 0.9) during final 1000 iterations over constant ΞΌ to the end: shifts from the transient (acceleration) phase to the fine convergence (estimation) phase. High momentum near a minimum causes overshooting and oscillation that prevent precise convergence; reducing it allows the optimizer to settle into a better local minimum. The reduction is applied late to avoid sacrificing the long-range progress made possible by high momentum.
-
Fixed learning rate (no decay) over learning rate schedules: the paper sweeps a grid of constant learning rates and picks the best. No annealing or decay is used. This simplifies the hyperparameter space (only one learning rate parameter rather than a schedule) and relies on the momentum schedule to control the effective step size over time. The momentum reduction at the end of training implicitly reduces the effective step size in high-curvature directions, providing some of the benefit that learning rate decay would confer.
4. Key Insights and Innovations
Innovation 1: The Failure of First-Order Methods Is a Diagnosis of Two Independent Factors β Initialization AND Momentum β Not a Single Deficiency
The paper's most fundamental intellectual move is reframing the deep network optimization problem from a question about which optimizer to use to a question about which conditions must be satisfied for any first-order optimizer to succeed. Prior work treated the difficulty of training deep networks as a monolithic problem β either "SGD doesn't work" (leading researchers to develop pre-training or second-order methods) or "SGD works fine with good initialization" (the emerging counter-narrative from Glorot & Bengio (2010) and Chapelle & Erhan (2011)). The paper cuts through this binary by identifying two independent, jointly necessary conditions: the network must start in a region of parameter space from which gradient-based optimization is possible (initialization), AND the optimizer must efficiently navigate the low-curvature directions that dominate the loss landscape during the transient phase (momentum). Neither condition alone is sufficient, and prior work failed because it systematically satisfied at most one of them.
This is a diagnostic framework, not just an empirical recipe. The paper's abstract states it precisely: "poorly initialized networks cannot be trained with momentum and well-initialized networks perform markedly worse when the momentum is absent or poorly tuned." The symmetry is the insight β it argues that the field's prior negative results on momentum (Orr, 1996; LeCun et al., 1998) were largely artifacts of testing momentum on poorly initialized networks, while the field's positive results with good initializations (Chapelle & Erhan, 2011) were limited by inadequate momentum. Both communities made the same logical error: assuming that success or failure of the training pipeline was attributable to a single factor, when the two factors interact multiplicatively.
The significance of this reframing extends beyond the paper's empirical results. It provides a unified explanation for why the literature contained contradictory findings: papers that found SGD worked (Glorot & Bengio, 2010) happened to use good initializations but missed the additional gains from momentum; papers that found momentum ineffective (Wiegerinck et al., 1999) used poor initializations where no amount of momentum could help; papers that found pre-training necessary (Hinton et al., 2006) were compensating for poor random initialization with a better starting point, but didn't realize that a better random initialization plus momentum could achieve the same effect. The paper resolves these contradictions not by picking a winning side but by showing they were answering different parts of the same question.
The evidence supporting this two-factor diagnosis is embedded in the structure of the experiments. Table 1 shows that fixing initialization (sparse initialization) and varying momentum (ΞΌ_max from 0 to 0.999) produces a more than 6Γ reduction in training error on Curves (0.48 β 0.074). Table 3 shows that fixing momentum and varying the initialization scale produces order-of-magnitude changes (0.074 for scale 1 vs. 16 for scale 0.25). The interaction is visible in the RNN results (Table 5): with the ESN-inspired initialization, even ΞΌ = 0 achieves non-trivial results on the addition problem (0.82 zero-one loss vs. 0.82 for the bias-only baseline β essentially no learning), but high momentum brings the error down by three orders of magnitude (to 0.00025 with NAG at ΞΌβ = 0.995). The initialization makes learning possible; the momentum makes it efficient.
This is not an incremental improvement over prior work β it is a fundamental conceptual reframing. Before this paper, the question was "can first-order methods train deep networks?" After this paper, the question becomes "what initialization and momentum schedule are jointly optimal for this architecture and task?" The paper changed what it means to investigate optimization for deep learning.
Innovation 2: The Transient Phase, Not Asymptotic Convergence, Is the Bottleneck β and Momentum Matters Precisely There
The paper's second major conceptual contribution is a re-diagnosis of where the computational cost in deep network training actually occurs, and consequently a re-interpretation of what optimization theory is relevant. Prior theoretical analyses of momentum (Polyak, 1964; Orr, 1996; Wiegerinck et al., 1999) evaluated algorithms by their asymptotic local convergence rate β how quickly they approach the exact minimum once already in its vicinity, in the limit of many iterations. Under this metric, momentum provides no benefit in the stochastic setting: the convergence rate remains O(1/T) with or without momentum when gradients are noisy, because the bottleneck is gradient variance (the Ο/βT term), not curvature (the L/T term). This theoretical result, combined with experiments confirming it on shallow networks, led the field to conclude that momentum was at best a minor tuning knob and at worst a distraction.
The paper's insight β building on the distinction made by Darken & Moody (1993) but applying it to deep networks for the first time β is that this asymptotic analysis applies to the wrong phase of learning. Deep network training is dominated by the transient phase: the initial period where the optimizer moves from a random initialization through highly non-convex terrain to reach a good region of parameter space. During this phase, the gradient signal is not pure noise β there are persistent directions of reduction corresponding to the low-curvature valleys of the loss landscape β and the L/T term in the convergence bound (which momentum improves from O(1/T) to O(1/TΒ²)) dominates over the Ο/βT term. Momentum provides its theoretical acceleration precisely when it matters most.
The paper makes this argument explicit in Section 2, citing the convergence rate of accelerated stochastic gradient methods (Lan, 2010): O(L/TΒ² + Ο/βT). The acceleration from momentum only affects the first term, and this term is dominant early in training. As training proceeds and the optimizer approaches a minimum, the second term takes over, and momentum's benefit disappears β which is exactly why the paper recommends reducing ΞΌ to 0.9 for the final 1000 iterations (Table 2 shows this fine-tuning phase improves error from 0.096 to 0.074 on Curves, a ~23% reduction). The two-phase schedule (high ΞΌ for the transient phase, reduced ΞΌ for fine convergence) is a direct operationalization of this theoretical insight.
What makes this more than a simple application of known theory is the empirical demonstration that the transient phase is long enough to dominate total computation. The paper argues that "while asymptotically it is the second phase which must eventually dominate computation time, in practice it seems that for deeper networks in particular, the first phase dominates overall computation time" (Section 3). This is an empirical claim about the structure of deep network loss landscapes, not a theoretical one. The evidence is in the sensitivity to ΞΌ_max: if the asymptotic phase dominated, the choice of ΞΌ_max would matter little, since all values would eventually converge to similar minima. Instead, Table 1 shows that ΞΌ_max = 0.999 with NAG achieves 0.074 on Curves while ΞΌ_max = 0 achieves 0.48 β a 6.5Γ difference β after the same 750,000 iterations. The optimizer with low ΞΌ hasn't just converged more slowly; it has failed to reach a comparably good region of parameter space at all, because it couldn't make sufficient progress along low-curvature directions during the transient phase.
This insight fundamentally changes how one should think about optimizer design for deep learning. The relevant metric is not asymptotic convergence rate but transient-phase efficiency β how quickly and reliably the optimizer can traverse the complex, ill-conditioned terrain between a random initialization and a good basin of attraction. Second-order methods like HF were effective precisely because they addressed this phase (through curvature reweighting), but the paper shows that momentum, properly configured, addresses it nearly as well with far less complexity. The implication is that the field's prior focus on asymptotic theory systematically misled researchers about which algorithmic properties matter for deep learning, and that returning to the transient-phase analysis of Darken & Moody (1993) β largely ignored in the intervening two decades β is essential for understanding deep network optimization.
Innovation 3: Nesterov's Method as Curvature-Aware Momentum β Unifying Two Historically Separate Research Traditions
The paper's third contribution is a theoretical bridge between classical momentum (from the numerical optimization tradition of Polyak, 1964) and Nesterov's accelerated gradient (from the convex optimization theory tradition of Nesterov, 1983). Before this paper, these were treated as distinct algorithms with separate derivations, separate convergence theories, and separate communities of use. Nesterov's method was formulated using an "estimate sequence" technique that bore no obvious resemblance to momentum; the standard presentation used auxiliary sequences and a momentum-like parameter that was varied according to a specific recurrence (the famous ). Classical momentum was the simple, intuitive velocity accumulation of Polyak (1964). The two methods coexisted without a clear connection.
The paper's reformulation of NAG (Equations 3-4, derived in Appendix A.1) shows that Nesterov's method is operationally identical to classical momentum with a single change: evaluate the gradient at instead of at . This is a conceptual unification of considerable power. It means that all the intuition practitioners had developed about momentum β velocity as accumulated gradient history, acceleration along persistent directions, damping of oscillations β applies directly to Nesterov's method, which inherits those properties but with an additional stabilizing mechanism. And conversely, all the theoretical guarantees developed for Nesterov's method in the convex optimization literature (the O(1/TΒ²) convergence rate for smooth convex functions, the accelerated rates for strongly convex functions) provide a theoretical foundation for understanding why the momentum heuristic works.
The geometric intuition (Figure 1) and the formal analysis on quadratics (Theorem 2.1) provide the mechanism for this unification. NAG's lookahead gradient evaluation produces a curvature-dependent effective momentum: . In high-curvature directions (large Ξ»), momentum is automatically reduced, preventing the oscillations that plague CM. In low-curvature directions (small Ξ»), momentum is essentially unchanged, preserving acceleration. Classical momentum applies uniform momentum and must therefore compromise: set ΞΌ high enough to accelerate low-curvature directions and suffer oscillations in high-curvature ones, or set ΞΌ low enough for stability and sacrifice acceleration. NAG resolves this tension automatically, without requiring per-direction curvature estimates.
This is a fundamental algorithmic insight, not an incremental parameter tuning. The difference between NAG and CM is a single line of code (evaluate the gradient at a slightly different point), but the consequence is a qualitatively different optimization behavior: NAG tolerates ΞΌ values (0.995, 0.999) that would cause CM to oscillate wildly or diverge. Table 1 confirms this sharply: at ΞΌ_max = 0.999, CM achieves 0.10 on Curves (worse than ΞΌ_max = 0.9, which achieves 0.096 at best learning rate), while NAG achieves 0.074. On Faces, CM at ΞΌ_max = 0.999 achieves 9.3 while NAG achieves 7.7. On the RNN problems (Table 5), the pattern is even starker: for the addition problem, NAG at ΞΌβ = 0.995 achieves 0.00025 while CM at the same ΞΌβ achieves 0.036 β over 100Γ worse. High momentum is essential for these problems, and only NAG can use it stably.
The significance of this unification extends beyond the empirical results. It means that practitioners can treat NAG as a drop-in replacement for CM that is strictly more robust to the choice of ΞΌ and Ξ΅. It means that the large body of convex optimization theory for Nesterov's method (which had seemed abstract and disconnected from neural network practice) provides directly applicable guidance for hyperparameter scheduling in deep learning β the paper's momentum schedule is explicitly motivated by Nesterov's theoretical schedules. And it suggests that future optimizer design should focus on mechanisms that provide curvature-awareness without explicit curvature computation, of which NAG's lookahead trick is a particularly elegant example. The connection to Hessian-Free optimization in Section 5 β where HF's CG initializations are framed as another curvature-aware momentum mechanism β reinforces this as a general principle rather than a one-off trick.
Innovation 4: Hessian-Free Optimization as a Momentum Method β Blurring the First-Order / Second-Order Boundary
The paper's final conceptual contribution is a re-interpretation of Hessian-Free optimization as occupying a continuum with momentum methods, rather than being a categorically different class of algorithm. This reframing has implications for both understanding why first-order methods can match second-order ones and for designing better optimization algorithms of either type.
The standard narrative before this paper treated HF (Martens, 2010) as a second-order method that succeeded where first-order methods failed because it used curvature information to reweight gradient steps. The implicit assumption was that the curvature matrix itself β the explicit computation and inversion of Hessian-vector products β was the source of HF's advantage. The paper challenges this assumption by identifying momentum-like mechanisms within HF that are independent of (and potentially more important than) the curvature reweighting.
The key observation is that HF's practice of "hot-starting" conjugate gradient from the previous solution β rather than from zero, as in standard truncated-Newton methods β effectively persists information across outer iterations in a manner structurally analogous to momentum. When CG is run for many iterations per HF update, this persistence matters less (CG converges the quadratic well regardless of initialization). But when CG is terminated early (as is standard in HF to limit computational cost), the initialization from the previous solution strongly shapes the resulting update. In the extreme of a single CG iteration per HF update, HF becomes equivalent to NAG with a curvature-dependent learning rate. The number of CG iterations thus acts as a dial between momentum-like behavior (few iterations) and full second-order behavior (many iterations).
This framing makes the paper's empirical results β that well-tuned NAG can match or exceed HF on deep autoencoders β conceptually coherent rather than surprising. If HF already derives much of its benefit from a momentum-like mechanism, then a sufficiently well-tuned explicit momentum method should be able to recover similar performance, while avoiding the computational cost of Hessian-vector products. The remaining advantage of HF (its ability to use curvature information to set per-direction learning rates) is real but modest on these problems, as the CG vs. NAG comparison on a real training quadratic (Appendix Figure 5) suggests.
The paper strengthens this argument by showing that making HF more momentum-like improves it. The modified HF variant (HFβ in Table 1) achieves 0.058 on Curves vs. 0.074 for the best NAG and 0.11 for standard HF, suggesting that the momentum aspect of HF was underexploited in the standard configuration. This is not a result that follows naturally from thinking of HF as a pure second-order method β under that framing, HF should be improved by making curvature estimates more accurate, not by strengthening momentum persistence. The fact that a momentum-inspired modification improves HF is evidence for the paper's thesis that momentum-like persistent information is the core mechanism shared across both algorithm classes, and that understanding this mechanism leads to improvements regardless of which class one works in.
This insight has had lasting impact on the field. Modern optimizers like Adam (Kingma & Ba, 2015), which postdate this paper by two years, can be seen as combining explicit momentum (the first-moment estimate) with curvature-aware per-dimension learning rates (the second-moment estimate) β essentially a hybrid of the momentum and curvature-reweighting aspects that this paper identified as complementary. The paper's analytical framework β decomposing optimization algorithms into how they persist information across iterations and how they adapt to curvature β remains the standard lens through which new optimizers are understood and compared.
The evidence for this reframing is in Table 1 (showing that NAG, HFβ , and HF* occupy a spectrum of performance rather than separate clusters, with NAG achieving the best results on two of three problems), in the scaling of performance with ΞΌ_max (showing that momentum strength is the primary driver of final error, consistent with the momentum-as-core-mechanism hypothesis), and in the CG vs. NAG comparison (Appendix Figure 5, showing that CG converges faster but not dramatically so, suggesting that the quadratic model's limited validity may cap the benefit of exact curvature information). Together, these results support the paper's central claim: the boundary between first-order and second-order methods is blurrier than previously believed, and the path to better deep network optimizers lies in understanding the momentum-like mechanisms that both classes share, not in pursuing ever-more-accurate curvature estimates.
5. Experimental Analysis
Evaluation Methodology
-
Dataset. The paper uses three deep autoencoder problems from Hinton & Salakhutdinov (2006): Curves, MNIST, and Faces (described in Appendix A.2). These are some of the deepest neural networks with published results at the time, ranging from 7 to 11 layers, and have become standard benchmarks for deep network optimization (as cited by Martens, 2010; Glorot & Bengio, 2010; Chapelle & Erhan, 2011). For recurrent neural networks, the paper uses four artificial sequence modeling problems from Hochreiter & Schmidhuber (1997): addition (T=80), multiplication (T=80), mem-5 (T=200), and mem-20 (T=80), where T is the temporal span over which dependencies must be learned (details in Appendix A.3). These tasks were specifically designed to test long-range temporal dependency learning and were considered impossibly difficult for first-order methods prior to Martens & Sutskever (2011).
-
Base model(s). For the autoencoder experiments, the paper trains standard feed-forward neural networks with sigmoid nonlinearities, following the exact architectures from Hinton & Salakhutdinov (2006): Curves uses a 784-400-200-100-50-25-6 encoder and symmetric decoder (11 layers total), MNIST uses a 784-1000-500-250-30 encoder/decoder (7 layers), and Faces uses a 625-2000-1000-500-30 encoder/decoder (7 layers). For the RNN experiments, a single recurrent neural network with 100 standard tanh hidden units is used, identical to the model in Martens & Sutskever (2011). The paper argues that these architectures are "representative" deep models whose optimization difficulty motivated the development of both pre-training and Hessian-Free methods β making them the right testbed for demonstrating that first-order methods suffice.
-
Metrics. All autoencoder experiments report training squared error β the mean squared reconstruction error on the training set β not test error. The paper explicitly justifies this choice: "test error depends strongly on the amount of overfitting in these problems, which in turn depends on the type and amount of regularization used during training. While regularization is an issue of vital importance... it is outside the scope of our discussion" (Section 3). For RNN experiments, the paper reports zero-one loss (fraction of incorrect predictions), described as "more interpretable" than the squared error or cross-entropy being minimized. For the addition and multiplication problems, a prediction is considered incorrect if the error in the final output exceeds 0.04. For the bit memorization problems, the fraction of timesteps computed incorrectly is reported.
-
Baselines. The paper compares against several published results treated as baselines: (1) Plain SGD without momentum (ΞΌ = 0), which represents the default first-order method whose inadequacy motivated the paper; (2) Classical momentum (CM) at various ΞΌ values, serving as the direct comparison point for NAG; (3) The results of Chapelle & Erhan (2011) (labeled SGDC in Table 1), who used 1.7M SGD steps with tanh networks and the Glorot & Bengio (2010) initialization β representing the state-of-the-art for well-initialized first-order methods prior to this work; (4) Standard Hessian-Free optimization from Martens (2010) (labeled HF* in Table 1), representing the best published results on these benchmarks; (5) A bias-only baseline for RNNs (Table 5, "biases" column), representing the error of an RNN that learns only output biases while ignoring the hidden state β effectively the error floor for a network that fails to learn temporal dependencies.
-
Generation budget / compute accounting. The primary unit of compute is the number of parameter updates (iterations): 750,000 for all autoencoder experiments and 50,000 for all RNN experiments. All autoencoder experiments use mini-batches of size 200; all RNN experiments use mini-batches of 100 sequences. The paper does not report wall-clock time or FLOP counts, and all comparisons between methods are at equal numbers of updates. This means that HF, which is substantially more expensive per update (requiring multiple CG iterations with Hessian-vector products), is compared against SGD at the same iteration count rather than the same wall-clock time β a choice that is generous to the SGD methods being advocated. The paper does not discuss this asymmetry, though the practical implication is clear: if SGD achieves better results in the same number of updates, and each SGD update is cheaper, the advantage is further amplified in practice.
-
Cross-validation / statistical protocol. The paper does not employ cross-validation or hold-out validation sets for hyperparameter selection. Instead, for each combination of momentum type and ΞΌ_max (or ΞΌβ for RNNs), the learning rate Ξ΅ is selected by grid search over a discrete set ({0.05, 0.01, 0.005, 0.001, 0.0005, 0.0001} for autoencoders; {10β»Β³, 10β»β΄, 10β»β΅, 10β»βΆ} for RNNs), and the configuration achieving the lowest final training error on the training set is reported β effectively selecting hyperparameters on the test metric. For the RNN experiments, results are averaged over 4 different random seeds (stated in Section 4.2), providing some measure of statistical reliability. For the autoencoder experiments, no averaging over seeds is reported, and the paper does not state whether the results in Table 1 represent single runs or averages. The absence of a separate hyperparameter validation set means the reported training errors may reflect some degree of overfitting to the training data through hyperparameter selection, though the paper argues this is mitigated by the fact that "undertrained solutions are known to perform poorly on both the training and test sets (underfitting)" for these problems (Section 3).
Main Quantitative Results
Deep Autoencoder Training Errors (Table 1)
The central result for feed-forward networks is Table 1, which reports the final training squared error on Curves, MNIST, and Faces after 750,000 parameter updates. The headline finding is that NAG with high ΞΌ_max achieves the lowest training errors on all three problems, surpassing both previously published Hessian-Free results and the SGD results of Chapelle & Erhan (2011).
On Curves (the 11-layer autoencoder): The best NAG result (ΞΌ_max = 0.999 with the optimal learning rate) achieves a training squared error of 0.074. This compares to 0.48 for SGD without momentum (ΞΌ=0) β a 6.5Γ reduction β demonstrating the magnitude of improvement attributable to momentum. The best CM result at the same ΞΌ_max (0.999) achieves only 0.10, meaning NAG outperforms CM by approximately 26% at the highest momentum setting. Standard HF from Martens (2010) (HF*) achieves 0.11, while the improved momentum-inspired HF variant (HFβ ) achieves 0.058 β the single best result on this problem. Chapelle & Erhan (2011)'s SGD result (SGDC) is 0.16, approximately 2.2Γ worse than the best NAG result. The progression with increasing ΞΌ_max is monotonic for NAG: 0.48 (ΞΌ=0) β 0.16 (0.9) β 0.096 (0.99) β 0.091 (0.995) β 0.074 (0.999). For CM, the progression is non-monotonic: 0.15 (0.9) β 0.10 (0.99) β 0.10 (0.995) β 0.10 (0.999), showing that CM plateaus and then degrades as ΞΌ increases beyond 0.99.
On MNIST (the 7-layer autoencoder): The best NAG result (ΞΌ_max = 0.99) achieves 0.73. SGD without momentum achieves 2.1 β a 2.9Γ reduction. Unlike Curves, the optimal ΞΌ_max for NAG on MNIST is 0.99, not 0.999 (0.73 for 0.99 vs. 0.75 for 0.995 vs. 0.80 for 0.999), indicating that the optimal momentum strength is problem-dependent. The best CM result is 0.77 at ΞΌ_max = 0.99. SGDC achieves 0.9, meaning the best NAG result represents a 19% improvement over the prior state-of-the-art for first-order methods. HF* achieves 1.40 and HFβ achieves 0.69 β the latter is slightly better than NAG (0.73) but the gap is small (approximately 5%).
On Faces (the 7-layer autoencoder on a different dataset): The best NAG result (ΞΌ_max = 0.999) achieves 7.7, with ΞΌ_max = 0.995 achieving a nearly identical 7.8. SGD without momentum achieves 36.4 β a 4.7Γ reduction, the largest relative improvement across the three problems. CM at ΞΌ_max = 0.999 achieves 9.3, meaning NAG outperforms CM by approximately 17% at the highest momentum. SGDC is not reported for Faces. HF* achieves 12.0 and HFβ achieves 7.5 β HFβ slightly edges out NAG (7.5 vs. 7.7), but by a margin of only ~2.6%.
Pattern across problems: The optimal ΞΌ_max varies by problem (0.999 for Curves and Faces, 0.99 for MNIST), suggesting that the optimal momentum strength is not a universal constant but depends on the optimization landscape. NAG consistently outperforms CM at high ΞΌ values (0.995 and 0.999), with the gap widening as ΞΌ increases β consistent with Theorem 2.1's prediction that NAG's curvature-dependent effective momentum provides greater stability. The performance gap between SGD (ΞΌ=0) and the best NAG result ranges from 2.9Γ (MNIST) to 6.5Γ (Curves), demonstrating that momentum is not a minor tuning knob but a dominant factor in final training error.
Effect of Momentum Reduction at End of Training (Table 2)
Table 2 quantifies the benefit of reducing ΞΌ to 0.9 during the final 1000 parameter updates. For NAG with the optimal configuration: Curves improves from 0.096 to 0.074 (a 23% relative reduction in error); MNIST improves from 1.20 to 0.73 (a 39% reduction); Faces improves from 10.83 to 7.7 (a 29% reduction). These improvements are substantial β the fine-tuning phase accounts for a significant fraction of the total error reduction, confirming the paper's two-phase model where high momentum enables long-range progress during the transient phase but must be dialed back for precise convergence. Notably, the "before" values (0.096, 1.20, 10.83) are themselves already better than the ΞΌ=0 baselines (0.48, 2.1, 36.4), meaning the fine-tuning phase builds on top of the gains from high-momentum optimization rather than compensating for its failures.
The paper's claim that reducing ΞΌ too early is detrimental is supported indirectly by the structure of the results: the optimal configuration uses high ΞΌ for the vast majority of training (749,000 of 750,000 iterations) and only reduces it at the very end. If reducing ΞΌ earlier were beneficial, a configuration with a lower ΞΌ_max throughout training would outperform, but Table 1 shows the opposite β higher ΞΌ_max consistently achieves lower final error (for NAG) even before the fine-tuning reduction.
Recurrent Neural Network Results (Table 5)
Table 5 reports the zero-one losses on the four long-term dependency problems after 50,000 parameter updates, averaged over 4 random seeds. The headline finding is that NAG with high ΞΌβ successfully trains standard RNNs on problems previously considered impossible for first-order methods, while CM at the same ΞΌβ values performs substantially worse.
Addition (T=80): The bias-only baseline achieves 0.82, meaning a network that ignores the hidden state gets 82% of examples wrong. SGD without momentum (ΞΌβ = 0) achieves 0.82 β identical to the bias-only baseline β confirming that plain SGD completely fails to learn temporal dependencies on this problem. NAG at ΞΌβ = 0.9 achieves 0.39, at ΞΌβ = 0.98 achieves 0.02, and at ΞΌβ = 0.995 achieves 0.00025 β effectively solving the problem with near-zero error. CM at the same ΞΌβ values achieves 0.43 (0.9), 0.62 (0.98), and 0.036 (0.995). The gap between NAG and CM at ΞΌβ = 0.995 is over 100Γ (0.00025 vs. 0.036), and CM at ΞΌβ = 0.995 is actually worse than at ΞΌβ = 0.9 (0.036 vs. 0.43, where smaller is better β wait, 0.036 < 0.43, so CM does improve from 0.9 to 0.995, but the best CM result at ΞΌβ = 0.9 is 0.43, and the best at 0.995 is 0.036 β the text in the table shows 0.43, 0.62, 0.036 for CM at ΞΌβ = 0.9, 0.98, 0.995 respectively, so there is a non-monotonic pattern). The best CM result overall on addition is 0.036 at ΞΌβ = 0.995 β still nearly 150Γ worse than the best NAG result of 0.00025.
Multiplication (T=80): The bias-only baseline achieves 0.84. SGD achieves 0.84 β again identical. NAG achieves 0.48 (ΞΌβ = 0.9), 0.36 (0.98), and 0.22 (0.995) β a substantial improvement over baseline but not the near-zero error achieved on addition, indicating that multiplication is a harder problem. CM achieves 0.0013 (0.9), 0.029 (0.98), and 0.025 (0.995). Strikingly, the best CM result (0.0013 at ΞΌβ = 0.9) is substantially better than the best NAG result (0.22 at ΞΌβ = 0.995) β the only case where CM outperforms NAG. However, the paper notes that the multiplication problem required centering of inputs and outputs to be solved, which Martens & Sutskever (2011) did not require. The CM result at ΞΌβ = 0.9 is surprisingly strong, and the paper does not explain this anomaly. The non-monotonic behavior for CM (0.0013 β 0.029 β 0.025) suggests sensitivity to the interaction between momentum and learning rate.
Mem-5 (T=200): The bias-only baseline achieves 2.5 (note: this is a percentage or scaled loss β the units differ across problems). SGD achieves 2.5 β identical to baseline. NAG achieves 1.27 (ΞΌβ = 0.9), 1.02 (0.98), and 0.96 (0.995). CM achieves 0.63 (0.9), 1.12 (0.98), and 1.09 (0.995). Here, CM at ΞΌβ = 0.9 achieves the best result overall (0.63 vs. 0.96 for NAG), though the paper emphasizes that NAG shows monotonic improvement with increasing ΞΌβ while CM degrades as ΞΌβ increases beyond 0.9 β a pattern consistent with NAG's greater tolerance for high momentum.
Mem-20 (T=80): The bias-only baseline achieves 8.0. SGD achieves 8.0. NAG achieves 5.37 (ΞΌβ = 0.9), 2.77 (0.98), and 0.0144 (0.995) β a dramatic improvement that effectively solves the problem at the highest ΞΌβ, improving by over 500Γ from the SGD baseline. CM achieves 0.00005 (0.9), 1.75 (0.98), and 0.0017 (0.995). This is the most extreme non-monotonic pattern for CM: 0.00005 at ΞΌβ = 0.9 is the single best result on this problem (better than NAG's best of 0.0144 by a factor of ~288), but CM at ΞΌβ = 0.98 degrades to 1.75 before recovering partially to 0.0017 at ΞΌβ = 0.995. The paper does not provide a detailed explanation for this extreme sensitivity, but the result highlights that while NAG is robust and predictable as ΞΌβ increases (monotonic improvement), CM can achieve excellent results at particular ΞΌβ values but is brittle β small changes in the momentum parameter cause order-of-magnitude performance swings.
Summary pattern across RNN problems: NAG with high ΞΌβ consistently achieves strong results that improve monotonically as ΞΌβ increases (with the multiplication problem being a partial exception, where improvement saturates at 0.22). CM shows non-monotonic, brittle behavior β sometimes outperforming NAG at specific ΞΌβ values (mem-20 at ΞΌβ = 0.9, multiplication at ΞΌβ = 0.9) but degrading substantially at others, making it unreliable without problem-specific tuning. The paper's claim that "momentum methods can cope with long-range temporal dependency training tasks" is strongly supported for NAG at high ΞΌβ (0.98β0.995), where results on addition (0.00025) and mem-20 (0.0144) represent near-perfect solutions to problems that plain SGD fails on entirely. The comparison with Martens & Sutskever (2011) is less direct β the paper states those results "appear to be moderately better and more robust" but does not provide a side-by-side numerical comparison in Table 5, instead characterizing the HF approach as using "a specialized update damping technique whose benefits seemed mostly limited to training RNNs to solve these kinds of extreme temporal dependency problems."
Ablation Studies and Robustness Checks
Initialization scale for sparse initialization (Table 3): On the Curves dataset with the optimal NAG configuration, varying the scale multiplier applied to the sparse initialization weights produces: scale 1.0 (default) β 0.074 error; scale 2.0 β 0.083 (12% worse); scale 4.0 β 0.35 (4.7Γ worse); scale 0.5 β 16 (216Γ worse); scale 0.25 β 16 (216Γ worse). The initialization is modestly robust to scaling up by a factor of 2, but fails catastrophically when scaled down β halving the initial weights produces a two-order-of-magnitude error increase, demonstrating that weights that are too small are far more damaging than weights that are too large for this architecture and initialization scheme. The paper states that "factor 5 we did not achieve sensible results" for scaling by 5, without providing a numerical value. The sensitivity to downward scaling is consistent with the vanishing gradient problem: if weights are too small, signals cannot propagate through 11 layers of sigmoid nonlinearities, and the network fails to learn.
Momentum type comparison across ΞΌ_max values (Table 1): This is effectively an ablation of the NAG mechanism β comparing CM and NAG at identical ΞΌ_max values reveals the contribution of the lookahead gradient evaluation. For Curves, the NAG advantage over CM grows with ΞΌ_max: at ΞΌ_max = 0.9, NAG (0.16) vs. CM (0.15) β negligible difference; at 0.99, NAG (0.096) vs. CM (0.10) β 4% advantage; at 0.995, NAG (0.091) vs. CM (0.10) β 9% advantage; at 0.999, NAG (0.074) vs. CM (0.10) β 26% advantage. This monotonic growth in NAG's relative advantage as ΞΌ increases is precisely what Theorem 2.1 predicts: NAG's effective momentum reduction in high-curvature directions scales with Ρλ, and this matters more when ΞΌ is large. For MNIST, the pattern is less pronounced: NAG (0.73) vs. CM (0.77) at ΞΌ_max = 0.99, with both degrading at 0.999 (NAG: 0.80, CM: 0.90). For Faces: NAG (7.7) vs. CM (9.3) at ΞΌ_max = 0.999. The ablation confirms that the NAG mechanism provides the largest benefits at the highest ΞΌ values, exactly where CM's uniform momentum becomes unstable.
Momentum schedule design (implicit): The paper does not ablate the specific formula for ΞΌ_t (Equation 5) against alternative schedules (e.g., constant ΞΌ, linear increase, cosine schedule). The two RNN results where CM at ΞΌβ = 0.9 surprisingly outperforms NAG (multiplication: CM 0.0013 vs. NAG 0.22; mem-20: CM 0.00005 vs. NAG 0.0144) suggest that the relationship between ΞΌ schedule and final performance is complex and that the optimal schedule may differ between NAG and CM. The paper attributes NAG's advantage primarily to tolerance of higher ΞΌ, but does not explore whether a different schedule for CM (e.g., starting lower and increasing more gradually) could recover similar performance at high ΞΌ_max.
Learning rate grid (implicit): The choice of learning rate is ablated through grid search, but the paper does not report the optimal Ξ΅ for each configuration, making it impossible to assess whether NAG and CM require different learning rates for the same ΞΌ_max, or whether the optimal Ξ΅ varies with ΞΌ_max in a predictable way. The RNN results use a different grid ({10β»Β³, 10β»β΄, 10β»β΅, 10β»βΆ}) than the autoencoders ({0.05, ..., 0.0001}), reflecting the observation that RNNs require "a particularly small learning rate" β but this is an empirical finding, not an ablated design choice.
HF momentum-like modification (Table 1, HFβ vs. HF):* The improved HF variant, which was modified to "behave even more like NAG" (details in Appendix A.6), achieves substantially lower errors than standard HF on all three problems: Curves: 0.058 vs. 0.11 (47% reduction); MNIST: 0.69 vs. 1.40 (51% reduction); Faces: 7.5 vs. 12.0 (38% reduction). This is an ablation of the hypothesis that HF derives its benefits partly from momentum-like mechanisms β if HF's advantage were purely from curvature reweighting, making it more momentum-like should not produce such large improvements. The fact that it does supports the paper's conceptual framework. However, the HFβ results also slightly surpass the best NAG results on Curves (0.058 vs. 0.074) and Faces (7.5 vs. 7.7), indicating that even with strong momentum, there remains a small residual advantage to incorporating explicit curvature information.
Centering for RNN problems (Section 4.2): The paper states that centering (mean subtraction of inputs and outputs) was "important to reliably solve all of the training problems," and that Martens & Sutskever (2011) did not require centering for the same problems. The multiplication problem is specifically called out as one that required centering, while "the other problems are already centered." This is not a formal ablation β the paper does not report results without centering β but it represents a negative result: the claimed approach (ESN-inspired initialization + NAG) alone was insufficient for the multiplication problem without additional data preprocessing that the prior HF approach did not need.
Task-dependent input-to-hidden scale for RNNs (Section 4.1): The paper reports that the optimal scale for input-to-hidden weights depends on whether the task has many irrelevant distractor inputs. For tasks with distractors (addition, multiplication), a scale of 0.001Β·N(0, 1) is used; for tasks without distractors, a larger scale of 0.1Β·N(0, 1) is used. This is presented as a finding from experimentation ("Having experimented with multiple scales we found that a Gaussian draw with a standard deviation of 0.001 achieved a good balance") but is not a formal ablation with results reported across multiple scale choices for each task. The paper also distinguishes between "add"/"mul" input connections (scale 0.001) and "mem" input connections (scale 0.1) within the same task, indicating that even within a single problem, different input subspaces benefit from different initialization scales β a level of task-specific tuning that represents a limitation on the generality of the proposed initialization.
Critical Assessment
The experiments provide strong support for the paper's central empirical claim β that NAG with a well-designed initialization can train deep autoencoders and RNNs to levels previously achieved only by Hessian-Free optimization β but several qualifying observations are necessary about what the experiments do and do not demonstrate.
On the claim that initialization and momentum jointly account for the HF gap: The evidence is convincing but incomplete. Table 1 shows that NAG + sparse initialization surpasses standard HF (HF*) on two of three problems and matches it on the third (MNIST: 0.73 vs. HFβ 0.69). However, the comparison is not entirely clean. HF* results include L2 regularization (as the paper notes in Section 3: "the previously published results on HF used L2 regularization, so they cannot be directly compared"), while the NAG results use no regularization. The paper runs a modified HF without L2 (HFβ ) to provide a fairer comparison, and HFβ slightly outperforms NAG on Curves (0.058 vs. 0.074) and Faces (7.5 vs. 7.7), and is essentially tied on MNIST (0.69 vs. 0.73). This means the most accurate characterization is: NAG closes most of the gap with HF (compared to the 2-4Γ gap between HF and prior SGD results), but a well-tuned HF still holds a small advantage of ~5-25% depending on the problem.* The paper's abstract claims "levels of performance that were previously achievable only with Hessian-Free optimization" β this is true relative to the published HF* numbers, but the HFβ comparison suggests that the gap is narrowed rather than fully eliminated when HF is also optimized.
A notable gap in the experimental design is the single-run reporting for autoencoder experiments. The paper does not state whether Table 1 results are averages over multiple seeds or single runs. Given that the RNN experiments use 4-seed averaging explicitly, the omission for autoencoders is conspicuous and raises questions about the reliability of the precise numerical comparisons, particularly for small differences (e.g., NAG 0.074 vs. HFβ 0.058 on Curves β is this a 21% difference or within run-to-run variance?).
On the claim that first-order methods fail due to poor initialization AND inadequate momentum: The evidence for initialization sensitivity is strong but limited to a single ablation (Table 3). The paper shows that scaling the sparse initialization by 0.5 or 0.25 causes catastrophic failure (error jumps from 0.074 to 16). However, the paper does not compare sparse initialization against alternative initialization schemes under the same optimized momentum configuration. It is possible that other initializations (e.g., the Glorot & Bengio (2010) normalized initialization) could achieve similar results with the same momentum schedule, which would weaken the claim that sparse initialization is specifically required. The only indirect comparison is to Chapelle & Erhan (2011) (SGDC in Table 1: 0.16 on Curves), who used the Glorot & Bengio initialization with plain SGD β but since their momentum configuration was different (no NAG, no schedule), the comparison conflates initialization and optimization effects. A direct ablation comparing sparse vs. dense vs. Glorot initialization, all with the optimal NAG schedule, is missing.
For RNNs, the situation is more complex. The paper's ESN-inspired initialization involves multiple interacting choices: spectral radius 1.1, 15 fan-in sparsity, task-dependent input scales, input/output centering, and different scales for different input types. With this many degrees of freedom, it is difficult to attribute success to any specific aspect of the initialization. The paper does not ablate these choices individually β for example, what happens if the spectral radius is 1.0 or 1.2 instead of 1.1? What if the fan-in is 50 instead of 15? The claim that "previous attempts to train... recurrent neural networks from random initializations have likely failed due to poor initialization schemes" (abstract) is plausible given the results, but the paper has not demonstrated that the specific initialization choices are necessary rather than merely sufficient.
On the claim that the transient phase, not asymptotic convergence, dominates deep network training: This is a conceptual claim supported indirectly by the structure of results, not by a direct experiment. The evidence is: (1) higher ΞΌ_max achieves lower final error (Table 1), and if the asymptotic phase dominated, the choice of ΞΌ_max would matter less since all configurations would eventually converge similarly; (2) reducing ΞΌ at the very end of training provides a substantial final improvement (Table 2), consistent with a phase transition from transient to asymptotic behavior; (3) the RNN results show that ΞΌβ = 0 (SGD) fails to learn entirely within 50,000 iterations on problems that NAG with high ΞΌβ nearly solves. However, the paper does not run experiments for substantially longer durations to test whether low-ΞΌ configurations eventually catch up β for example, would SGD with ΞΌ=0 on Curves reach 0.074 if run for 7.5M iterations (10Γ longer) instead of 750K? If so, the claim that momentum is essential (rather than merely an accelerator) would be weakened. The paper's framing as a "diagnosis of failure" implicitly assumes that the training budget is fixed and practical, which is reasonable from an engineering perspective, but the asymptotic question matters for understanding whether the optimization landscape itself is fundamentally navigable by first-order methods given unlimited time.
On the claim that NAG is fundamentally more stable than CM (Theorem 2.1 and geometric intuition): The evidence is strong for the high-ΞΌ regime. Across all three autoencoder problems and three of four RNN problems, NAG at the highest ΞΌ values (0.995, 0.999) substantially outperforms CM at the same values. The RNN results contain interesting counterexamples β CM at ΞΌβ = 0.9 achieving 0.00005 on mem-20 vs. NAG's 0.0144, and CM at ΞΌβ = 0.9 achieving 0.0013 on multiplication vs. NAG's 0.22 β which complicate the narrative. These results suggest that while NAG is more robust to high ΞΌ (monotonic improvement as ΞΌ increases), CM can achieve better peak performance at specific lower ΞΌ values on some problems. The paper's claim should perhaps be qualified: NAG tolerates higher ΞΌ than CM for a given Ξ΅, enabling stronger momentum and better results in most cases, but CM is not uniformly worse β it can excel when precisely tuned, but the tuning is brittle and problem-specific.
On the RNN results vs. Martens & Sutskever (2011): The paper claims its results "fall just short" of those reported by Martens & Sutskever (2011), but the comparison is qualitative rather than quantitative in the main text. Table 5 does not include the corresponding HF numbers side-by-side, making it impossible for a reader to assess the magnitude of the gap without consulting the prior paper. The paper acknowledges that the HF approach "achieved lower error rates and their initialization was chosen with less care," and that centering was required for multiplication while HF did not need it. The absence of a direct numerical comparison in the main results table is a weakness β it forces the reader to take the paper's characterization of "falling just short" on faith.
Missing experiments that would strengthen the paper: (1) A direct comparison of CM and NAG with the same momentum schedule but varying ΞΌ continuously to map out the stability regions β the paper samples only 4-5 ΞΌ values per problem, which may miss narrow regions where CM excels; (2) Experiments with much longer training runs to determine whether the relative rankings of methods change with increased budget; (3) Test error results for the autoencoders β while the paper argues that underfitting means training error is a reliable proxy, reporting test error would allow assessment of whether the lower training errors from high momentum translate to better generalization or merely more aggressive overfitting; (4) An ablation of the momentum schedule formula (Equation 5) against simpler alternatives (constant ΞΌ, linear increase) to demonstrate that the specific schedule matters rather than just the final ΞΌ_max; (5) A direct comparison between the ESN-inspired initialization and alternative RNN initializations under identical momentum configurations; (6) Reporting the optimal learning rate for each configuration to allow analysis of the interaction between Ξ΅ and ΞΌ β a table showing which Ξ΅ was selected for each ΞΌ_max would be highly informative.
On generalizability: All autoencoder experiments use sigmoid nonlinearities and the same three datasets. The paper does not test on ReLU networks (which were becoming common by 2013, following Krizhevsky et al., 2012), on convolutional architectures, or on tasks beyond reconstruction. The RNN experiments use a single architecture (100 tanh units) on four synthetic problems. Whether the findings generalize to other nonlinearities, architectures, or real-world sequence tasks (speech, language) is not addressed. The paper's claim that its framework provides "a simple to understand and easy to use framework for deep learning" (Section 1) thus overstates the demonstrated scope β the results are confined to sigmoidal autoencoders and small RNNs on synthetic temporal problems.
On the practical significance of the RNN results: While solving the Hochreiter & Schmidhuber (1997) problems with first-order methods is a striking proof of concept, these are small synthetic tasks (the RNN has only 100 hidden units and the problems involve a single scalar output). The paper does not demonstrate that the approach scales to the larger RNNs and real datasets (e.g., character-level language modeling, speech recognition) where HF had shown success (Sutskever et al., 2011; Mikolov et al., 2012). The claim that the results are "sufficient for most practical purposes" (Section 4.2) is speculative without such experiments.
In summary, the experiments convincingly demonstrate that NAG with the proposed momentum schedule and initialization achieves dramatic improvements over plain SGD and classical momentum on the tested benchmarks, closing most (though not all) of the gap with Hessian-Free optimization. The evidence for the specific theoretical claims β that the transient phase dominates, that Theorem 2.1 explains the NAG-CM difference, that the proposed schedule is optimal β is more circumstantial, relying on consistency between results and theory rather than direct falsification of alternatives. The paper's most important contribution is the empirical demonstration itself: deep networks can be trained from random initializations with first-order methods, and momentum (specifically Nesterov-style, properly scheduled) is the critical missing ingredient that prior work underestimated.
6. Limitations and Trade-offs
6.1 All Results Are on a Single Benchmark Family with a Single Model Architecture and Activation Function
The assumption or constraint. Every experiment in this paper β both feed-forward and recurrent β uses sigmoidal activation functions (standard sigmoid for autoencoders, tanh for RNNs) and a fixed set of benchmark problems: the three deep autoencoder datasets from Hinton & Salakhutdinov (2006) for feed-forward networks, and the four synthetic long-term dependency problems from Hochreiter & Schmidhuber (1997) for RNNs. The paper does not test on ReLU networks, convolutional architectures, or real-world sequence tasks. The base model for autoencoders is a specific family of fully-connected sigmoidal autoencoders; for RNNs, it is a single architecture of 100 tanh hidden units. The authors are transparent that their scope is limited β the abstract and introduction frame the contribution specifically around these "deep autoencoders" and "temporal networks" β but the practical reader needs to know whether the findings transfer to the architectures and nonlinearities that became dominant shortly after this paper's publication.
The consequence. The paper's central finding β that NAG with a specific momentum schedule and initialization matches HF on deep network training β may not hold for networks with ReLU activations, which have fundamentally different gradient propagation properties. Sigmoid and tanh saturate at the extremes, creating the vanishing gradient problem that the paper's initialization is specifically designed to mitigate; ReLU networks suffer from a different pathology (dying ReLUs) and may not benefit from the same initialization principles. Similarly, convolutional networks have extensive parameter sharing that changes the curvature structure of the optimization landscape, and modern RNN architectures (LSTMs, GRUs) have gating mechanisms that explicitly address the vanishing gradient problem the paper's momentum schedule is compensating for. The paper demonstrates that momentum solves a particular failure mode (slow progress along low-curvature directions caused by sigmoidal saturation and poor conditioning), but whether this is the dominant failure mode in other architectures is unknown. A practitioner training a modern ReLU-based network or an LSTM cannot conclude from this paper that the specific ΞΌ schedule or NAG's advantage over CM will persist β the experiments provide no evidence either way.
What evidence exists in the paper. The evidence for this limitation is the absence of any experiments beyond sigmoidal activations and the stated benchmark families. Section 3 introduces the autoencoder experiments with "the networks we trained used the standard sigmoid nonlinearity" and Section 4 specifies "100 standard tanh hidden units" for RNNs. There is no mention of ReLU, no convolutional experiments, and no real-world sequence tasks (e.g., character-level language modeling, where Sutskever et al. (2011) had demonstrated HF's effectiveness). The paper notes this implicitly in Section 4.2 when it states that the RNN results "seem sufficient for most practical purposes" β a claim about generalizability that the experiments do not directly support. The ablation studies are limited to scaling the sparse initialization (Table 3) and do not compare initialization strategies across activation functions.
Mitigation status. The paper does not address this limitation. It does not claim generalizability beyond the tested architectures, but it also does not explicitly warn readers that the findings may be activation-function-specific or architecture-specific. The title's broad framing ("On the importance of initialization and momentum in deep learning") and the abstract's claim that the methods "can train both DNNs and RNNs" imply greater generality than the experiments demonstrate. No future work is suggested to test on alternative nonlinearities or architectures.
6.2 The Difficulty Estimation Cost Appears as an Unaccounted-for Practical Overhead in Deployment
The assumption or constraint. The paper's training recipe depends on an extensive hyperparameter sweep to select the learning rate Ξ΅ and the momentum ceiling ΞΌ_max jointly. For the autoencoder experiments, this involves running the full 750,000-iteration training for each combination of momentum type (NAG or CM), each ΞΌ_max value (5 choices: 0, 0.9, 0.99, 0.995, 0.999), and each candidate learning rate (6 choices: 0.05 to 0.0001) β a total of up to 2 Γ 5 Γ 6 = 60 full training runs per problem to identify the single best configuration reported in Table 1. The RNN experiments involve a similar sweep over ΞΌ_0 values and learning rates, averaged over 4 seeds. The cost of this grid search is not included in the paper's accounting of computational requirements. The paper presents the optimal configuration as the result, not the process of finding it.
The consequence. The reported results β for example, NAG achieving 0.074 on Curves with ΞΌ_max = 0.999 and some optimal Ξ΅ β represent the achievable performance given oracle knowledge of the best hyperparameters, not the performance a practitioner would obtain on a new problem without a comparable hyperparameter sweep. The sweep multiplies the effective computational cost by a factor of ~60Γ over what the headline numbers suggest. On a new task or architecture, a practitioner must either replicate this expensive sweep or risk selecting suboptimal hyperparameters, potentially failing to reproduce the paper's dramatic improvements. The gap between the best reported result and a "reasonable default" configuration is not measured. Tables 1 and 5 show substantial variation across ΞΌ_max values (e.g., on Curves, NAG at ΞΌ_max = 0.9 achieves 0.16 vs. 0.074 at ΞΌ_max = 0.999 β a factor of 2.2Γ difference), so picking the wrong ΞΌ_max or Ξ΅ produces significantly worse results. The paper provides no guidance for selecting hyperparameters without an exhaustive sweep.
What evidence exists in the paper. The evidence is in the experimental methodology itself. Section 3 states: "for each choice of ΞΌ_max, we report the learning rate that achieved the best training error" and "the learning rate Ξ΅ was chosen from {0.05, 0.01, 0.005, 0.001, 0.0005, 0.0001} in order to achieve the lowest final error training error." The paper does not report which learning rate was optimal for each ΞΌ_max, making it impossible for a reader to identify patterns (e.g., "higher ΞΌ always requires lower Ξ΅") that could guide hyperparameter selection on new problems. The RNN experiments use an analogous scheme (Section 4.2): "for each ΞΌ_0, we use the empirically best learning rate chosen from {10β»Β³, 10β»β΄, 10β»β΅, 10β»βΆ}." The variation across problems in optimal ΞΌ_max (0.999 for Curves and Faces, 0.99 for MNIST) demonstrates that the optimal hyperparameters are problem-dependent and cannot be transferred from one task to another.
Mitigation status. The paper does not acknowledge this as a limitation and does not suggest methods for reducing the hyperparameter search cost. The momentum schedule (Equation 5) reduces the ΞΌ hyperparameter to a single scalar (ΞΌ_max) rather than a full schedule, which is a form of dimensionality reduction, but the joint selection of ΞΌ_max and Ξ΅ still requires a two-dimensional grid search. No heuristics, rules of thumb, or adaptive methods are proposed. This is a significant gap between the paper's theoretical contribution (demonstrating that properly configured momentum works) and its practical deployability (providing a method to find that proper configuration without oracle access to final training error).
6.3 The Autoencoder Experiments Report Only Training Error, Not Test Error, Leaving Generalization Unmeasured
The assumption or constraint. All autoencoder results in Table 1 report training squared error exclusively. The paper explicitly justifies this choice in Section 3: "test error depends strongly on the amount of overfitting in these problems, which in turn depends on the type and amount of regularization used during training. While regularization is an issue of vital importance when designing systems of practical utility, it is outside the scope of our discussion." The paper additionally argues that "undertrained solutions are known to perform poorly on both the training and test sets (underfitting)" β implying that lower training error correlates with lower test error in the regimes studied β but provides no evidence for this claim on the specific configurations tested.
The consequence. The paper's central claim β that NAG with proper initialization achieves "levels of performance that were previously achievable only with Hessian-Free optimization" β is supported only for training error, not for the metric that ultimately matters in applications (generalization to unseen data). The relationship between training error and test error depends on the interaction between optimization and implicit regularization. Momentum methods that aggressively minimize training error may do so by converging to sharper minima that generalize worse (a phenomenon later formalized in the "sharp vs. flat minima" literature). Conversely, the paper's finding that reducing ΞΌ at the end of training improves training error (Table 2) might correspond to moving into a wider basin that generalizes better β but this is speculative without test error measurements. The strong momentum configurations (ΞΌ_max = 0.999) that achieve the best training errors might also be the configurations that overfit most aggressively, potentially yielding worse test performance than lower-momentum counterparts. The paper cannot distinguish between "the optimizer found a genuinely better solution that generalizes" and "the optimizer overfit the training set more thoroughly."
What evidence exists in the paper. The evidence is the complete absence of test error numbers for the autoencoder experiments. Section 3 states the justification but provides no data to support the claim that lower training error implies better generalization for these specific problems and optimization configurations. The RNN experiments use synthetic problems where the training data effectively defines the task (the model must learn the underlying function from examples), making the train/test distinction less relevant β but the autoencoder experiments use real datasets (MNIST, Faces) where overfitting is a genuine concern. Table 1 compares NAG results against HF* (which used L2 regularization) and SGDC (Chapelle & Erhan, 2011, who used tanh networks and may have had different generalization properties), but these comparisons are on training error only, making them potentially misleading: a method could achieve lower training error while generalizing worse, appearing superior in Table 1 but worse in practice.
Mitigation status. The paper explicitly acknowledges this limitation but dismisses it as outside scope. The argument that "undertrained solutions perform poorly on both training and test sets" is a claim about underfitting, not about the overfitting regime that high-momentum optimization might enter. The paper does not suggest future work to validate the training-test correlation, does not report test error even for the best single configuration as a sanity check, and does not qualify its headline claims (e.g., "lowest published results") as applying only to training error. A practitioner evaluating whether to adopt this method would need test error results to assess the generalization tradeoff β the paper provides none.
6.4 The RNN Results Require Extensive Task-Specific Initialization Tuning That Undermines Generality
The assumption or constraint. The ESN-inspired initialization for RNNs involves multiple interacting design choices that the paper determines through task-specific experimentation rather than a principled formula: the spectral radius of the hidden-to-hidden matrix (1.1), the sparsity pattern (15 fan-in), the scale of input-to-hidden weights (0.001 for operand inputs on distractor-heavy tasks, 0.1 for marker inputs and tasks without distractors), and input/output centering (required for multiplication but not for the other problems). The paper states that "having experimented with multiple scales we found that a Gaussian draw with a standard deviation of 0.001 achieved a good balance" (Section 4.1) and that "good choices for initial scale of the input-to-hidden weights depended a lot on the particular characteristics of the particular task (such as its dimensionality or the input variance)." This means the initialization recipe is not a fixed procedure but a set of guidelines that require problem-specific tuning.
The consequence. The paper's claim that its approach solves the RNN training problems "robustly" (Section 4.2) is qualified by the fact that the initialization was tuned to each task's characteristics. A practitioner facing a new sequence problem β particularly one with unknown properties (e.g., an unfamiliar distractor structure, unknown relevant temporal scales, real-valued rather than synthetic inputs) β cannot simply apply the recipe from Table 4 and expect success. The distinction between "add/mul" connections (scale 0.001) and "mem" connections (scale 0.1) on the addition and multiplication problems requires knowing in advance which inputs are which, information that may not be available in real-world tasks. The requirement for input/output centering on the multiplication problem (but not on the others) further demonstrates fragility: Martens & Sutskever (2011) solved the same problems without centering using HF, meaning the first-order approach depends on preprocessing that the second-order approach did not require. The paper's statement that the HF results "achieved lower error rates and their initialization was chosen with less care" (Section 4.2) is an implicit acknowledgment that the current approach requires more careful initialization engineering to compensate for the loss of second-order curvature information.
What evidence exists in the paper. The evidence is in the level of detail in Section 4.1 and Table 4. The separate scales for different input types within the same task (add vs. mem), the task-dependent choice of 0.001 vs. 0.1, the problem-specific need for centering, and the spectral radius of 1.1 (which "worked well on all tasks" but was presumably found through experimentation rather than derived from task properties) all indicate substantial task-specific engineering. The paper does not ablate these choices individually β there is no experiment showing what happens if the spectral radius is 1.0 or 1.2, or if the input scale is 0.01 instead of 0.001. The reader cannot distinguish which aspects of the initialization are essential and which are incidental. The four RNN tasks, while challenging, are all from the same family of synthetic long-term dependency problems and share structural properties (scalar or low-dimensional outputs, known temporal structure, binary or synthetic inputs) that may not reflect the diversity of real sequence modeling tasks.
Mitigation status. The paper does not frame this as a limitation or propose methods for automating the initialization choices. The guidelines in Table 4 are presented as a recipe, but the text reveals they are the product of empirical tuning. The paper acknowledges the HF approach "used a specialized update damping technique whose benefits seemed mostly limited to training RNNs to solve these kinds of extreme temporal dependency problems" (Section 5), implying that the current approach may also have limited scope, but does not discuss the initialization tuning burden directly. The lack of ablation makes it impossible for a practitioner to prioritize which initialization parameters are most critical to tune on a new task.
6.5 The Comparison Against Hessian-Free Optimization Confounds Algorithmic Differences with Regularization and Hyperparameter Tuning
The assumption or constraint. The paper's central comparative claim β that NAG with proper initialization and momentum matches or exceeds Hessian-Free optimization β rests on comparisons that confound the optimization algorithm with other experimental choices. The published HF results (HF* in Table 1, from Martens (2010)) used L2 regularization (weight decay) during training, while the NAG experiments use no regularization. The paper acknowledges this: "the previously published results on HF used L2 regularization, so they cannot be directly compared" (Section 3). To address this, the paper runs a modified version of HF without L2 regularization (HFβ ) that also incorporates the paper's own momentum-inspired modifications β making it a different algorithm from the published HF*. Meanwhile, the NAG results are obtained with an extensive hyperparameter sweep (60+ configurations per problem to select ΞΌ_max and Ξ΅ jointly), while it is unclear whether the HF* and HFβ results benefited from comparable hyperparameter optimization. The comparison therefore pits an aggressively optimized NAG against either a regularized HF (HF*) or a modified and potentially suboptimally tuned HF (HFβ ).
The consequence. The paper cannot cleanly attribute the performance differences to the optimization algorithm alone. When NAG (0.074 on Curves) outperforms HF* (0.11), it is unclear whether the advantage comes from NAG being a better optimizer or from the absence of L2 regularization (which biases the solution away from the training-error-minimizing parameters). When HFβ (0.058 on Curves) outperforms NAG (0.074), it is unclear whether this represents a genuine advantage of second-order curvature information, an artifact of the momentum-inspired modifications to HF, or simply the result of a particularly favorable hyperparameter configuration for HFβ . The fair comparison would be: NAG vs. HF, both without regularization, both with comparable hyperparameter optimization effort, on the same architecture, measuring the same metric. This comparison does not exist in the paper. The HFβ results are reported without the extensive methodological detail provided for the NAG experiments (how was the damping parameter chosen? How many CG iterations? Was there a comparable hyperparameter sweep?), making it impossible to assess whether HFβ was given the same degree of optimization as NAG.
What evidence exists in the paper. Table 1 contains the three-way comparison (HF*, HFβ , NAG/CM). The methodological asymmetry is visible in the paper's description: Section 3 provides detailed justification for the momentum schedule, the learning rate grid, and the fine-tuning phase for NAG; Appendix A.6 describes the HFβ modifications briefly (the main text only says "see sec. A.6 of the appendix"). The learning rate for NAG was selected from a 6-value grid per ΞΌ_max configuration; the comparable selection procedure for HFβ 's meta-parameters is not described. The paper notes the HF* regularization issue explicitly but does not discuss whether HF* would benefit from the same hyperparameter tuning effort applied to NAG.
Mitigation status. The paper acknowledges the regularization confound but does not resolve it. The HFβ variant partially addresses the issue (by removing regularization and adding momentum-like modifications), but it introduces a new confound (algorithm modification) and does not include the hyperparameter tuning transparency provided for NAG. The paper presents the results as evidence that "the boundary between first-order and second-order methods is blurrier than commonly assumed" (Section 5), which is a conceptual claim that the confounded comparison can support, but the stronger claim that NAG "can train both DNNs and RNNs to levels of performance that were previously achievable only with Hessian-Free optimization" (abstract) requires a cleaner comparison than is provided. A reader cannot determine from this paper whether, given equal hyperparameter tuning effort, HF would regain a clear advantage over NAG on these problems.
6.6 The Momentum Schedule Is Fixed and Coarse, Providing No Mechanism for Per-Dimension or Adaptive Adjustment
The assumption or constraint. The momentum schedule (Equation 5) applies the same scalar ΞΌ_t to all parameters at each iteration. The schedule is purely a function of the iteration count t and the ceiling ΞΌ_max β it does not adapt to the observed behavior of the optimization (e.g., gradient variance, loss reduction rate, parameter-level curvature estimates). Theorem 2.1 shows that NAG automatically applies lower effective momentum in high-curvature directions through the mechanism ΞΌ_eff = ΞΌ(1 β λΡ), which provides some per-dimension adaptation, but this adaptation is fixed by the local curvature Ξ» at each step, not by any learned or scheduled mechanism. Moreover, the schedule itself is coarse: ΞΌ changes only every 250 iterations (when βt/250β increments), producing a stair-step function that jumps at discrete intervals rather than evolving smoothly. For the RNN experiments, the schedule is even coarser: ΞΌ = 0.9 for 1000 iterations, then constant at ΞΌ_0 thereafter β a single step change.
The consequence. The fixed schedule cannot respond to differences between problems, between phases of training on a single problem, or between parameters within a network. A problem that requires sustained high momentum for 500,000 iterations will be suboptimally served by the same schedule as a problem that benefits from high momentum for only 200,000 iterations. Within a single training run, some parameters (e.g., those in early layers vs. late layers, or weights vs. biases) may benefit from different momentum strengths at different times β the scalar schedule provides no mechanism for this. The discrete stair-step nature means that momentum is constant for 249 iterations at a time, potentially causing the optimizer to oscillate at the start of each new ΞΌ plateau (as momentum abruptly increases) or to make insufficient progress at the end (as the optimizer saturates the benefit of the current ΞΌ before the next increase). The paper's finding that reducing ΞΌ at the very end of training (final 1000 iterations) provides substantial improvements (Table 2) suggests that the fixed schedule's transition from high ΞΌ to lower ΞΌ for fine convergence is mistimed β if the schedule naturally reduced ΞΌ at the right moment, the manual reduction would not provide additional benefit. The fact that it does implies the schedule is suboptimal.
What evidence exists in the paper. The evidence is in the schedule design itself and in the existence of Table 2. The ΞΌ_t formula (Equation 5) has several arbitrary constants: the base period of 250 iterations, the specific functional form 2^{-1 - logβ(βt/250β + 1)}, and the use of a hard ceiling ΞΌ_max rather than a smooth asymptotic approach. The paper motivates the schedule theoretically (blending Nesterov's schedules for strongly convex and general convex functions) but does not ablate the specific constants or the functional form. The fact that the optimal ΞΌ_max varies across problems (0.999 for Curves and Faces, 0.99 for MNIST) confirms that the schedule is not universally optimal, but the paper provides no method for determining the right ΞΌ_max without a full sweep. The RNN schedule is even less principled: ΞΌ = 0.9 for 1000 iterations is a heuristic with no theoretical motivation provided.
Mitigation status. The paper does not discuss adaptive momentum methods or per-parameter momentum schedules. At the time of publication (2013), adaptive methods like AdaGrad (Duchi et al., 2011) and RMSProp (Tieleman & Hinton, 2012) existed but were not considered. The paper's contribution is in demonstrating that a well-chosen fixed schedule works, not in proposing an adaptive mechanism. However, the paper does not acknowledge the fixed schedule as a limitation or suggest that adaptive momentum (where ΞΌ_t is adjusted based on optimization progress) could provide further improvements. The discrete stair-step nature is presented without justification, and no comparison to a smooth schedule (e.g., ΞΌ_t = ΞΌ_max Β· (1 β exp(βt/Ο)) or a cosine schedule) is provided. This is a significant omission given that the paper's core argument is that momentum scheduling matters β if scheduling matters, then the specific form of the schedule matters, and the paper has not demonstrated that its chosen form is better than alternatives.
7. Implications and Future Directions
How This Work Changes the Landscape
This paper's primary effect is to reframe the deep network optimization problem from a question of algorithm choice to a question of condition satisfaction. Before this work, the field's mental model was approximately: "first-order methods cannot train deep networks β therefore we need pre-training or second-order methods." After this work, the mental model becomes: "first-order methods CAN train deep networks, but only when TWO conditions are simultaneously met β the initialization places the network in a trainable region of parameter space, AND the momentum is configured to accelerate efficiently through the transient phase." This is more than an incremental improvement on prior recipes; it is a diagnostic framework that explains why previous attempts failed and provides a checklist for making them succeed.
The magnitude of this shift is best understood by what it resolves. The paper reconciles the apparently contradictory bodies of evidence that had accumulated by 2013: the success of Hessian-Free optimization from random initializations (Martens, 2010; Martens & Sutskever, 2011) versus the partial but incomplete success of well-initialized SGD (Glorot & Bengio, 2010; Chapelle & Erhan, 2011). The resolution is that both camps were partially right. The first-order proponents correctly identified that initialization was important but missed that momentum scheduling was the missing piece needed to close the remaining gap. The second-order proponents correctly identified that something beyond plain SGD was needed, but attributed the gap to curvature information when it was actually addressable by properly configured momentum. The paper thus converts a confusing set of results into a coherent picture: the field's prior negative conclusions about momentum (Orr, 1996; LeCun et al., 1998) were artifacts of testing momentum on poorly initialized networks where no optimizer could succeed, combined with schedules (or lack thereof) that provided inadequate acceleration during the transient phase.
The paper also reframes the relationship between first-order and second-order methods in a way that has implications for optimizer design research. By showing that HF can be viewed as a momentum method with CG-based initialization persistence (Section 5), and by demonstrating that making HF more momentum-like improves it (HFβ vs. HF* in Table 1), the paper suggests that momentum-like persistent information propagation is the core mechanism shared across both classes. This shifts the research question from "how do we get better curvature estimates?" toward "how do we more effectively persist and propagate gradient information across iterations?" β a framing that would influence the development of adaptive methods like Adam and RMSProp in subsequent years. The curvature reweighting that second-order methods provide is real but modest in its benefit (the 5-25% gap between NAG and HFβ in Table 1), suggesting that the field had overinvested in curvature estimation relative to momentum scheduling.
The paper's most important methodological contribution may be the reframing of what phase of optimization matters. The argument that the transient phase, not asymptotic convergence, dominates computation time in deep network training (building on Darken & Moody, 1993 but applying it quantitatively to deep architectures) provides a criterion for evaluating optimizers that was not widely used at the time. Before this paper, optimizers were typically compared by asymptotic convergence rates on convex test functions or by final test error after fixed training. The paper argues that the relevant metric is transient-phase efficiency β how quickly an optimizer traverses from a random initialization to a good basin β and that this is what high momentum provides. This focus on the transient phase is validated by the sensitivity to ΞΌ_max (Table 1: error varies by 6.5Γ from ΞΌ=0 to ΞΌ=0.999 on Curves) and by the existence of a distinct fine-tuning phase requiring reduced ΞΌ (Table 2). A research program that only compares final errors or asymptotic rates would miss the dynamics that actually determine success or failure.
One perhaps underappreciated implication is that the paper implicitly argues against the necessity of learning rate decay in deep network training. The experiments use a constant learning rate throughout (swept per configuration but fixed per run), relying on the momentum schedule and the final ΞΌ reduction to control the effective step size. This is a departure from the common practice (then and now) of decaying the learning rate over time. The paper does not make this argument explicitly, but the results suggest that momentum scheduling can substitute for some of the functions that learning rate decay is typically asked to perform β particularly the transition from large-scale exploration to fine convergence. This is a testable hypothesis that the paper enables but does not itself investigate.
In terms of which research directions become more or less attractive: the paper makes curvature-estimation-focused optimizer research less urgent (if NAG can match HF on these benchmarks, curvature reweighting provides diminishing returns) and makes momentum scheduling and initialization design more central. It also makes the study of the transient phase β its duration, its curvature structure, its dependence on architecture β a first-class research topic rather than an inconvenience to be bypassed with pre-training.
Follow-Up Research This Work Enables
1. Direct measurement of transient phase duration across architectures and tasks. The paper argues that the transient phase dominates deep network training, but provides only indirect evidence (sensitivity to ΞΌ_max, benefit of final ΞΌ reduction). A direct experiment would instrument the optimization trajectory to measure when the optimizer transitions from transient (persistent gradient structure, L/T-dominated convergence) to asymptotic (noise-dominated, Ο/βT-limited) behavior. One approach: compute the auto-correlation of gradient directions at multiple timescales throughout training, identifying the point where the gradient becomes effectively uncorrelated noise. This would allow measuring the transient phase duration for different architectures (feed-forward vs. convolutional vs. recurrent), depths, nonlinearities (sigmoid vs. ReLU), and tasks. If the transient phase is shorter for ReLU networks (which have less severe vanishing gradient problems), the benefit of high momentum would be reduced β directly testing the generalizability of the paper's findings beyond sigmoidal architectures. The paper's framework predicts that architectures with longer transient phases benefit more from high momentum; measuring the transient phase directly would test this prediction.
2. Learned per-parameter momentum schedules via hypergradient descent. The paper demonstrates that the momentum schedule matters enormously, but the proposed schedule (Equation 5) is a fixed, scalar function of the iteration count with hand-tuned constants (the 250-iteration period, the specific stair-step form). A natural extension is to treat the momentum coefficient as a learnable parameter β or more ambitiously, as a learnable per-parameter function β optimized by hypergradient descent (differentiating through the optimization process). This was computationally prohibitive in 2013 but is routine with modern automatic differentiation frameworks. A concrete experiment: on the Curves autoencoder, initialize ΞΌ as a per-layer or per-parameter vector, optimize it jointly with the model parameters using a validation set (or using the training loss with a short-horizon meta-objective), and compare the resulting schedule against the hand-designed one. The paper's Theorem 2.1 suggests that different eigendirections of the Hessian should optimally receive different effective momentum (ΞΌ_eff = ΞΌ(1 β λΡ)), which a per-parameter schedule could approximate without explicit curvature computation. If learned schedules substantially outperform the hand-designed schedule, it would validate the paper's claim that momentum scheduling is critical while showing that the specific formula is not optimal β a useful refinement. If learned schedules fail to beat the hand-designed version, it would suggest that the specific increasing pattern matters more than per-parameter adaptation, an important negative result.
3. Systematic comparison of NAG against adaptive methods (AdaGrad, RMSProp, Adam) on the same benchmarks. The paper compares NAG against CM and HF, but adaptive per-parameter learning rate methods (AdaGrad by Duchi et al., 2011; RMSProp by Tieleman & Hinton, 2012) existed at the time and were not tested. These methods address ill-conditioning through per-dimension learning rate scaling (dividing gradients by the square root of accumulated squared gradients), which is a different mechanism than momentum for handling curvature variation. A natural question: does NAG with a good schedule outperform RMSProp on the deep autoencoder and RNN benchmarks, or do adaptive learning rates provide similar or greater benefit? The paper's framework suggests that per-dimension learning rate adaptation (addressing curvature variation across parameters) and momentum (addressing persistent directions along low-curvature valleys) are complementary β adaptive methods might handle the high-curvature oscillation problem that NAG's lookahead addresses, while momentum handles the low-curvature acceleration that adaptive methods struggle with alone. A direct comparison on the Curves and MNIST autoencoders, with both NAG and RMSProp/Adam given comparable hyperparameter tuning budgets, would clarify whether one mechanism dominates or whether combining them (as Adam eventually did) is necessary for optimal performance. The RNN benchmarks from Table 5 are particularly interesting here because they stress the temporal credit assignment problem β do adaptive methods help with exploding/vanishing gradients in RNNs as much as strong momentum does?
4. Stress-testing the ESN-inspired RNN initialization on real-world sequence tasks. The paper's RNN results are confined to four synthetic long-term dependency problems with known structure. A necessary follow-up is to test the same initialization + NAG recipe on the character-level language modeling tasks where Sutskever et al. (2011) and Mikolov et al. (2012) demonstrated HF's effectiveness. Specifically: train a standard RNN with 100-500 tanh hidden units on the Penn Treebank or text8 datasets for character prediction, using the ESN-inspired initialization (spectral radius 1.1, 15 fan-in, task-tuned input scales) with NAG at ΞΌ = 0.995, and compare against the published HF results. The paper's claim that its approach is "sufficient for most practical purposes" (Section 4.2) is untested without such experiments. The central question is whether the initialization tuning burden observed on synthetic tasks (separate input scales for operand vs. marker connections, task-dependent choice of 0.001 vs. 0.1, centering requirements) becomes prohibitive on real tasks where the input structure is less cleanly separable. If the recipe transfers with minimal tuning β e.g., using a single input scale sweep rather than per-input-type tuning β it would substantially strengthen the paper's practical claims. If it requires extensive per-task engineering comparable to the synthetic experiments, it would establish a boundary on the method's generality and suggest that the initialization sensitivity is a fundamental limitation rather than an artifact of the synthetic problems.
5. Ablation of the specific momentum schedule form against simpler alternatives. The paper's schedule (Equation 5) has a specific functional form motivated by Nesterov's theoretical work, but the paper never ablates it against simpler schedules β constant ΞΌ, linear increase from 0.5 to ΞΌ_max, cosine annealing, or the classic Nesterov schedule ΞΌ_t = 1 β 3/(t + 5). A clean ablation on the Curves autoencoder would fix ΞΌ_max (say 0.999) and the learning rate, and vary only the schedule shape, measuring final training error after 750,000 iterations with the final ΞΌ reduction applied identically across all variants. If the proposed schedule significantly outperforms simpler alternatives, it validates the theoretical motivation and demonstrates that schedule shape matters independently of the final ΞΌ value. If a linear or cosine schedule matches or exceeds the proposed stair-step, it would simplify the practical recipe and suggest that the specific form is less important than the general principle of increasing ΞΌ over time β an important qualification to the paper's claims. This is a simple experiment (a few training runs per schedule shape) with high information value for practitioners trying to adopt the method.
6. Investigation of the correct-to-incorrect transition in the optimization analogue β when does reducing momentum too early cause irrecoverable performance loss? The paper warns against reducing ΞΌ too early but provides only anecdotal evidence ("in our experiments we found that doing this was detrimental"). A systematic experiment would run the Curves autoencoder with the optimal NAG configuration (ΞΌ_max = 0.999) but vary the iteration at which ΞΌ is reduced to 0.9: at 100K, 200K, 300K, ..., 750K iterations, measuring final training error. This would map out the irrecoverability curve β the point at which early ΞΌ reduction permanently sacrifices performance vs. merely slowing convergence. If the curve shows a sharp threshold (e.g., reducing ΞΌ before 500K iterations causes 2Γ worse final error, while reducing after 500K has no penalty), it would provide direct evidence for the paper's two-phase model and give practitioners a concrete criterion for when to transition. If the curve is smooth with no threshold, it would suggest that the transient/asymptotic distinction is less crisp than the paper claims and that the benefit of high ΞΌ accrues continuously rather than in a distinct phase. This experiment is straightforward to run and would test one of the paper's central conceptual claims.
Practical Applications and Downstream Use Cases
1. Training deep fully-connected networks without pre-training. The most direct application is replacing greedy layerwise pre-training pipelines with end-to-end training from random initialization. On the Curves autoencoder, the best NAG configuration achieves training error 0.074 vs. 0.16 for the prior best first-order result (Chapelle & Erhan, 2011) β a 2.2Γ improvement β using a simpler pipeline (no per-layer auxiliary objectives, no multi-phase training, just a single optimization run). For practitioners building deep autoencoders or similar fully-connected architectures for dimensionality reduction, feature learning, or unsupervised pre-training of downstream tasks, this represents a substantial reduction in engineering complexity. The recipe is concrete: sparse initialization (15 fan-in per unit), NAG with the schedule of Equation 5 and ΞΌ_max = 0.99β0.999 (selected by a small grid search), constant learning rate, and a final 1% of training at reduced ΞΌ = 0.9. No pre-training stages, no layerwise training, no specialized second-order optimizers.
2. Enabling standard RNNs on tasks with moderate long-range dependencies. Before this work, training a standard RNN (without LSTM gating or HF optimization) on tasks with temporal dependencies spanning more than ~10-20 steps was considered essentially impossible. The paper's results show that the ESN-inspired initialization + NAG at ΞΌ = 0.995 can train a vanilla 100-unit RNN to near-zero error on the addition problem with T=80 (error 0.00025, Table 5) and on mem-20 with T=80 (error 0.0144). This opens the possibility of using simpler RNN architectures for sequence tasks where LSTMs would previously have been considered mandatory, reducing model complexity and the associated hyperparameter burden (forget gate bias, etc.). The practical benefit is most relevant in resource-constrained settings (embedded systems, on-device processing) where the additional parameters and gating computations of an LSTM are costly. The caveat is the initialization sensitivity: the recipe requires tuning the spectral radius (1.1 is a good default) and input weight scales per task, which may limit plug-and-play adoption.
3. A principled hyperparameter selection strategy for momentum methods β sweep ΞΌ_max, not ΞΌ_t. The paper's momentum schedule reduces the momentum hyperparameter from a full schedule (potentially many parameters) to a single scalar ΞΌ_max, with the schedule shape fixed by Equation 5. This dramatically reduces the hyperparameter search space: a practitioner can sweep ΞΌ_max over {0.9, 0.99, 0.995, 0.999} rather than designing a custom schedule. Combined with the paper's recommendation on the learning rate grid ({0.05 to 0.0001} for feed-forward, {10β»Β³ to 10β»βΆ} for recurrent) and the final ΞΌ reduction to 0.9, this provides a concrete, replicable protocol for configuring momentum on a new deep learning problem. The 4-6Γ improvement in training error from the worst to best ΞΌ_max (Table 1) means that even a coarse sweep over 4-5 values captures most of the available gain, without requiring the 60-configuration full grid search the paper used. This is an immediately actionable workflow for practitioners: (a) use sparse initialization (or a modern equivalent like He initialization), (b) fix the schedule shape from Equation 5, (c) sweep ΞΌ_max and Ξ΅ on a validation set, (d) train with the best pair, (e) reduce ΞΌ to 0.9 for the final ~1% of training.
4. Accelerating hyperparameter search for Hessian-Free optimization via momentum-inspired modifications. The paper's finding that making HF more momentum-like improved results (HFβ vs. HF*, Table 1: 0.058 vs. 0.11 on Curves, a 47% error reduction) provides a concrete direction for practitioners who use or maintain HF-based training pipelines. The modification β strengthening the CG initialization persistence to behave more like NAG's velocity accumulation β is described in Appendix A.6 and represents a relatively small change to an existing HF implementation. For applications where HF was already deployed (speech recognition, language modeling as in Sutskever et al., 2011), this suggests a low-cost path to improved performance without changing the overall optimization framework. More broadly, the insight that CG initializations serve a momentum-like function suggests that any truncated-Newton method can benefit from explicitly optimizing this initialization strategy rather than treating it as an implementation detail.