Authors:
Christopher W. Bartlett
1
;
Jamie Bossenbroek
2
;
Yukie Ueyama
1
;
Patricia E. Mccallinhart
1
;
Aaron J. Trask
1
and
William C. Ray
1
Affiliations:
1
The Abigail Wexner Research Institute at Nationwide Children’s Hospital, Columbus, Ohio, U.S.A.
;
2
Department of Computer Science and Engineering, Ohio State University College of Engineering, Columbus, Ohio, U.S.A.
Keyword(s):
Machine Learning, Health, Invasive, Non-invasive, Model, Overfitting.
Abstract:
Early stopping is an extremely common tool to minimize overfitting, which would otherwise be a cause of poor generalization of the model to novel data. However, early stopping is a heuristic that, while effective, primarily relies on ad hoc parameters and metrics. Optimizing when to stop remains a challenge. In this paper, we suggest that for some biomedical applications, a natural dichotomy of invasive/non-invasive measurements of a biological system can be exploited to provide objective advice on early stopping. We discuss the conditions where invasive measurements of a biological process should provide better predictions than non-invasive measurements, or at best offer parity. Hence, if data from an invasive measurement is available locally, or from the literature, that information can be leveraged to know with high certainty whether a model of non-invasive data is overfitted. We present paired invasive/non-invasive cardiac and coronary artery measurements from two mouse strains,
one of which spontaneously develops type 2 diabetes, posed as a classification problem. Examination of the various stopping rules shows that generalization is reduced with more training epochs and commonly applied stopping rules give widely different generalization error estimates. The use of an empirically derived training ceiling is demonstrated to be helpful as added information to leverage early stopping in order to reduce overfitting.
(More)