Efficient Architecture Search for Continual Learning

By Qiang Gao, Zhipeng Luo, Diego Klabjan. 2020

75 citations recorded by the source corpus.

Continual learning with neural networks, which aims to learn a sequence of tasks, is an important learning framework in artificial intelligence (AI). However, it often confronts three challenges: 1) overcome the catastrophic forgetting problem; 2) adapt the current network to new tasks; and 3) control its model complexity. To reach these goals, we propose a novel approach named continual learning with efficient architecture search (CLEAS). CLEAS works closely with neural architecture search (NAS), which leverages reinforcement learning techniques to search for the best neural architecture that fits a new task. In particular, we design a neuron-level NAS controller that decides which old neurons from previous tasks should be reused (knowledge transfer) and which new neurons should be added (to learn new knowledge). Such a fine-grained controller allows finding a very concise architecture that can fit each new task well. Meanwhile, since we do not alter the weights of the reused neurons, we perfectly memorize the knowledge learned from the previous tasks. We evaluate CLEAS on numerous sequential classification tasks, and the results demonstrate that CLEAS outperforms other state-of-the-art alternative methods, achieving higher classification accuracy while using simpler neural architectures.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Bernt Schiele
  2. Mehrdad Farajtabar
  3. Qianru Sun
  4. Yaoyao Liu
  5. Piyush Rai
  6. Seyed Iman Mirzadeh
  7. Vinay Kumar Verma
  8. Lawrence Carin

In Computer Vision and Pattern Recognition

  1. Bernt Schiele
  2. Qianru Sun
  3. Luc Van Gool
  4. Mehrdad Farajtabar
  5. Lawrence Carin
  6. Dacheng Tao
  7. Yaoyao Liu
  8. Trevor Darrell

Citations in the corpus