English

A picture of the space of typical learnable tasks

Machine Learning 2023-07-25 v4

Abstract

We develop information geometric techniques to understand the representations learned by deep networks when they are trained on different tasks using supervised, meta-, semi-supervised and contrastive learning. We shed light on the following phenomena that relate to the structure of the space of tasks: (1) the manifold of probabilistic models trained on different tasks using different representation learning methods is effectively low-dimensional; (2) supervised learning on one task results in a surprising amount of progress even on seemingly dissimilar tasks; progress on other tasks is larger if the training task has diverse classes; (3) the structure of the space of tasks indicated by our analysis is consistent with parts of the Wordnet phylogenetic tree; (4) episodic meta-learning algorithms and supervised learning traverse different trajectories during training but they fit similar models eventually; (5) contrastive and semi-supervised learning methods traverse trajectories similar to those of supervised learning. We use classification tasks constructed from the CIFAR-10 and Imagenet datasets to study these phenomena.

Keywords

Cite

@article{arxiv.2210.17011,
  title  = {A picture of the space of typical learnable tasks},
  author = {Rahul Ramesh and Jialin Mao and Itay Griniasty and Rubing Yang and Han Kheng Teoh and Mark Transtrum and James P. Sethna and Pratik Chaudhari},
  journal= {arXiv preprint arXiv:2210.17011},
  year   = {2023}
}