Skip to content
DraconiticResearch
Search Sources API
203,234 studies31 sourcesUpdated 27 min ago
PreprintarXiv licencearXiv

LocalProp: Neuro-Localized Memory-Efficient Backpropagation

Abstract

The current deep learning training paradigm employs end-to-end backpropagation, regardless of the training stage, i.e. pre-training or fine-tuning. However, backpropagating through the entire model is neither biologically plausible nor memory efficient, since learning inside the brain is highly localized. Therefore, we propose LocalProp, a training procedure that locally updates the weights of a model. Our neuro-localized weight updates follow the "pre-training then fine-tuning" paradigm, where the pre-training is based on I-JEPA. After locally updating the weights, a pruning operation is performed, followed by a short final fine-tuning phase. Pruning helps by sending the learning signal from higher blocks to lower blocks. We perform experiments on several datasets, including large-scale benchmarks such as ImageNet, and empirically show that LocalProp reaches good performance at a fraction of GPU peak memory. By varying the number of jointly optimized blocks, we identify gradient-propagation span as a practical control over the accuracy-memory trade-off.

Related

  1. Localized charges, reset noise, and boundary memory in matrix-product-conserving quantum chains
    Ron Rubin
    arXivPreprintarXiv licenceFull textMathematical physicsQuantum physics & quantum informationarXiv
  2. Jacobian-Free Unrolling: memory efficient unrolled network for 3D SPECT Reconstruction
    Corentin Constanza, Théo Kaprelian, Ane Etxebeste +1
    IEEE Transactions on Radiation and Plasma Medical SciencesPeer-reviewedPublisher termsFull textSystems & synthetic biologyOptimisation & controlHAL
  3. PowerStep: Memory-Efficient Adaptive Optimization via ℓp-Norm Steepest Descent
    Yao Lu, Dengdong Fan, Shixun Zhang +1
    arXivPreprintCC BY 4.0Full textNumerical analysis & scientific computingOptimisation & controlarXiv
  4. Finite-Memory Control of POMDPs: Fundamental Limits and Efficient Design
    Emrecan Kutay, Atilla Eryilmaz, Ness Shroff
    arXivPreprintCC BY 4.0Full textGeneral mathematicsarXiv
  5. Low-Rank Friction for Memory-Efficient Transformer Pretraining
    Rajit Rajpal, Benedict Leimkuhler
    arXivPreprintCC BY 4.0Full textStatisticsarXiv
BibTeX
@misc{grigore2026localprop,
  title = {{LocalProp: Neuro-Localized Memory-Efficient Backpropagation}},
  author = {Diana-Nicoleta Grigore and Iuliana Georgescu and Radu Tudor Ionescu},
  howpublished = {arXiv},
  year = {2026},
  month = sep,
  doi = {10.48550/arxiv.2609.32517},
  eprint = {2609.32517},
  archivePrefix = {arXiv},
  url = {https://arxiv.org/abs/2609.32517},
}
RIS
TY  - UNPB
TI  - LocalProp: Neuro-Localized Memory-Efficient Backpropagation
AU  - Diana-Nicoleta Grigore
AU  - Iuliana Georgescu
AU  - Radu Tudor Ionescu
PY  - 2026
DA  - 2026/09/26
PB  - arXiv
DO  - 10.48550/arxiv.2609.32517
UR  - https://arxiv.org/abs/2609.32517
AB  - The current deep learning training paradigm employs end-to-end backpropagation, regardless of the training stage, i.e. pre-training or fine-tuning. However, backpropagating through the entire model is neither biologically plausible nor memory efficient, since learning inside the brain is highly localized. Therefore, we propose LocalProp, a training procedure that locally updates the weights of a model. Our neuro-localized weight updates follow the "pre-training then fine-tuning" paradigm, where the pre-training is based on I-JEPA. After locally updating the weights, a pruning operation is performed, followed by a short final fine-tuning phase. Pruning helps by sending the learning signal from higher blocks to lower blocks. We perform experiments on several datasets, including large-scale benchmarks such as ImageNet, and empirically show that LocalProp reaches good performance at a fraction of GPU peak memory. By varying the number of jointly optimized blocks, we identify gradient-propagation span as a practical control over the accuracy-memory trade-off.
ER  -