Maintaining Plasticity in Deep Continual Learning

التفاصيل البيبلوغرافية
العنوان: Maintaining Plasticity in Deep Continual Learning
المؤلفون: Dohare, Shibhansh, Hernandez-Garcia, J. Fernando, Rahman, Parash, Mahmood, A. Rupam, Sutton, Richard S.
سنة النشر: 2023
المجموعة: Computer Science
مصطلحات موضوعية: Computer Science - Machine Learning
الوصف: Modern deep-learning systems are specialized to problem settings in which training occurs once and then never again, as opposed to continual-learning settings in which training occurs continually. If deep-learning systems are applied in a continual learning setting, then it is well known that they may fail to remember earlier examples. More fundamental, but less well known, is that they may also lose their ability to learn on new examples, a phenomenon called loss of plasticity. We provide direct demonstrations of loss of plasticity using the MNIST and ImageNet datasets repurposed for continual learning as sequences of tasks. In ImageNet, binary classification performance dropped from 89% accuracy on an early task down to 77%, about the level of a linear network, on the 2000th task. Loss of plasticity occurred with a wide range of deep network architectures, optimizers, activation functions, batch normalization, dropout, but was substantially eased by L2-regularization, particularly when combined with weight perturbation. Further, we introduce a new algorithm -- continual backpropagation -- which slightly modifies conventional backpropagation to reinitialize a small fraction of less-used units after each example and appears to maintain plasticity indefinitely.
نوع الوثيقة: Working Paper
URL الوصول: http://arxiv.org/abs/2306.13812
رقم الانضمام: edsarx.2306.13812
قاعدة البيانات: arXiv