Modern deep neural networks rely on depth and model complexity to enhance their representational power, yet excessive network depth introduces a critical challenge: supervision vanishing in intermediate layers. In this paper, we propose Locality-aware Backward Propagation (LBP), a training framework that resolves this through a dual-phase mechanism: in Stage 1, strict gradient confinement ensures segment-specific subgraph optimization without interference from distant gradients; in Stage 2, global coherence is harmonized via multi-loss-driven updates. By decoupling localized supervision from global objectives, LBP not only alleviates supervision vanishing but also enables stable gradient accumulation, where global attention mechanisms amplify parameter sensitivity. The experimental results demonstrate that our proposed method not only achieves competitive performance on CNN architectures across both CIFAR-10/100 datasets but also delivers remarkable improvements on Vision Transformers.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Locality-Aware Backward Propagation for Deep Neural Networks

  • Shangke Yuan,
  • Qing Du,
  • Mingkui Tan

摘要

Modern deep neural networks rely on depth and model complexity to enhance their representational power, yet excessive network depth introduces a critical challenge: supervision vanishing in intermediate layers. In this paper, we propose Locality-aware Backward Propagation (LBP), a training framework that resolves this through a dual-phase mechanism: in Stage 1, strict gradient confinement ensures segment-specific subgraph optimization without interference from distant gradients; in Stage 2, global coherence is harmonized via multi-loss-driven updates. By decoupling localized supervision from global objectives, LBP not only alleviates supervision vanishing but also enables stable gradient accumulation, where global attention mechanisms amplify parameter sensitivity. The experimental results demonstrate that our proposed method not only achieves competitive performance on CNN architectures across both CIFAR-10/100 datasets but also delivers remarkable improvements on Vision Transformers.