Plastic and stable gated classifiers for continual learning
: Kuo Nicholas I-Hsien, Harandi Mehrtash, Fourrier Nicolas, Walder Christian, Ferraro Gabriela, Suominen Hanna
: IEEE Computer Society Conference on Computer Vision and Pattern Recognition workshops
Publisher: IEEE Computer Society
: 2021
: IEEE Computer Society Conference on Computer Vision and Pattern Recognition Workshops
: 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW)
: IEEE Computer Society Conference on Computer Vision and Pattern Recognition Workshops
: 3548
: 3553
: 978-1-6654-4900-7
: 978-1-6654-4899-4
: 2160-7516
DOI: https://doi.org/10.1109/CVPRW53098.2021.00394
Conventional neural networks are mostly high in plasticity but low in stability. Hence, catastrophic forgetting tends to occur over the sequential training of multiple tasks and a backbone learner loses its ability in solving a previously learnt task. Several studies have shown that catastrophic forgetting can be partially mitigated through freezing the feature extractor weights while only sequentially training the classifier network. Though these are effective methods in retaining knowledge, forgetting could still become severe if the classifier network is over-parameterised over many tasks. As a remedy, this paper presents a novel classifier design with high stability. Highway-Connection Classifier Networks (HCNs) leverage gated units to alleviate forgetting. When employed alone, they exhibit strong robustness against forgetting. In addition, they synergise well with many existing and popular continual learning archetypes. We release our codes at https://github.com/Nic5472K/CLVISION2021_CVPR_HCN