Continual learning using Bayesian neural networks
File(s)CBLN__Fianl_Version_.pdf (689.45 KB)
Accepted version
Author(s)
Li, Honglin
Barnaghi, Payam
Enshaeifar, Shirin
Ganz, Frieder
Type
Journal Article
Abstract
Continual learning models allow them to learn and adapt to new changes and tasks over time. However, in continual and sequential learning scenarios, in which the models are trained using different data with various distributions, neural networks (NNs) tend to forget the previously learned knowledge. This phenomenon is often referred to as catastrophic forgetting. The catastrophic forgetting is an inevitable problem in continual learning models for dynamic environments. To address this issue, we propose a method, called continual Bayesian learning networks (CBLNs), which enables the networks to allocate additional resources to adapt to new tasks without forgetting the previously learned tasks. Using a Bayesian NN, CBLN maintains a mixture of Gaussian posterior distributions that are associated with different tasks. The proposed method tries to optimize the number of resources that are needed to learn each task and avoids an exponential increase in the number of resources that are involved in learning multiple tasks. The proposed method does not need to access the past training data and can choose suitable weights to classify the data points during the test time automatically based on an uncertainty criterion. We have evaluated the method on the MNIST and UCR time-series data sets. The evaluation results show that the method can address the catastrophic forgetting problem at a promising rate compared to the state-of-the-art models.
Date Issued
2020-08-31
Date Acceptance
2020-08-08
Citation
IEEE Transactions on Neural Networks and Learning Systems, 2020, 32 (9), pp.4243-4252
ISSN
2162-2388
Publisher
Institute of Electrical and Electronics Engineers
Start Page
4243
End Page
4252
Journal / Book Title
IEEE Transactions on Neural Networks and Learning Systems
Volume
32
Issue
9
Copyright Statement
© 2020 IEEE. Personal use of this material is permitted. Permission from IEEE must be obtained for all other uses, in any current or future media, including reprinting/republishing this material for advertising or promotional purposes, creating new collective works, for resale or redistribution to servers or lists, or reuse of any copyrighted component of this work in other works. This is the Accepted Manuscript version of a published work appearing in IEEE Transactions on Neural Networks and Learning Systems, https://doi.org/10.1109/TNNLS.2020.3017292
Sponsor
Medical Research Council
Identifier
https://www.ncbi.nlm.nih.gov/pubmed/32866104
Grant Number
UKDRI-7002
Subjects
Science & Technology
Technology
Computer Science, Artificial Intelligence
Computer Science, Hardware & Architecture
Computer Science, Theory & Methods
Engineering, Electrical & Electronic
Computer Science
Engineering
Task analysis
Adaptation models
Training
Bayes methods
Modeling
Uncertainty
Gaussian distribution
Bayesian neural networks (BNNs)
catastrophic forgetting
continual learning
incremental learning
uncertainty
SYSTEMS
cs.LG
cs.LG
cs.NE
stat.ML
Artificial Intelligence & Image Processing
Publication Status
Published
Coverage Spatial
United States
Date Publish Online
2020-08-31