NetworkFF: Unified Layer Optimization in Forward-Only Neural Networks
Salar Beigzad
TL;DR
The paper tackles the memory and plausibility limitations of backpropagation by leveraging Forward-Forward learning. It introduces Collaborative Forward-Forward (CFF) that enables inter-layer coordination via fixed and adaptive collaboration schemes. Extensive experiments on MNIST and Fashion-MNIST show that CFF improves accuracy over baseline FF, with adaptive CFF providing the largest gains. The results suggest that inter-layer collaboration preserves memory efficiency and biological plausibility while boosting convergence and performance, with implications for neuromorphic and energy-constrained AI.
Abstract
The Forward-Forward algorithm eliminates backpropagation's memory constraints and biological implausibility through dual forward passes with positive and negative data. However, conventional implementations suffer from critical inter-layer isolation, where layers optimize goodness functions independently without leveraging collective learning dynamics. This isolation constrains representational coordination and limits convergence efficiency in deeper architectures. This paper introduces Collaborative Forward-Forward (CFF) learning, extending the original algorithm through inter-layer cooperation mechanisms that preserve forward-only computation while enabling global context integration. Our framework implements two collaborative paradigms: Fixed CFF (F-CFF) with constant inter-layer coupling and Adaptive CFF (A-CFF) with learnable collaboration parameters that evolve during training. The collaborative goodness function incorporates weighted contributions from all layers, enabling coordinated feature learning while maintaining memory efficiency and biological plausibility. Comprehensive evaluation on MNIST and Fashion-MNIST demonstrates significant performance improvements over baseline Forward-Forward implementations. These findings establish inter-layer collaboration as a fundamental enhancement to Forward-Forward learning, with immediate applicability to neuromorphic computing architectures and energy-constrained AI systems.
