Learning from the Past: Continual Meta-Learning via Bayesian Graph Modeling
This addresses the problem of retaining long-term knowledge in few-shot learning for AI systems, representing a novel method for a known bottleneck rather than an incremental improvement.
The paper tackles catastrophic forgetting and robustness issues in meta-learning by proposing a continual meta-learning approach with Bayesian Graph Neural Networks, achieving a 42.8% performance improvement on miniImageNet 5-way 1-shot classification compared to state-of-the-art methods.
Meta-learning for few-shot learning allows a machine to leverage previously acquired knowledge as a prior, thus improving the performance on novel tasks with only small amounts of data. However, most mainstream models suffer from catastrophic forgetting and insufficient robustness issues, thereby failing to fully retain or exploit long-term knowledge while being prone to cause severe error accumulation. In this paper, we propose a novel Continual Meta-Learning approach with Bayesian Graph Neural Networks (CML-BGNN) that mathematically formulates meta-learning as continual learning of a sequence of tasks. With each task forming as a graph, the intra- and inter-task correlations can be well preserved via message-passing and history transition. To remedy topological uncertainty from graph initialization, we utilize Bayes by Backprop strategy that approximates the posterior distribution of task-specific parameters with amortized inference networks, which are seamlessly integrated into the end-to-end edge learning. Extensive experiments conducted on the miniImageNet and tieredImageNet datasets demonstrate the effectiveness and efficiency of the proposed method, improving the performance by 42.8% compared with state-of-the-art on the miniImageNet 5-way 1-shot classification task.