Nested Learning: The Illusion of Deep Learning Architectures โ€” part 14/14

Cleaned-text code points [196798, 211657). Bibliography omitted here; source.md includes it.

Revisiting the momentum term without using the chain rule in backpropagation, W โ„“ t + 1 = W โ„“ t + ๐’Ž โ„“ t + 1 \displaystyle{W_{\ell}}_{{}_{t+1}}={W_{\ell}}_{{}_{t}}+{\bm{m}_{\ell}}_{{}_{t+1}} ๐’Ž โ„“ t + 1 = ฮฑ โ„“ , t + 1 ๐’Ž โ„“ t โˆ’ ฮท โ„“ , t + 1 โˆ‡ W โ„“ t โ„’ ( W โ„“ t ; ๐’™ t + 1 ) , \displaystyle{\bm{m}_{\ell}}_{{}_{t+1}}=\alpha_{\ell,t+1}{\bm{m}_{\ell}}_{{}_{t}}-\eta_{\ell,t+1}\nabla_{{W_{\ell}}_{{}_{t}}}\mathcal{L}\left({W_{\ell}}_{{}_{t}};\bm{x}_{t+1}\right), (100) one can interpret the momentum as a key or value-less associative memory ( Behrouz et al., 2025b , see Section 5) , where the gradient terms โˆ‡ W โ„“ t โ„’ ( W โ„“ t ; ๐’™ t + 1 ) \nabla_{{W_{\ell}}_{{}_{t}}}\mathcal{L}\left({W_{\ell}}_{{}_{t}};\bm{x}_{t+1}\right) are compressed into the momentum. We expect from a powerful momentum term to perfectly memorize all the past gradients in the training process so it can better model the current update to the weights. To this end, one can start from defining a simple objective as: โ„’ ~ t = โˆ‘ i = 1 t โˆฅ ๐’Ž โ„“ t โŠ™ ๐’ˆ โ„“ i + 1 โˆ’ ๐‘ท โ„“ t โˆฅ 2 2 + ฮป โ„“ โˆฅ ๐’Ž โ„“ t โˆฅ 2 F , \displaystyle\tilde{\mathcal{L}}_{t}=\sum_{i=1}^{t}\|{\bm{m}_{\ell}}_{{}_{t}}\odot{\bm{g}_{\ell}}_{{}_{i+1}}-\bm{P}_{\ell_{t}}\|_{2}^{2}\>\>+\>\>\lambda_{\ell}\|{\bm{m}_{\ell}}_{{}_{t}}\|^{2}_{F}, (101) where ๐’ˆ โ„“ t + 1 = โˆ’ โˆ‡ W โ„“ t โ„’ ( W โ„“ t ; ๐’™ t + 1 ) {\bm{g}_{\ell}}_{{}_{t+1}}=-\nabla_{{W_{\ell}}_{{}_{t}}}\mathcal{L}\left({W_{\ell}}_{{}_{t}};\bm{x}_{t+1}\right) . This objective aims to find a momentum term that does not simply map the gradients to 1 (which also results in limited memory management in momentum), but it maps gradients to a global property of past data samples. The more expressive this global property is, the more accurately the momentum can incorporate the compressed information from past. Accordingly, the objective in Equation 101 admits an optimal associative memory that maps gradients to ๐‘ท โ„“ t + 1 \bm{P}_{\ell_{t+1}} as: ๐’Ž โ„“ , i ( t ) โˆ— = [ ( ๐‘ฏ โ„“ , i ( t ) + ฮป โ„“ โ€‹ ๐‘ฐ ) โˆ’ 1 ] โŠ™ ( ๐‘ด โ„“ , i ( t ) ) = [ ( ๐‘ฏ โ„“ , i ( t ) + ฮป โ„“ โ€‹ ๐‘ฐ ) โˆ’ 1 ] โŠ™ ๐‘ด ~ โ„“ , i + 1 ( t ) โŠ™ ๐‘ท โ„“ t , where \displaystyle{\bm{m}^{(t)}_{\ell,i}}^{*}=\left[\left(\bm{H}^{(t)}_{\ell,i}\>+\>\lambda_{\ell}\>\bm{I}\right)^{-1}\right]\odot\left(\bm{M}^{(t)}_{\ell,i}\right)=\left[\left(\bm{H}^{(t)}_{\ell,i}\>+\>\lambda_{\ell}\>\bm{I}\right)^{-1}\right]\odot\tilde{\bm{M}}^{(t)}_{\ell,i+1}\odot\bm{P}_{\ell_{t}},\qquad\text{where} ๐‘ด ( t ) โ„“ , i + 1 = ๐‘ด ( t ) โ„“ , i + ฮฒ 1 ๐’ˆ โ„“ i + 1 โŠ™ ๐‘ท โ„“ t = ๐‘ด ~ ( t ) โ„“ , i + 1 โŠ™ ๐‘ท โ„“ t , \displaystyle\bm{M}^{(t)}_{\ell,i+1}=\bm{M}^{(t)}_{\ell,i}\>+\>\beta_{1}\>{\bm{g}_{\ell}}_{{}_{i+1}}\odot\bm{P}_{\ell_{t}}=\tilde{\bm{M}}^{(t)}_{\ell,i+1}\odot\bm{P}_{\ell_{t}}, ๐‘ด ~ ( t ) โ„“ , i + 1 = ๐‘ด ( t ) โ„“ , i + ฮฒ 1 ๐’ˆ โ„“ i + 1 , \displaystyle\tilde{\bm{M}}^{(t)}_{\ell,i+1}=\bm{M}^{(t)}_{\ell,i}\>+\>\beta_{1}\>{\bm{g}_{\ell}}_{{}_{i+1}}, ๐‘ฏ ( t ) โ„“ , i + 1 = ๐‘ฏ ( t ) โ„“ , i + ฮฒ 2 ๐’ˆ โ„“ i + 1 โŠ™ ๐’ˆ โ„“ i + 1 = ๐‘ฏ ( t ) โ„“ , i + ฮฒ 2 ๐’ˆ โ„“ i + 1 2 . \displaystyle\bm{H}^{(t)}_{\ell,i+1}=\bm{H}^{(t)}_{\ell,i}+\beta_{2}\>{\bm{g}_{\ell}}_{{}_{i+1}}\odot{\bm{g}_{\ell}}_{{}_{i+1}}=\bm{H}^{(t)}_{\ell,i}+\beta_{2}\>{\bm{g}_{\ell}}_{{}_{i+1}}^{2}. (102) Given this solution, one can write the update step as: W โ„“ i + 1 = W โ„“ i โˆ’ ฮท t ๐’Ž โ„“ , i ( t ) โˆ— = W โ„“ i โˆ’ ฮท i [ ( ๐‘ฏ โ„“ , i ( t ) + ฮป โ„“ ๐‘ฐ ) โˆ’ 1 ] โŠ™ ๐‘ด ~ โ„“ , i + 1 ( t ) โŠ™ ๐‘ท โ„“ t . \displaystyle{W_{\ell}}_{{}_{i+1}}={W_{\ell}}_{{}_{i}}-\eta_{t}\>{\bm{m}^{(t)}_{\ell,i}}^{*}={W_{\ell}}_{{}_{i}}-\eta_{i}\>\left[\left(\bm{H}^{(t)}_{\ell,i}\>+\>\lambda_{\ell}\>\bm{I}\right)^{-1}\right]\odot\tilde{\bm{M}}^{(t)}_{\ell,i+1}\odot\bm{P}_{\ell_{t}}. (103) We start with a simple case, where ๐‘ท โ„“ t \bm{P}_{\ell_{t}} is the summation of the square of past gradients: i.e., ๐‘ท โ„“ t = โˆ‘ i = 1 t ๐’ˆ โ„“ i + 1 2 \bm{P}_{\ell_{t}}=\sum_{i=1}^{t}\bm{g}_{\ell_{i+1}}^{2} . With the choice of ฮป โ†’ 0 \lambda\rightarrow 0 , Equation 103 recovers simple gradient descent with momentum. That is, let ฮป = 0 \lambda=0 , we have: W โ„“ i + 1 = W โ„“ i โˆ’ ฮท t ๐’Ž โ„“ , i ( t ) โˆ— = W โ„“ i โˆ’ ฮท i ( ๐‘ฏ โ„“ , i ( t ) ) โˆ’ 1 โŠ™ ๐‘ด ~ โ„“ , i + 1 ( t ) โŠ™ ๐‘ท โ„“ t โŸ ๐‘ฏ โ„“ , i ( t ) / ฮฒ 2 = W โ„“ i โˆ’ ฮท t ฮฒ 2 ๐‘ด ~ โ„“ , i + 1 ( t ) , \displaystyle{W_{\ell}}_{{}_{i+1}}={W_{\ell}}_{{}_{i}}-\eta_{t}\>{\bm{m}^{(t)}_{\ell,i}}^{*}={W_{\ell}}_{{}_{i}}-\eta_{i}\>\left(\bm{H}^{(t)}_{\ell,i}\right)^{-1}\odot\tilde{\bm{M}}^{(t)}_{\ell,i+1}\odot\underset{\bm{H}^{(t)}_{\ell,i}/\beta_{2}}{\underbrace{\bm{P}_{\ell_{t}}}}={W_{\ell}}_{{}_{i}}-\eta_{t}\beta_{2}\>\tilde{\bm{M}}^{(t)}_{\ell,i+1}, (104) which based on the definition of ๐‘ด ~ โ„“ , i + 1 ( t ) \tilde{\bm{M}}^{(t)}_{\ell,i+1} , this update rule is equivalent to gradient descent with momentum. Next, we explore a more sophisticated design choice, where we use ๐‘ท โ„“ t \bm{P}_{\ell_{t}} as the variance of data samples before token t + 1 t+1 . In this case, formally, ๐‘ท โ„“ t = โˆ‘ i = 1 t ๐’ˆ โ„“ i + 1 2 \bm{P}_{\ell_{t}}=\sqrt{\sum_{i=1}^{t}\bm{g}_{\ell_{i+1}}^{2}} and so the update rule is as follows: W โ„“ i + 1 = W โ„“ i โˆ’ ฮท t ๐’Ž โ„“ , i ( t ) โˆ— = W โ„“ i โˆ’ ฮท i [ ( ๐‘ฏ โ„“ , i ( t ) + ฮป โ„“ ๐‘ฐ ) โˆ’ 1 ] โŠ™ ๐‘ด ~ โ„“ , i + 1 ( t ) โŠ™ ๐‘ท โ„“ t โŸ ๐‘ฏ โ„“ , i ( t ) 1 2 / ฮฒ 2 โ‰ˆ W โ„“ i โˆ’ ฮท t ฮฒ 2 ๐‘ด ~ โ„“ , i ( t ) ๐‘ฏ โ„“ , i ( t ) 1 / 2 + ฮต \displaystyle{W_{\ell}}_{{}_{i+1}}={W_{\ell}}_{{}_{i}}-\eta_{t}\>{\bm{m}^{(t)}_{\ell,i}}^{*}={W_{\ell}}_{{}_{i}}-\eta_{i}\>\left[\left(\bm{H}^{(t)}_{\ell,i}\>+\>\lambda_{\ell}\>\bm{I}\right)^{-1}\right]\odot\tilde{\bm{M}}^{(t)}_{\ell,i+1}\odot\underset{\bm{H}^{(t)^{\frac{1}{2}}}_{\ell,i}/\sqrt{\beta_{2}}}{\underbrace{\bm{P}_{\ell_{t}}}}\approx{W_{\ell}}_{{}_{i}}-\frac{\eta_{t}}{\sqrt{\beta_{2}}}\>\frac{\tilde{\bm{M}}^{(t)}_{\ell,i}}{\bm{H}^{(t)^{1/2}}_{\ell,i}\>+\varepsilon} (105) which is equivalent to the popular Adam optimizer ( Kingma and Ba, 2014 ) . Therefore, Adam is an optimal associative memory given the L 2 L_{2} regression objective that is defined in Equation 101 . In fact, the update component of Adam at each state aims to learn a mapping between the gradients and their variance (as a global property of past data samples). One interesting point about this formulation is about the frequency of elements. Looking at the first and second momentum in Adam optimizer, the frequency of update for both of these memories are the same as both are updated after each sample. Furthermore, the computation of each of them can be done in parallel and so are independent. Accordingly, it is one of the few examples that two components without any internal gradient flow, are placed in the same frequency and so level (see Section 3.2 ).

