Supervised Local Training with Backward Links for Deep Neural Networks
KAUST DepartmentDivision of Computer, Electrical and Mathematical Sciences and Engineering, King Abdullah University of Science and Technology, Thuwal, Saudi Arabia
Electrical and Computer Engineering Program
Computer, Electrical and Mathematical Science and Engineering (CEMSE) Division
Advanced Membranes and Porous Materials Research Center
Permanent link to this recordhttp://hdl.handle.net/10754/690008
MetadataShow full item record
AbstractThe restricted training pattern in the standard BP requires end-to-end error propagation, causing large memory costs and prohibiting model parallelization. Existing local training methods aim to resolve the training obstacles by completely cutting off the backward path between modules and isolating their gradients. These methods prevent information exchange between modules and result in inferior performance. This work proposes a novel local training algorithm, BackLink, which introduces inter-module backward dependency and facilitates information to flow backward along with the network. To preserve the computational advantage of local training, BackLink restricts the error propagation length within the module. Extensive experiments performed in various deep convolutional neural networks demonstrate that our method consistently improves the classification performance of local training algorithms over other methods. For example, our method can surpass the conventional greedy local training method by 6.45% in accuracy in ResNet32 classifying CIFAR100 and recent work by 2.58% in ResNet110 classifying STL-10 with much lower complexity, respectively. Analysis of computational costs reveals that small overheads are incurred in GPU memory costs and runtime on multiple GPUs. Our method can lead up to a 79% reduction in memory cost and 52% in simulation runtime in ResNet110 compared to the standard BP. Therefore, our method could create new opportunities for improving training algorithms towards better efficiency for real-time learning applications.
CitationGuo, W., Fouda, M. E., Eltawil, A. M., & Salama, K. N. (2023). Supervised Local Training with Backward Links for Deep Neural Networks. IEEE Transactions on Artificial Intelligence, 1–14. https://doi.org/10.1109/tai.2023.3251313
SponsorsThis work was funded by the King Abdullah University of Science and Technology (KAUST) AI Initiative, Saudi Arabia.