English

MOLE: MOdular Learning FramEwork via Mutual Information Maximization

Machine Learning 2026-05-28 v1 Artificial Intelligence

Abstract

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.

Keywords

Cite

@article{arxiv.2308.07772,
  title  = {MOLE: MOdular Learning FramEwork via Mutual Information Maximization},
  author = {Tianchao Li and Yulong Pei},
  journal= {arXiv preprint arXiv:2308.07772},
  year   = {2026}
}

Comments

accepted by icml llw

R2 v1 2026-06-28T11:56:03.904Z