Addressing Catastrophic Forgetting in Few-Shot Problems
Authors:
Pauching Yap,
Hippolyt Ritter,
David Barber
Abstract:
Neural networks are known to suffer from catastrophic forgetting when trained on sequential datasets. While there have been numerous attempts to solve this problem in large-scale supervised classification, little has been done to overcome catastrophic forgetting in few-shot classification problems. We demonstrate that the popular gradient-based model-agnostic meta-learning algorithm (MAML) indeed…
▽ More
Neural networks are known to suffer from catastrophic forgetting when trained on sequential datasets. While there have been numerous attempts to solve this problem in large-scale supervised classification, little has been done to overcome catastrophic forgetting in few-shot classification problems. We demonstrate that the popular gradient-based model-agnostic meta-learning algorithm (MAML) indeed suffers from catastrophic forgetting and introduce a Bayesian online meta-learning framework that tackles this problem. Our framework utilises Bayesian online learning and meta-learning along with Laplace approximation and variational inference to overcome catastrophic forgetting in few-shot classification problems. The experimental evaluations demonstrate that our framework can effectively achieve this goal in comparison with various baselines. As an additional utility, we also demonstrate empirically that our framework is capable of meta-learning on sequentially arriving few-shot tasks from a stationary task distribution.
△ Less
Submitted 21 June, 2021; v1 submitted 30 April, 2020;
originally announced May 2020.
Robust Group Comparison Using Non-Parametric Block-Based Statistics
Authors:
Geng Chen,
Pei Zhang,
Ke Li,
Chong-Yaw Wee,
Wenliang Pan,
Yafeng Wu,
Panteleimon Giannakopoulos,
Sven Haller,
Dinggang Shen,
Pew-Thian Yap
Abstract:
Voxel-based analysis methods localize brain structural differences by performing voxel-wise statistical comparisons on two groups of images aligned to a common space. This procedure requires highly accurate registration as well as a sufficiently large dataset. However, in practice, the registration algorithms are not perfect due to noise, artifacts, and complex structural variations. The sample si…
▽ More
Voxel-based analysis methods localize brain structural differences by performing voxel-wise statistical comparisons on two groups of images aligned to a common space. This procedure requires highly accurate registration as well as a sufficiently large dataset. However, in practice, the registration algorithms are not perfect due to noise, artifacts, and complex structural variations. The sample size is also limited due to low disease prevalence, recruitment difficulties, and demographic matching issues. To address these issues, in this paper, we propose a method, called block-based statistic (BBS), for robust group comparison. BBS consists of two major components: Block matching and permutation test. Specifically, based on two group of images aligned to a common space, we first perform block matching so that structural misalignments can be corrected. Then, based on results given by block matching, we conduct robust non-parametric statistical inference based on permutation test. Extensive experiments were performed on synthetic data and the real diffusion MR data of mild cognitive impairment patients. The experimental results indicate that BBS significantly improves statistical power, notwithstanding the small sample size.
△ Less
Submitted 24 July, 2018;
originally announced July 2018.