A critical look at the current train/test split in machine learning
Not on the map. Why?
Abstract (arXiv)
The randomized or cross-validated split of training and testing sets has been adopted as the gold standard of machine learning for decades. The establishment of these split protocols are based on two assumptions: (i)-fixing the dataset to be eternally static so we could evaluate different machine learning algorithms or models; (ii)-there is a complete set of annotated data available to researchers or industrial practitioners. However, in this article, we intend to take a closer and critical look at the split protocol itself and point out its weakness and limitation, especially for industrial applications. In many real-world problems, we must acknowledge that there are numerous situations where assumption (ii) does not hold. For instance, for interdisciplinary applications like drug discovery, it often requires real lab experiments to annotate data which poses huge costs in both time and financial considerations. In other words, it can be very difficult or even impossible to satisfy assumption (ii). In this article, we intend to access this problem and reiterate the paradigm of active learning, and investigate its potential on solving problems under unconventional train/test split protocols. We further propose a new adaptive active learning architecture (AAL) which involves an adaptation policy, in comparison with the traditional active learning that only unidirectionally adds data points to the training pool. We primarily justify our points by extensively investigating an interdisciplinary drug-protein binding problem. We additionally evaluate AAL on more conventional machine learning benchmarking datasets like CIFAR-10 to demonstrate the generalizability and efficacy of the new framework.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Builds on 17 of its 40 references that are in the map. First posted on arXiv 8 Jun 2021.
Builds on
17 of 40 references.
References · 17
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,861
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
- 2020FixMatch: Simplifying Semi-Supervised Learning with Consistency and Confidencecited by 2,080
- 2017Learning to Compare: Relation Network for Few-Shot Learningcited by 1,139
- 2017Deep Bayesian Active Learning with Image Datacited by 812
- 2019ReMixMatch: Semi-Supervised Learning with Distribution Alignment and Augmentation Anchoringcited by 349
- 2019Variational Adversarial Active Learningcited by 284
- 2015Semi-Supervised Learning with Ladder Networkscited by 246
- 2018Learning to Propagate Labels: Transductive Propagation Network for Few-shot Learningcited by 208
- 2018DeepDTA: Deep Drug-Target Binding Affinity Predictioncited by 171
- 2018Adversarial Active Learning for Deep Networks: a Margin Based Approachcited by 155
- 2018Task-Agnostic Meta-Learning for Few-shot Learningcited by 151
5 more
- 2017Generative Adversarial Active Learningcited by 69
- 2016Protein-Ligand Scoring with Convolutional Neural Networkscited by 57
- 2019Semantic Redundancies in Image-Classification Datasets: The 10% You Don't Needcited by 43
- 2017Atomic Convolutional Networks for Predicting Protein-Ligand Binding Affinitycited by 21
- 2018RelationNet2: Deep Comparison Columns for Few-Shot Learningcited by 3
Built on by
Citing papers per year: 2024: 3.
3 papers.
- 2024A Philosophical Introduction to Language Models -- Part I: Continuity With Classic Debatescited by 22
- 2024Two-Phase Segmentation Approach for Accurate Left Ventricle Segmentation in Cardiac MRI using Machine Learningcited by 0
- 2024Survival of the Cheapest: Cost-Aware Hardware Adaptation for Adversarial Robustnesscited by 0
- 2024Survival of the Cheapest: Cost-Aware Hardware Adaptation for Adversarial Robustness11 Sep 2024
- 2024Two-Phase Segmentation Approach for Accurate Left Ventricle Segmentation in Cardiac MRI using Machine Learning29 Jul 2024
- 2024A Philosophical Introduction to Language Models -- Part I: Continuity With Classic Debates8 Jan 2024