On minimizing the training set fill distance in machine learning regression
Not on the map. Why?
Abstract (arXiv)
For regression tasks one often leverages large datasets for training predictive machine learning models. However, using large datasets may not be feasible due to computational limitations or high data labelling costs. Therefore, suitably selecting small training sets from large pools of unlabelled data points is essential to maximize model performance while maintaining efficiency. In this work, we study Farthest Point Sampling (FPS), a data selection approach that aims to minimize the fill distance of the selected set. We derive an upper bound for the maximum expected prediction error, conditional to the location of the unlabelled data points, that linearly depends on the training set fill distance. For empirical validation, we perform experiments using two regression models on three datasets. We empirically show that selecting a training set by aiming to minimize the fill distance, thereby minimizing our derived bound, significantly reduces the maximum prediction error of various regression models, outperforming alternative sampling approaches by a large margin. Furthermore, we show that selecting training sets with the FPS can also increase model stability for the specific case of Gaussian kernel regression approaches.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 10 of its 58 references that are in the map. First posted on arXiv 20 Jul 2023.
Builds on
10 of 58 references.
References · 10
- 2017SchNet: A continuous-filter convolutional neural network for modeling quantum interactionscited by 571
- 2018Lipschitz regularity of deep neural networks: analysis and efficient estimationcited by 415
- 2018Regularisation of Neural Networks by Enforcing Lipschitz Continuitycited by 314
- 2020Machine Learning Force Fieldscited by 207
- 2021Spherical Message Passing for 3D Graph Networkscited by 168
- 2015Gaussian Processes for Data-Efficient Learning in Robotics and Controlcited by 98
- 2022GemNet-OC: Developing Graph Neural Networks for Large and Diverse Molecular Simulation Datasetscited by 61
- 2018Active Learning for Regression Using Greedy Samplingcited by 26
- 2020Metrics for Benchmarking and Uncertainty Quantification: Quality, Applicability, and a Path to Best Practices for Machine Learning in Chemistrycited by 9
- 2020Improving Sample and Feature Selection with Principal Covariates Regressioncited by 9
Built on by
Citing papers per year: 2025: 1; 2026: 1.
2 papers.