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

面向深度协作学习的知识转移图

Knowledge Transfer Graph for Deep Collaborative Learning

Chubu University(中部大学)

Soma Minami, Tsubasa Hirakawa, Takayoshi Yamashita, Hironobu Fujiyoshi

摘要:

通过大量的手工设计,利用其输出或中间激活在多个网络之间进行知识转移,从简单的师生方法(知识蒸馏)演变为双向群组方法(深度相互学习)。影响知识转移的关键因素包括网络规模、网络数量、转移方向和损失函数的设计。然而,由于这些因素在组合时是巨大的,并且变得错综复杂,传统的知识转移方法只能探索有限的组合。在本文中,我们提出了一种新的基于图的方法,更灵活和多样化的知识转移组合。为了实现知识转移,我们提出了一种新的图形表示称为知识转移图,它提供了一个统一的知识转移视图,并有可能代表不同的知识转移模式。我们还提出了四个门函数,引入到损失函数。控制梯度的四个门可以提供知识转移的多种组合。搜索图结构使我们能够发现更有效的知识转移方法比手动设计的。在CIFAR-10、CIFAR-100和Tiny-ImageNet数据集上的实验结果表明,该方法实现了显著的性能提升,并且能够找到显着的图结构。

英文摘要:

Knowledge transfer among multiple networks using their outputs or intermediate activations have evolved through extensive manual design from a simple teacher-student approach (knowledge distillation) to a bidirectional cohort one (deep mutual learning). The key factors of such knowledge transfer involve the network size, the number of networks, the transfer direction, and the design of the loss function. However, because these factors are enormous when combined and become intricately entangled, the methods of conventional knowledge transfer have explored only limited combinations. In this paper, we propose a new graph-based approach for more flexible and diverse combinations of knowledge transfer. To achieve the knowledge transfer, we propose a novel graph representation called knowledge transfer graph that provides a unified view of the knowledge transfer and has the potential to represent diverse knowledge transfer patterns. We also propose four gate functions that are introduced into loss functions. The four gates, which control the gradient, can deliver diverse combinations of knowledge transfer. Searching the graph structure enables us to discover more effective knowledge transfer methods than a manually designed one. Experimental results on the CIFAR-10, -100, and Tiny-ImageNet datasets show that the proposed method achieved significant performance improvements and was able to find remarkable graph structures.

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

↑