LocoProp: Enhancing BackProp via Local Loss Optimization
Not on the map. Why?
Abstract (arXiv)
Second-order methods have shown state-of-the-art performance for optimizing deep neural networks. Nonetheless, their large memory requirement and high computational complexity, compared to first-order methods, hinder their versatility in a typical low-budget setup. This paper introduces a general framework of layerwise loss construction for multilayer neural networks that achieves a performance closer to second-order methods while utilizing first-order optimizers only. Our methodology lies upon a three-component loss, target, and regularizer combination, for which altering each component results in a new update rule. We provide examples using squared loss and layerwise Bregman divergences induced by the convex integral functions of various transfer functions. Our experiments on benchmark models and datasets validate the efficacy of our new approach, reducing the gap between first-order and second-order optimizers.
Key facts. Cited by 15 papers in the map (data through 7 Oct 2026). Builds on 23 of its 51 references that are in the map. First posted on arXiv 11 Jun 2021.
Builds on
23 of 51 references.
References · 23
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,283
- 2018Adafactor: Adaptive Learning Rates with Sublinear Memory Costcited by 896
- 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 776
- 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 413
- 2019Limitations of the Empirical Fisher Approximation for Natural Gradient Descentcited by 213
- 2014Difference Target Propagationcited by 144
- 2016Decoupled Neural Interfaces using Synthetic Gradientscited by 135
- 2020Practical Quasi-Newton Methods for Training Deep Neural Networkscited by 67
- 2016Training Neural Networks Without Gradients: A Scalable ADMM Approachcited by 61
- 2019Robust Bi-Tempered Logistic Loss Based on Bregman Divergencescited by 61
- 2018Global Convergence of Block Coordinate Descent in Deep Learningcited by 38
11 more
- 2017Convergent Block Coordinate Descent for Training Tikhonov Regularized Deep Neural Networkscited by 26
- 2018Lifted Neural Networkscited by 19
- 2019Memory-Efficient Adaptive Optimizationcited by 19
- 2018Fenchel Lifted Networks: A Lagrange Relaxation of Neural Network Trainingcited by 18
- 2018A Proximal Block Coordinate Descent Algorithm for Deep Neural Network Trainingcited by 15
- 2018Lifted Proximal Operator Machinescited by 14
- 2019Contrastive Learning for Lifted Networkscited by 9
- 2017Proximal Backpropagationcited by 7
- 2020Guided Learning of Nonconvex Models through Successive Functional Gradient Optimizationcited by 7
- 2019An Implicit Form of Krasulina's k-PCA Update without the Orthonormality Constraintcited by 4
- 2016ParMAC: distributed optimisation of nested functions, with application to learning binary autoencoderscited by 2
Built on by
Citing papers per year: 2021: 2; 2022: 3; 2023: 5; 2024: 1; 2025: 2; 2026: 2.
15 papers.
Sort:
By area:
- 2021Large-Scale Differentially Private BERTcited by 117
- 2022Gradient Descent on Neurons and its Link to Approximate Second-Order Optimizationcited by 29
- 2023On the Parameterization of Second-Order Optimization Effective Towards the Infinite Widthcited by 14
- 2023Block-local learning with probabilistic latent representationscited by 6
- 2023Target-based Surrogates for Stochastic Optimizationcited by 4
- 2024Local Loss Optimization in the Infinite Width: Stable Parameterization of Predictive Coding Networks and Target Propagationcited by 4
- 2021Target Propagation via Regularized Inversioncited by 3
- 2022An alternative approach to train neural networks using monotone variational inequalitycited by 1
- 2022Layerwise Bregman Representation Learning with Applications to Knowledge Distillationcited by 1
- 2023FOSI: Hybrid First and Second Order Optimizationcited by 1
- 2023Data-efficient operator learning for solving high Mach number fluid flow problemscited by 0
- 2025Selective Matching Losses -- Not All Scores Are Created Equalcited by 0
- 2026SOLO: Pretraining Billion-Parameter Language Models with Shared-Output Local Learning28 Sep 2026
- 2026Analog-Friendly Predictive Coding without Activation Derivatives26 Sep 2026
- 2025An Adaptive Method Stabilizing Activations for Enhanced Generalization10 Jun 2025
- 2025Selective Matching Losses -- Not All Scores Are Created Equal4 Jun 2025
- 2024Local Loss Optimization in the Infinite Width: Stable Parameterization of Predictive Coding Networks and Target Propagation4 Nov 2024
- 2023On the Parameterization of Second-Order Optimization Effective Towards the Infinite Width19 Dec 2023
- 2023Data-efficient operator learning for solving high Mach number fluid flow problems28 Nov 2023
- 2023Block-local learning with probabilistic latent representations24 May 2023
- 2023FOSI: Hybrid First and Second Order Optimization16 Feb 2023
- 2023Target-based Surrogates for Stochastic Optimization6 Feb 2023
- 2022Layerwise Bregman Representation Learning with Applications to Knowledge Distillation15 Sep 2022
- 2022An alternative approach to train neural networks using monotone variational inequality17 Feb 2022
3 more
- 2022Gradient Descent on Neurons and its Link to Approximate Second-Order Optimization28 Jan 2022
- 2021Target Propagation via Regularized Inversion2 Dec 2021
- 2021Large-Scale Differentially Private BERT3 Aug 2021
- 2023Target-based Surrogates for Stochastic Optimizationcited by 4
- 2023Target-based Surrogates for Stochastic Optimization6 Feb 2023
- 2021Large-Scale Differentially Private BERTcited by 117
- 2022Gradient Descent on Neurons and its Link to Approximate Second-Order Optimizationcited by 29
- 2023On the Parameterization of Second-Order Optimization Effective Towards the Infinite Widthcited by 14
- 2023Block-local learning with probabilistic latent representationscited by 6
- 2024Local Loss Optimization in the Infinite Width: Stable Parameterization of Predictive Coding Networks and Target Propagationcited by 4
- 2021Target Propagation via Regularized Inversioncited by 3
- 2022An alternative approach to train neural networks using monotone variational inequalitycited by 1
- 2022Layerwise Bregman Representation Learning with Applications to Knowledge Distillationcited by 1
- 2023FOSI: Hybrid First and Second Order Optimizationcited by 1
- 2023Data-efficient operator learning for solving high Mach number fluid flow problemscited by 0
- 2025Selective Matching Losses -- Not All Scores Are Created Equalcited by 0
- 2025An Adaptive Method Stabilizing Activations for Enhanced Generalizationcited by 0
12 most cited of 14 All 14 (JSON)
- 2026SOLO: Pretraining Billion-Parameter Language Models with Shared-Output Local Learning28 Sep 2026
- 2026Analog-Friendly Predictive Coding without Activation Derivatives26 Sep 2026
- 2025An Adaptive Method Stabilizing Activations for Enhanced Generalization10 Jun 2025
- 2025Selective Matching Losses -- Not All Scores Are Created Equal4 Jun 2025
- 2024Local Loss Optimization in the Infinite Width: Stable Parameterization of Predictive Coding Networks and Target Propagation4 Nov 2024
- 2023On the Parameterization of Second-Order Optimization Effective Towards the Infinite Width19 Dec 2023
- 2023Data-efficient operator learning for solving high Mach number fluid flow problems28 Nov 2023
- 2023Block-local learning with probabilistic latent representations24 May 2023
- 2023FOSI: Hybrid First and Second Order Optimization16 Feb 2023
- 2022Layerwise Bregman Representation Learning with Applications to Knowledge Distillation15 Sep 2022
- 2022An alternative approach to train neural networks using monotone variational inequality17 Feb 2022
- 2022Gradient Descent on Neurons and its Link to Approximate Second-Order Optimization28 Jan 2022
12 newest of 14 All 14 (JSON)