A Simple Lifelong Learning Approach
Joshua T. Vogelstein,Jayanta Dey,Hayden S. Helm,Will LeVine,Ronak D. Mehta,Tyler M. Tomita,Haoyin Xu,Ali Geisa,Qingyang Wang,Gido M. van de Ven,Chenyu Gao,Weiwei Yang,Bryan Tower,Jonathan Larson,Christopher M. White,Carey E. Priebe
2024-06-12
Abstract:In lifelong learning, data are used to improve performance not only on the present task, but also on past and future (unencountered) tasks. While typical transfer learning algorithms can improve performance on future tasks, their performance on prior tasks degrades upon learning new tasks (called forgetting). Many recent approaches for continual or lifelong learning have attempted to maintain performance on old tasks given new tasks. But striving to avoid forgetting sets the goal unnecessarily low. The goal of lifelong learning should be to use data to improve performance on both future tasks (forward transfer) and past tasks (backward transfer). In this paper, we show that a simple approach -- representation ensembling -- demonstrates both forward and backward transfer in a variety of simulated and benchmark data scenarios, including tabular, vision (CIFAR-100, 5-dataset, Split Mini-Imagenet, and Food1k), and speech (spoken digit), in contrast to various reference algorithms, which typically failed to transfer either forward or backward, or both. Moreover, our proposed approach can flexibly operate with or without a computational budget.
Artificial Intelligence,Machine Learning