Saved in:
Bibliographic Details
Main Authors: Abbe, Emmanuel, Bengio, Samy, Lotfi, Aryo, Rizk, Kevin
Format: Preprint
Published: 2023
Subjects:
Online Access:https://arxiv.org/abs/2301.13105
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909396702855168
author Abbe, Emmanuel
Bengio, Samy
Lotfi, Aryo
Rizk, Kevin
author_facet Abbe, Emmanuel
Bengio, Samy
Lotfi, Aryo
Rizk, Kevin
contents This paper considers the learning of logical (Boolean) functions with a focus on the generalization on the unseen (GOTU) setting, a strong case of out-of-distribution generalization. This is motivated by the fact that the rich combinatorial nature of data in certain reasoning tasks (e.g., arithmetic/logic) makes representative data sampling challenging, and learning successfully under GOTU gives a first vignette of an 'extrapolating' or 'reasoning' learner. We study how different network architectures trained by (S)GD perform under GOTU and provide both theoretical and experimental evidence that for sparse functions and a class of network models including instances of Transformers, random features models, and linear networks, a min-degree-interpolator is learned on the unseen. More specifically, this means an interpolator of the training data that has minimal Fourier mass on the higher degree basis elements. These findings lead to two implications: (1) we provide an explanation to the length generalization problem for Boolean functions (e.g., Anil et al. 2022); (2) we introduce a curriculum learning algorithm called Degree-Curriculum that learns monomials more efficiently by incrementing supports. Finally, we discuss extensions to other models or non-sparse regimes where the min-degree bias may still occur or fade, as well as how it can be potentially corrected when undesirable.
format Preprint
id arxiv_https___arxiv_org_abs_2301_13105
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Generalization on the Unseen, Logic Reasoning and Degree Curriculum
Abbe, Emmanuel
Bengio, Samy
Lotfi, Aryo
Rizk, Kevin
Machine Learning
This paper considers the learning of logical (Boolean) functions with a focus on the generalization on the unseen (GOTU) setting, a strong case of out-of-distribution generalization. This is motivated by the fact that the rich combinatorial nature of data in certain reasoning tasks (e.g., arithmetic/logic) makes representative data sampling challenging, and learning successfully under GOTU gives a first vignette of an 'extrapolating' or 'reasoning' learner. We study how different network architectures trained by (S)GD perform under GOTU and provide both theoretical and experimental evidence that for sparse functions and a class of network models including instances of Transformers, random features models, and linear networks, a min-degree-interpolator is learned on the unseen. More specifically, this means an interpolator of the training data that has minimal Fourier mass on the higher degree basis elements. These findings lead to two implications: (1) we provide an explanation to the length generalization problem for Boolean functions (e.g., Anil et al. 2022); (2) we introduce a curriculum learning algorithm called Degree-Curriculum that learns monomials more efficiently by incrementing supports. Finally, we discuss extensions to other models or non-sparse regimes where the min-degree bias may still occur or fade, as well as how it can be potentially corrected when undesirable.
title Generalization on the Unseen, Logic Reasoning and Degree Curriculum
topic Machine Learning
url https://arxiv.org/abs/2301.13105