Leveraging Information Flow for Knowledge Transfer in Continual Learning
摘要
Continual Learning (CL) has proven an effective method of avoiding Catastrophic Forgetting (CF) when sequentially training neural networks. This improves network efficiency while facilitating the Knowledge Transfer (KT) between tasks. CL serves as an ideal setting for studying KT and network behavior. In particular, pruning methods for CL train subnetworks to handle the sequential tasks which allows us to take a structured approach to investigating KT. This subnetwork approach allows for the promotion of Forward KT (FKT) and complete prevention of negative Backward KT (BKT) and CF. Understanding which weights to share between tasks is crucial for FKT as sharing all weights can worsen accuracy. This paper demonstrates how the information flow (IF) within the network can reflect task similarity and usefulness, informing optimal sharing decisions to improve FKT between diverse tasks. We leverage IF to determine which weights to share to promote FKT. By doing this, we match the optimal sharing decisions for each implemented dataset, and compare the resulting accuracies against several benchmark methods which emphasize promoting FKT and BKT. Experiments are run on three CL datasets designed to emphasize variation in task complexity and similarity, using both ResNet-18 and VGG-16.