arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.02428cs.LG
收藏

利用神经单元动力学实现有效且可扩展的类增量学习

Harnessing Neural Unit Dynamics for Effective and Scalable Class-Incremental Learning

School of Artificial Intelligence and Automation, Huazhong University of Science and Technology(华中科技大学人工智能与自动化学院);School of Information and Control Engineering, China University of Mining and Technology(中国矿业大学信息与控制工程学院)

Depeng Li, Tianqi Wang, Junwei Chen, Wei Dai, Zhigang Zeng

AI总结:

提出基于神经单元动力学的类增量学习模型AutoActivator,通过自适应网络扩展与推理时神经单元激活机制,在无排练和最小扩展设置下有效防止遗忘并保持可扩展性。

摘要:

类增量学习(CIL)旨在训练模型从非平稳数据流中学习新类而不遗忘旧类。本文提出一种新型连接主义模型,通过定制神经单元动力学来调整神经网络在CIL中的行为。在每个训练会话中,该模型引入监督机制指导网络扩展,其增长规模与新到达任务的内在复杂度紧凑匹配。这构建了近最小网络,同时允许模型在无法充分容纳新类时扩展其容量。在推理阶段,它自动重新激活所需神经单元以检索知识,并保持其余单元未激活以防止干扰。我们将此模型命名为AutoActivator,它既有效又可扩展。为深入理解神经单元动力学,我们通过学习序列映射的通用近似定理,对模型的收敛性进行了理论分析,这在CIL领域尚属未被充分探索的方向。实验表明,该方法在不同骨干网络的无排练和最小扩展设置下均取得了强劲的CIL性能。

英文摘要:

Class-incremental learning (CIL) aims to train a model to learn new classes from non-stationary data streams without forgetting old ones. In this paper, we propose a new kind of connectionist model by tailoring neural unit dynamics that adapt the behavior of neural networks for CIL. In each training session, it introduces a supervisory mechanism to guide network expansion whose growth size is compactly commensurate with the intrinsic complexity of a newly arriving task. This constructs a near-minimal network while allowing the model to expand its capacity when cannot sufficiently hold new classes. At inference time, it automatically reactivates the required neural units to retrieve knowledge and leaves the remaining inactivated to prevent interference. We name our model AutoActivator, which is effective and scalable. To gain insights into the neural unit dynamics, we theoretically analyze the model's convergence property via a universal approximation theorem on learning sequential mappings, which is under-explored in the CIL community. Experiments show that our method achieves strong CIL performance in rehearsal-free and minimal-expansion settings with different backbones.

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

↑