Modeling Structured Data Learning with Restricted Boltzmann Machines in the Teacher-Student Setting

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Thériault, Robin, Tosello, Francesco, Tantari, Daniele
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912383298961408
author Thériault, Robin
Tosello, Francesco
Tantari, Daniele
author_facet Thériault, Robin
Tosello, Francesco
Tantari, Daniele
contents Restricted Boltzmann machines (RBM) are generative models capable to learn data with a rich underlying structure. We study the teacher-student setting where a student RBM learns structured data generated by a teacher RBM. The amount of structure in the data is controlled by adjusting the number of hidden units of the teacher and the correlations in the rows of the weights, a.k.a. patterns. In the absence of correlations, we validate the conjecture that the performance is independent of the number of teacher patters and hidden units of the student RBMs, and we argue that the teacher-student setting can be used as a toy model for studying the lottery ticket hypothesis. Beyond this regime, we find that the critical amount of data required to learn the teacher patterns decreases with both their number and correlations. In both regimes, we find that, even with a relatively large dataset, it becomes impossible to learn the teacher patterns if the inference temperature used for regularization is kept too low. In our framework, the student can learn teacher patterns one-to-one or many-to-one, generalizing previous findings about the teacher-student setting with two hidden units to any arbitrary finite number of hidden units.
format Preprint
id arxiv_https___arxiv_org_abs_2410_16150
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Modeling Structured Data Learning with Restricted Boltzmann Machines in the Teacher-Student Setting
Thériault, Robin
Tosello, Francesco
Tantari, Daniele
Machine Learning
Disordered Systems and Neural Networks
Restricted Boltzmann machines (RBM) are generative models capable to learn data with a rich underlying structure. We study the teacher-student setting where a student RBM learns structured data generated by a teacher RBM. The amount of structure in the data is controlled by adjusting the number of hidden units of the teacher and the correlations in the rows of the weights, a.k.a. patterns. In the absence of correlations, we validate the conjecture that the performance is independent of the number of teacher patters and hidden units of the student RBMs, and we argue that the teacher-student setting can be used as a toy model for studying the lottery ticket hypothesis. Beyond this regime, we find that the critical amount of data required to learn the teacher patterns decreases with both their number and correlations. In both regimes, we find that, even with a relatively large dataset, it becomes impossible to learn the teacher patterns if the inference temperature used for regularization is kept too low. In our framework, the student can learn teacher patterns one-to-one or many-to-one, generalizing previous findings about the teacher-student setting with two hidden units to any arbitrary finite number of hidden units.
title Modeling Structured Data Learning with Restricted Boltzmann Machines in the Teacher-Student Setting
topic Machine Learning
Disordered Systems and Neural Networks
url https://arxiv.org/abs/2410.16150