MAN++: Scaling Momentum Auxiliary Network for Supervised Local Learning in Vision Tasks

TL;DR

MAN++ introduces a novel auxiliary network with EMA-based inter-block communication and learnable bias, enabling efficient supervised local learning with reduced memory usage across vision tasks.

Contribution

The paper proposes MAN++, a new auxiliary network that enhances local learning by using EMA of parameters and learnable bias, improving performance and efficiency.

Findings

MAN++ achieves comparable performance to end-to-end training.

It significantly reduces GPU memory consumption.

Effective across multiple vision tasks and architectures.

Abstract

Deep learning typically relies on end-to-end backpropagation for training, a method that inherently suffers from issues such as update locking during parameter optimization, high GPU memory consumption, and a lack of biological plausibility. In contrast, supervised local learning seeks to mitigate these challenges by partitioning the network into multiple local blocks and designing independent auxiliary networks to update each block separately. However, because gradients are propagated solely within individual local blocks, performance degradation occurs, preventing supervised local learning from supplanting end-to-end backpropagation. To address these limitations and facilitate inter-block information flow, we propose the Momentum Auxiliary Network++ (MAN++). MAN++ introduces a dynamic interaction mechanism by employing the Exponential Moving Average (EMA) of parameters from adjacent blocks to enhance communication across the network. The auxiliary network, updated via EMA, effectively bridges the information gap between blocks. Notably, we observed that directly applying EMA parameters can be suboptimal due to feature discrepancies between local blocks. To resolve this issue, we introduce a learnable scaling bias that balances feature differences, thereby further improving performance. We validate MAN++ through extensive experiments on tasks that include image classification, object detection, and image segmentation, utilizing multiple network architectures. The experimental results demonstrate that MAN++ achieves performance comparable to end-to-end training while significantly reducing GPU memory usage. Consequently, MAN++ offers a novel perspective for supervised local learning and presents a viable alternative to conventional training methods.