Going beyond element-wise update, we reformulate Equation 101 with outer-product operation as: โ„’ ~ t = โˆ‘ i = 1 t โˆฅ ๐’Ž โ„“ t ๐’ˆ โ„“ i + 1 โˆ’ ๐‘ท โ„“ t โˆฅ 2 2 + ฮป โ„“ โˆฅ ๐’Ž โ„“ t โˆฅ 2 F , \displaystyle\tilde{\mathcal{L}}_{t}=\sum_{i=1}^{t}\|{\bm{m}_{\ell}}_{{}_{t}}{\bm{g}_{\ell}}_{{}_{i+1}}-\bm{P}_{\ell_{t}}\|_{2}^{2}\>\>+\>\>\lambda_{\ell}\|{\bm{m}_{\ell}}_{{}_{t}}\|^{2}_{F}, (106) where ๐’ˆ โ„“ t + 1 = โˆ’ โˆ‡ W โ„“ t โ„’ ( W โ„“ t ; ๐’™ t + 1 ) {\bm{g}_{\ell}}_{{}_{t+1}}=-\nabla_{{W_{\ell}}_{{}_{t}}}\mathcal{L}\left({W_{\ell}}_{{}_{t}};\bm{x}_{t+1}\right) . Similar to Equation 102 , finding the optimal solution to the above objective is defined as: ๐’Ž โ„“ , i ( t ) โˆ— = [ ( ๐‘ฏ โ„“ , i ( t ) + ฮป โ„“ โ€‹ ๐‘ฐ ) โˆ’ 1 ] โ€‹ ( ๐‘ด โ„“ , i ( t ) ) = [ ( ๐‘ฏ โ„“ , i ( t ) + ฮป โ„“ โ€‹ ๐‘ฐ ) โˆ’ 1 ] โ€‹ ๐‘ด ~ โ„“ , i + 1 ( t ) โ€‹ ๐‘ท โ„“ t , where \displaystyle{\bm{m}^{(t)}_{\ell,i}}^{*}=\left[\left(\bm{H}^{(t)}_{\ell,i}\>+\>\lambda_{\ell}\>\bm{I}\right)^{-1}\right]\left(\bm{M}^{(t)}_{\ell,i}\right)=\left[\left(\bm{H}^{(t)}_{\ell,i}\>+\>\lambda_{\ell}\>\bm{I}\right)^{-1}\right]\tilde{\bm{M}}^{(t)}_{\ell,i+1}\bm{P}_{\ell_{t}},\qquad\text{where} (107) ๐‘ด ( t ) โ„“ , i + 1 = ๐‘ด ( t ) โ„“ , i + ฮฒ 1 ๐‘ท โ„“ t ๐’ˆ โ„“ i + 1 โŠค = ๐‘ท โ„“ t ๐‘ด ~ ( t ) โ„“ , i + 1 , \displaystyle\bm{M}^{(t)}_{\ell,i+1}=\bm{M}^{(t)}_{\ell,i}\>+\>\beta_{1}\>\bm{P}_{\ell_{t}}\>{\bm{g}_{\ell}}_{{}_{i+1}}^{\top}=\bm{P}_{\ell_{t}}\>\tilde{\bm{M}}^{(t)}_{\ell,i+1}, (108) ๐‘ด ~ ( t ) โ„“ , i + 1 = ๐‘ด ( t ) โ„“ , i + ฮฒ 1 ๐’ˆ โ„“ i + 1 โŠค , \displaystyle\tilde{\bm{M}}^{(t)}_{\ell,i+1}=\bm{M}^{(t)}_{\ell,i}\>+\>\beta_{1}\>{\bm{g}_{\ell}}_{{}_{i+1}}^{\top}, (109) ๐‘ฏ ( t ) โ„“ , i + 1 = ๐‘ฏ ( t ) โ„“ , i + ฮฒ 2 ๐’ˆ โ„“ i + 1 ๐’ˆ โ„“ i + 1 โŠค . \displaystyle\bm{H}^{(t)}_{\ell,i+1}=\bm{H}^{(t)}_{\ell,i}+\beta_{2}\>{\bm{g}_{\ell}}_{{}_{i+1}}{\bm{g}_{\ell}}_{{}_{i+1}}^{\top}. (110) With a similar choice as Adam, i.e., letting ๐‘ท โ„“ t \bm{P}_{\ell_{t}} be the variance of the gradients so far, ๐‘ท โ„“ t = โˆ‘ i = 1 t ๐’ˆ โ„“ i + 1 โ€‹ ๐’ˆ โ„“ i + 1 โŠค \bm{P}_{\ell_{t}}=\sqrt{\sum_{i=1}^{t}\bm{g}_{\ell_{i+1}}\bm{g}_{\ell_{i+1}}^{\top}} , then the above solution can be simplified as: W โ„“ i + 1 = W โ„“ i โˆ’ ฮท t ๐’Ž โ„“ , i ( t ) โˆ— = W โ„“ i โˆ’ ฮท i [ ( ๐‘ฏ โ„“ , i ( t ) + ฮป โ„“ ๐‘ฐ ) โˆ’ 1 ] ๐‘ท โ„“ t โŸ ๐‘ฏ โ„“ , i ( t ) 1 2 / ฮฒ 2 ๐‘ด ~ โ„“ , i + 1 ( t ) โ‰ˆ W โ„“ i โˆ’ ฮท t ฮฒ 2 ๐‘ฏ โ„“ , i ( t ) โˆ’ 1 / 2 ๐‘ด ~ โ„“ , i ( t ) \displaystyle{W_{\ell}}_{{}_{i+1}}={W_{\ell}}_{{}_{i}}-\eta_{t}\>{\bm{m}^{(t)}_{\ell,i}}^{*}={W_{\ell}}_{{}_{i}}-\eta_{i}\>\left[\left(\bm{H}^{(t)}_{\ell,i}\>+\>\lambda_{\ell}\>\bm{I}\right)^{-1}\right]\>\underset{\bm{H}^{(t)^{\frac{1}{2}}}_{\ell,i}/\sqrt{\beta_{2}}}{\underbrace{\bm{P}_{\ell_{t}}}}\>\tilde{\bm{M}}^{(t)}_{\ell,i+1}\approx{W_{\ell}}_{{}_{i}}-\frac{\eta_{t}}{\sqrt{\beta_{2}}}\>\bm{H}^{(t)^{-1/2}}_{\ell,i}\>\tilde{\bm{M}}^{(t)}_{\ell,i} (111) The above formulation is the AdaGrad with momentum ( Dรฉfossez et al., 2022 ) and so generalizes the AdaGrad ( Duchi et al., 2011 ) , i.e., when ฮฒ 1 = 1 \beta_{1}=1 . Similarly, based on the connection of Adam optimizer ( Kingma and Ba, 2014 ) with other algorithms such as RMSProp ( Hinton et al., 2012 ) , SignSGD and its momentum-based variants ( Bernstein et al., 2018 ) , NAdam ( Dozat, 2016 ) , AMSGrad ( Reddi et al., 2016 ) , RAdam ( Liu et al., 2020 ) , and Lion ( Chen et al., 2023 ) , as well as considering AdaGradโ€™s connection with optimizers such as Shampoo ( Gupta et al., 2018 ) and Soap ( Vyas et al., 2025 ) โ€“i.e., as the approximation of the preconditioning termโ€“we can conclude that all these optimizers can be re-formulated as associative memory that aims to compress the gradients.

