固定网络的动态掩码学习:适应多任务的高效策略

0 下载量 41 浏览量 更新于2024-06-20 收藏 687KB PDF 举报
本文探讨了一种新颖的神经网络架构方法,旨在通过学习掩蔽权重来适应多个任务,而无需牺牲已学习任务的性能。这种方法借鉴了网络量化和规划的概念,提出了一种端到端可区分的机制,允许对固定深度神经网络的权重进行低开销的调整。研究者Arun Mallya、Dillon Davis和Svetlana Lazebnik来自伊利诺伊大学香槟分校,他们开发了一种技术,即在新任务上应用“规划块”,这些块可以应用于网络的原始权重,从而实现对网络的增量学习。 传统的深度网络微调往往导致旧任务性能下降的问题,也就是所谓的“为获取训练而学习任务”。作者们的目标是开发一种更高效的方法,既能增强网络的泛化能力,又能防止灾难性遗忘,同时减少额外参数的增加。他们提出的策略与传统的无遗忘学习(LwF)和弹性权重合并(EWC)不同,后者依赖于代理损失和正则化,而他们的方法则是通过迭代地学习掩码,而不是直接修改权重。 核心概念是“背负式掩码”,这是一种元素级别的操作,它可以在不改变底层网络结构的情况下,针对每个任务生成特定的权重掩码。这种掩码可以被设计成二进制,比如通过阈值函数,或者保持实数值,以有效地选择或抑制某些过滤器,仅在任务K上启用或关闭特定的网络连接。这种方法的优点在于它能够适应各种分类任务,即便这些任务的初始训练数据集(如ImageNet)与新任务之间存在较大差异。 作者们展示了他们的方法在处理不同任务时表现出与专用微调网络相当的性能,而且无论任务的顺序如何,都不会出现严重的任务遗忘或竞争问题。这表明他们的方法在保持网络灵活性和迁移学习能力方面取得了显著的成果,对于在资源有限且任务不断变化的场景下提升深度学习模型的实用性具有重要意义。 关键词:增量学习、二进制网络、灾难性遗忘、权重掩码、网络量化、固定骨干网络。 总结来说,这项研究为多任务学习提供了一种创新策略,通过掩码权重的学习来动态适应任务,这对于减少模型的复杂度、提高效率和保持旧任务性能具有潜在的实际价值。
2023-06-02 上传