English

Improving Generalization by Permutation Routing Across Model Copies

Machine Learning 2026-05-12 v1 Artificial Intelligence Machine Learning

Abstract

We introduce a use of the MM-cover (or MM-layer) transform for machine learning. The method replicates a model MM times, but instead of coupling the copies through parameter averaging or an explicit attractive force, as in replicated SGD or Elastic SGD, it rewires the contexts in which local learning messages are computed. Each local loss is evaluated on a routed model whose parameters are drawn from different copies according to permutations sampled from a structured mixing kernel QQ. Training then uses the original local update rule, while the resulting learning messages are redistributed across the copies through these routed computational paths. Thus QQ defines a topology for message transport and controls the long-loop structure of the lifted factor graph. We formulate this construction for perceptrons, committee machines, and multilayer perceptrons, showing that the same principle applies from discrete models to differentiable neural networks. The resulting framework provides a mechanism for improving generalization through structured message sharing rather than replica collapse or parameter-space coupling.

Keywords

Cite

@article{arxiv.2605.09256,
  title  = {Improving Generalization by Permutation Routing Across Model Copies},
  author = {Shuhei Kashiwamura and Timothee Leleu},
  journal= {arXiv preprint arXiv:2605.09256},
  year   = {2026}
}
R2 v1 2026-07-01T13:01:04.665Z