STAT-MECHLGMar 10, 2023

Tradeoff of generalization error in unsupervised learning

arXiv:2303.05718v21 citationsh-index: 22
Originality Incremental advance
AI Analysis

This addresses the fundamental problem of model selection in unsupervised learning for researchers, providing a theoretical framework that is incremental but clarifies an understudied aspect of generalization error.

The paper investigates whether unsupervised learning exhibits a tradeoff in generalization error, proposing a two-component tradeoff between model error and data error, where more complex models reduce model error but increase data error, especially with smaller datasets. This is validated using restricted Boltzmann machines on Ising model and TASEP configurations, showing optimal model complexity increases with data complexity.

Finding the optimal model complexity that minimizes the generalization error (GE) is a key issue of machine learning. For the conventional supervised learning, this task typically involves the bias-variance tradeoff: lowering the bias by making the model more complex entails an increase in the variance. Meanwhile, little has been studied about whether the same tradeoff exists for unsupervised learning. In this study, we propose that unsupervised learning generally exhibits a two-component tradeoff of the GE, namely the model error and the data error -- using a more complex model reduces the model error at the cost of the data error, with the data error playing a more significant role for a smaller training dataset. This is corroborated by training the restricted Boltzmann machine to generate the configurations of the two-dimensional Ising model at a given temperature and the totally asymmetric simple exclusion process with given entry and exit rates. Our results also indicate that the optimal model tends to be more complex when the data to be learned are more complex.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes