arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2311.01617cs.CVcs.AIcs.LGcs.NE

面向视觉任务持续学习的前瞻选择性可塑性

Look-Ahead Selective Plasticity for Continual Learning of Visual Tasks

University of Western Ontario(西安大略大学)

Rouzbeh Meshkinnejad, Jie Mei, Daniel Lizotte, Yalda Mohsenzadeh

AI总结:

本文受大脑事件模型启发,提出任务边界处的前瞻选择性可塑性机制,利用对比损失冗余性和新任务少量样本保留关键迁移参数,在多种持续学习场景取得最先进性能。

摘要:

对比表征学习已成为持续学习中一种颇具前景的技术,因为它能够学习对灾难性遗忘具有鲁棒性、并能很好泛化到未见未来任务的表征。以往的持续学习工作通过使用先前任务数据和已训练模型来解决遗忘问题。受大脑中创建并更新的事件模型启发,我们提出了一种发生在任务边界期间(即一个任务结束、另一个任务开始时)的新机制。通过观察对比损失在神经网络输出上诱发冗余的能力,我们的方法利用新任务的前几个样本,识别并保留对神经网络迁移能力贡献最大的参数,同时释放网络的其余部分以学习新特征。我们在包括 CIFAR10 和 TinyImagenet 在内的基准计算机视觉数据集上评估了所提出的方法,并在任务增量、类别增量和领域增量持续学习场景中展示了最先进的性能。

英文摘要:

Contrastive representation learning has emerged as a promising technique for continual learning as it can learn representations that are robust to catastrophic forgetting and generalize well to unseen future tasks. Previous work in continual learning has addressed forgetting by using previous task data and trained models. Inspired by event models created and updated in the brain, we propose a new mechanism that takes place during task boundaries, i.e., when one task finishes and another starts. By observing the redundancy-inducing ability of contrastive loss on the output of a neural network, our method leverages the first few samples of the new task to identify and retain parameters contributing most to the transfer ability of the neural network, freeing up the remaining parts of the network to learn new features. We evaluate the proposed methods on benchmark computer vision datasets including CIFAR10 and TinyImagenet and demonstrate state-of-the-art performance in the task-incremental, class-incremental, and domain-incremental continual learning scenarios.

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

↑