# Accumulated Decoupled Learning: Mitigating Gradient Staleness in Inter-Layer Model Parallelization

Source: [Accumulated Decoupled Learning: Mitigating Gradient Staleness in Inter-Layer Model Parallelization](<https://www.sota2.com/research/paper/68fa68e11e209131cea1eee7>)

Collection index: [llms.txt](<https://www.sota2.com/research/llms.txt>)

Data updated: 2025-10-25T22:48:52.480Z

Authors: Huiping Zhuang, Zhiping Lin, Kar-Ann Toh

Published: 2020-12-03T11:52:55.000Z

## Abstract

Decoupled learning is a branch of model parallelism which parallelizes the training of a network by splitting it depth-wise into multiple modules. Techniques from decoupled learning usually lead to stale gradient effect because of their asynchronous implementation, thereby causing performance degradation. In this paper, we propose an accumulated decoupled learning (ADL) which incorporates the gradient accumulation technique to mitigate the stale gradient effect. We give both theoretical and empirical evidences regarding how the gradient staleness can be reduced. We prove that the proposed method can converge to critical points, i.e., the gradients converge to 0, in spite of its asynchronous nature. Empirical validation is provided by training deep convolutional neural networks to perform classification tasks on CIFAR-10 and ImageNet datasets. The ADL is shown to outperform several state-of-the-arts in the classification tasks, and is the fastest among the compared methods.

## Recorded benchmark results

A research result applies to its stated dataset, split, metric and evaluation configuration. Compare like-for-like settings, preserve the metric's unit and direction, and distinguish paper-reported results from independent reproduction. The best result recorded here is not a claim of exhaustive, current global coverage.

## Original sources and implementation

[Original paper](<https://arxiv.org/abs/2012.03747>)

[PDF](<https://arxiv.org/pdf/2012.03747.pdf>)
