arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2209.07712cs.LGcs.CV
收藏

基于依赖保持超网络的持续学习

Continual Learning with Dependency Preserving Hypernetworks

Indian Institute of Technology(印度理工学院);Deakin University(迪肯大学)

Dupati Srikar Chandra, Sakshi Varshney, P. K. Srijith, Sunil Gupta

摘要:

人类在一生中不断学习,积累各种知识,并为未来的任务进行微调。当面对类似的目标时,如果连续任务之间的数据分布在学习过程中不是平稳的,那么神经网络就会遭受灾难性遗忘。解决这种持续学习(CL)问题的一种有效方法是使用超网络,该超网络为目标网络生成任务相关的权重。然而,现有的基于超网络的方法的持续学习性能受到跨层权重独立性的假设的影响,以保持参数效率。为了解决这个问题,我们提出了一种新的方法,使用一个依赖保持超网络来生成目标网络的权重,同时还保持参数的效率。我们建议使用基于递归神经网络(RNN)的超网络,它可以有效地生成层权重,同时允许它们之间的依赖关系。此外,我们提出了新的正则化和基于RNN的超网络的网络增长技术,以进一步提高持续学习的性能。为了证明所提出的方法的有效性,我们进行了几个图像分类持续学习任务和设置的实验。我们发现,基于RNN超网络的方法在所有这些CL设置和任务中的表现都优于基线。

英文摘要:

Humans learn continually throughout their lifespan by accumulating diverse knowledge and fine-tuning it for future tasks. When presented with a similar goal, neural networks suffer from catastrophic forgetting if data distributions across sequential tasks are not stationary over the course of learning. An effective approach to address such continual learning (CL) problems is to use hypernetworks which generate task dependent weights for a target network. However, the continual learning performance of existing hypernetwork based approaches are affected by the assumption of independence of the weights across the layers in order to maintain parameter efficiency. To address this limitation, we propose a novel approach that uses a dependency preserving hypernetwork to generate weights for the target network while also maintaining the parameter efficiency. We propose to use recurrent neural network (RNN) based hypernetwork that can generate layer weights efficiently while allowing for dependencies across them. In addition, we propose novel regularisation and network growth techniques for the RNN based hypernetwork to further improve the continual learning performance. To demonstrate the effectiveness of the proposed methods, we conducted experiments on several image classification continual learning tasks and settings. We found that the proposed methods based on the RNN hypernetworks outperformed the baselines in all these CL settings and tasks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