Information FOMO: The unhealthy fear of missing out on information. A method for removing misleading data for healthier models
Ethan Pickering, Themistoklis P. Sapsis
TL;DR
This work introduces Information FOMO, a Bayesian experimental-design–inspired sequential data-selection method that identifies information-rich samples while discarding data that harms model stability, addressing sample-wise double descent in fixed datasets. By coupling model and data, constructing a global surrogate output PDF, and using acquisition values $a(\mathbf{x})=w(\mathbf{x})\sigma^2(\mathbf{x})$ with $w(\mathbf{x})=p_{\mathbf{x}}(\mathbf{x})/p_\mu(\mu)$, the method achieves rapid, data-efficient convergence without traditional train/test/validation splits. The approach is demonstrated with Gaussian process regression on a 1D nonlinear map and with ensemble Deep Neural Networks on a high-dimensional dispersive wave model (MMT), showing reduced mean-squared and log-PDF errors and elimination of double descent while using roughly 1/20 of the data. Key contributions include a model-agnostic, data-driven acquisition strategy that dynamically adjusts the training set, the demonstration of convergence through a global output PDF, and practical guidance on shallow DNN ensembles for efficient uncertainty estimation. The results have broad implications for learning from small or fixed datasets and for compressing information in large-scale regimes without sacrificing predictive reliability.
Abstract
Misleading or unnecessary data can have out-sized impacts on the health or accuracy of Machine Learning (ML) models. We present a Bayesian sequential selection method, akin to Bayesian experimental design, that identifies critically important information within a dataset, while ignoring data that is either misleading or brings unnecessary complexity to the surrogate model of choice. Our method improves sample-wise error convergence and eliminates instances where more data leads to worse performance and instabilities of the surrogate model, often termed sample-wise ``double descent''. We find these instabilities are a result of the complexity of the underlying map and linked to extreme events and heavy tails. Our approach has two key features. First, the selection algorithm dynamically couples the chosen model and data. Data is chosen based on its merits towards improving the selected model, rather than being compared strictly against other data. Second, a natural convergence of the method removes the need for dividing the data into training, testing, and validation sets. Instead, the selection metric inherently assesses testing and validation error through global statistics of the model. This ensures that key information is never wasted in testing or validation. The method is applied using both Gaussian process regression and deep neural network surrogate models.
