Equivariant Experience Replay for Continual Learning in Imaging Inverse Problems

Overview & Synopsis

In this work, we propose Equivariant Experience Replay, a new paradigm for achieving continual learning of deep networks for solving ill-posed imaging inverse problems. Inspired by recent advances on unsupervised learning methods such as equivariant imaging/splitting, we find the exploitation of group-symmetry structure is particularly relevant and effective in continual learning setting for ill-posed inversion tasks. We therefore propose this new experience replay scheme, which is tailored for such inverse problems. Our numerical study demonstrates the superiority of our method compared to existing continual learning methods for inverse problems in medical imaging and image processing.