## Appendix C Delta Gradient Descent with Normalization

In Section 4.5 we discussed that gradient descent can be see as an associative memory, and be reformulated as: W t + 1 = arg โก min W โ€‹ โŸจ W โ€‹ ๐’™ t , โˆ‡ y t โ„’ โ€‹ ( W t , ๐’™ t ) โŸฉ + 1 2 โ€‹ ฮท t โ€‹ โ€– W โˆ’ W t โ€– 2 2 , \displaystyle W_{t+1}=\arg\min_{W}\>\>\langle W\bm{x}_{t},\nabla_{y_{t}}\mathcal{L}(W_{t};\bm{x}_{t})\rangle+\frac{1}{2\eta_{t}}\>\|W-W_{t}\|^{2}_{2}, (112) where each step aims at learning the negative of the gradient direction. Due to the fact that this learning rule only use update terms that depends on the current gradient, we defined ๐ฎ t = โˆ’ โˆ‡ y t โ„’ โ€‹ ( W t , ๐’™ t ) \mathbf{u}_{t}=-\nabla_{y_{t}}\mathcal{L}(W_{t};\bm{x}_{t}) , and extended the above process to Delta Gradient Descent with more expressive objective of โ„’ 2 \mathcal{L}_{2} regression loss: W t + 1 = arg โก min W โ€‹ 1 2 โ€‹ โ€– W โ€‹ ๐’™ t โˆ’ ๐ฎ t โ€– 2 2 + 1 2 โ€‹ ฮท t โ€‹ โ€– W โˆ’ W t โ€– 2 2 . \displaystyle W_{t+1}=\arg\min_{W}\>\>\frac{1}{2}\|W\bm{x}_{t}-\mathbf{u}_{t}\|^{2}_{2}+\frac{1}{2\eta_{t}}\|W-W_{t}\|^{2}_{2}. (113) We assume that ๐’™ t \bm{x}_{t} is normalized (e.g. in normalized memory systems or in neural networks with normalization layers, โ€– ๐’™ t โ€– 2 = ฮป \|\bm{x}_{t}\|_{2}=\lambda ). Taking gradient to optimize the above objective, 2 โ€‹ ( W t + 1 โ€‹ ๐’™ t โˆ’ โˆ‡ y t โ„’ โ€‹ ( W t , ๐’™ t ) ) โ€‹ ๐’™ t โŠค + 2 โ€‹ ฮท t โ€‹ ( W t + 1 โˆ’ W t ) = 0 , \displaystyle 2(W_{t+1}\bm{x}_{t}-\nabla_{y_{t}}\mathcal{L}(W_{t},\bm{x}_{t}))\bm{x}_{t}^{\top}+2\eta_{t}(W_{t+1}-W_{t})=0, (114) which results in: W t + 1 โ€‹ ( ๐’™ t โ€‹ ๐’™ t โŠค + ฮท t โ€‹ I ) = โˆ‡ y t โ„’ โ€‹ ( W t , ๐’™ t ) โ€‹ ๐’™ t โŠค + ฮท t โ€‹ W t , \displaystyle W_{t+1}(\bm{x}_{t}\bm{x}_{t}^{\top}+\eta_{t}I)=\nabla_{y_{t}}\mathcal{L}(W_{t},\bm{x}_{t})\bm{x}_{t}^{\top}+\eta_{t}W_{t}, (115) โ‡’ \displaystyle\Rightarrow\>\> W t + 1 = ( โˆ‡ y t โ„’ โ€‹ ( W t , x t ) โ€‹ x t โŠค + ฮท t โ€‹ W t ) โ€‹ ( ๐’™ t โ€‹ ๐’™ t โŠค + ฮท t โ€‹ I ) โˆ’ 1 . \displaystyle W_{t+1}=(\nabla_{y_{t}}\mathcal{L}(W_{t},x_{t})x_{t}^{\top}+\eta_{t}W_{t})(\bm{x}_{t}\bm{x}_{t}^{\top}+\eta_{t}I)^{-1}. (116) To compute ( ๐’™ t โ€‹ ๐’™ t โŠค + ฮท t โ€‹ I ) โˆ’ 1 (\bm{x}_{t}\bm{x}_{t}^{\top}+\eta_{t}I)^{-1} term with Sherman-Morrison lemma, we have: ( x t โ€‹ x t โŠค + ฮท t โ€‹ I ) โˆ’ 1 = 1 ฮท t โ€‹ ( I โˆ’ 1 ฮป 2 + ฮท t โ€‹ ๐’™ t โ€‹ ๐’™ t โŠค ) , \displaystyle(x_{t}x_{t}^{\top}+\eta_{t}I)^{-1}=\frac{1}{\eta_{t}}(I-\frac{1}{\lambda^{2}+\eta_{t}}\bm{x}_{t}\bm{x}_{t}^{\top}), (117) and so: W t + 1 = ( โˆ‡ y t โ„’ โ€‹ ( W t , ๐’™ t ) โ€‹ ๐’™ t โŠค + ฮท t โ€‹ W t ) โ€‹ 1 ฮท t โ€‹ ( I โˆ’ 1 ฮป 2 + ฮท t โ€‹ ๐’™ t โ€‹ ๐’™ t โŠค ) \displaystyle W_{t+1}=(\nabla_{y_{t}}\mathcal{L}(W_{t},\bm{x}_{t})\bm{x}_{t}^{\top}+\eta_{t}W_{t})\frac{1}{\eta_{t}}(I-\frac{1}{\lambda^{2}+\eta_{t}}\bm{x}_{t}\bm{x}_{t}^{\top}) (118) โ‡’ \displaystyle\Rightarrow\>\> W t + 1 = W t โ€‹ ( I โˆ’ 1 ฮป 2 + ฮท t โ€‹ ๐’™ t โ€‹ ๐’™ t โŠค ) + 1 ฮท t โ€‹ โˆ‡ y t โ„’ โ€‹ ( W t , ๐’™ t ) โ€‹ ๐’™ t โŠค โˆ’ 1 ฮป 2 โ€‹ ฮท t + ฮท t 2 โ€‹ โˆ‡ y t โ„’ โ€‹ ( W t , ๐’™ t ) โ€‹ ๐’™ t โŠค โ€‹ ๐’™ t โ€‹ ๐’™ t โŠค โŸ ฮป ฮป 2 โ€‹ ฮท t + ฮท t 2 โ€‹ โˆ‡ y t โ„’ โ€‹ ( W t , ๐’™ t ) โ€‹ ๐’™ t โŠค \displaystyle W_{t+1}=W_{t}\left(I-\frac{1}{\lambda^{2}+\eta_{t}}\bm{x}_{t}\bm{x}_{t}^{\top}\right)+\frac{1}{\eta_{t}}\nabla_{y_{t}}\mathcal{L}(W_{t},\bm{x}_{t})\bm{x}_{t}^{\top}-\underset{\frac{\lambda}{\lambda^{2}\eta_{t}+\eta^{2}_{t}}\nabla_{y_{t}}\mathcal{L}(W_{t},\bm{x}_{t})\bm{x}_{t}^{\top}}{\underbrace{\frac{1}{\lambda^{2}\eta_{t}+\eta^{2}_{t}}\nabla_{y_{t}}\mathcal{L}(W_{t},\bm{x}_{t})\bm{x}_{t}^{\top}\bm{x}_{t}\bm{x}_{t}^{\top}}} (119) โ‡’ \displaystyle\Rightarrow\>\> W t + 1 = W t โ€‹ ( I โˆ’ 1 ฮป 2 + ฮท t โ€‹ ๐’™ t โ€‹ ๐’™ t โŠค ) โˆ’ ( ฮป ฮป 2 โ€‹ ฮท t + ฮท t 2 โˆ’ 1 ฮท t ) โ€‹ โˆ‡ y t โ„’ โ€‹ ( W t , ๐’™ t ) โ€‹ ๐’™ t โŠค \displaystyle W_{t+1}=W_{t}\left(I-\frac{1}{\lambda^{2}+\eta_{t}}\bm{x}_{t}\bm{x}_{t}^{\top}\right)-\left(\frac{\lambda}{\lambda^{2}\eta_{t}+\eta^{2}_{t}}-\frac{1}{\eta_{t}}\right)\nabla_{y_{t}}\mathcal{L}(W_{t},\bm{x}_{t})\bm{x}_{t}^{\top} (120) โ‡’ \displaystyle\Rightarrow\>\> W t + 1 = W t โ€‹ ( I โˆ’ ฮฑ t โ€‹ ๐’™ t โ€‹ ๐’™ t โŠค ) โˆ’ ฮฒ โ€‹ โˆ‡ y t โ„’ โ€‹ ( W t , ๐’™ t ) โ€‹ ๐’™ t โŠค \displaystyle W_{t+1}=W_{t}\left(I-\alpha_{t}\bm{x}_{t}\bm{x}_{t}^{\top}\right)-\beta\>\>\nabla_{y_{t}}\mathcal{L}(W_{t},\bm{x}_{t})\bm{x}_{t}^{\top} (121)

## Instructions for reporting errors

We are continuing to improve HTML versions of papers, and your feedback helps enhance accessibility and mobile support. To report errors in the HTML that will help us improve conversion and rendering, choose any of the methods listed below:

Click the "Report Issue" ( ) button, located in the page header.

Tip: You can select the relevant text first, to include it in your report.

Our team has already identified the following issues . We appreciate your time reviewing and reporting rendering errors we may not have found yet. Your efforts will help us improve the HTML versions for all readers, because disability should not be a barrier to accessing research. Thank you for your continued support in championing open access for all.

Have a free development cycle? Help support accessibility at arXiv! Our collaborators at LaTeXML maintain a list of packages that need conversion , and welcome developer contributions .