A4 Refereed article in a conference publication
Plastic and stable gated classifiers for continual learning
Authors: Kuo Nicholas I-Hsien, Harandi Mehrtash, Fourrier Nicolas, Walder Christian, Ferraro Gabriela, Suominen Hanna
Conference name: IEEE Computer Society Conference on Computer Vision and Pattern Recognition workshops
Publisher: IEEE Computer Society
Publication year: 2021
Journal: IEEE Computer Society Conference on Computer Vision and Pattern Recognition Workshops
Book title : 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW)
Journal name in source: IEEE Computer Society Conference on Computer Vision and Pattern Recognition Workshops
First page : 3548
Last page: 3553
ISBN: 978-1-6654-4900-7
eISBN: 978-1-6654-4899-4
ISSN: 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