arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2308.07772cs.LGcs.AI

MOLE: 基于互信息最大化的模块化学习框架

MOLE: MOdular Learning FramEwork via Mutual Information Maximization

Department of Mathematics(数学系);Computer Science, Eindhoven University of Technology, Eindhoven, the Netherland(计算机科学系,埃因霍温理工大学,埃因霍温,荷兰)

Tianchao Li, Yulong Pei

AI总结:

提出一种异步局部学习框架MOLE,通过层间模块化与互信息最大化实现梯度隔离的局部优化,适用于向量、网格和图数据,并在图级别和节点级别任务上验证了通用性。

摘要:

本文介绍了一种名为模块化学习框架(MOLE)的异步局部学习框架,用于神经网络。该框架按层将神经网络模块化,通过互信息为每个模块定义训练目标,并通过互信息最大化依次训练每个模块。MOLE使训练成为跨模块梯度隔离的局部优化,这种方案比BP更具生物合理性。我们在向量、网格和图类型数据上进行了实验。特别地,该框架能够解决图类型数据的图级别和节点级别任务。因此,MOLE已被实验证明可普遍适用于不同类型的数据。

英文摘要:

This paper is to introduce an asynchronous and local learning framework for neural networks, named Modular Learning Framework (MOLE). This framework modularizes neural networks by layers, defines the training objective via mutual information for each module, and sequentially trains each module by mutual information maximization. MOLE makes the training become local optimization with gradient-isolated across modules, and this scheme is more biologically plausible than BP. We run experiments on vector-, grid- and graph-type data. In particular, this framework is capable of solving both graph- and node-level tasks for graph-type data. Therefore, MOLE has been experimentally proven to be universally applicable to different types of data.

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

↑