Performance of Regression Models as a Function of Experiment Noise

Gang Li, Jan Zrimec, Boyang Ji, Jun Geng, Johan Larsbrink, Aleksej Zelezniak, Jens Nielsen, Martin Km Engqvist

Research output: Contribution to journalJournal articleResearchpeer-review

11 Downloads (Pure)

Abstract

A challenge in developing machine learning regression models is that it is difficult to know whether maximal performance has been reached on the test dataset, or whether further model improvement is possible. In biology, this problem is particularly pronounced as sample labels (response variables) are typically obtained through experiments and therefore have experiment noise associated with them. Such label noise puts a fundamental limit to the metrics of performance attainable by regression models on the test dataset. We address this challenge by deriving an expected upper bound for the coefficient of determination (R 2) for regression models when tested on the holdout dataset. This upper bound depends only on the noise associated with the response variable in a dataset as well as its variance. The upper bound estimate was validated via Monte Carlo simulations and then used as a tool to bootstrap performance of regression models trained on biological datasets, including protein sequence data, transcriptomic data, and genomic data. The new method for estimating upper bounds for model performance on test data should aid researchers in developing ML regression models that reach their maximum potential. Although we study biological datasets in this work, the new upper bound estimates will hold true for regression models from any research field or application area where response variables have associated noise.
Original languageEnglish
JournalBioinformatics and Biology Insights
Volume15
Number of pages10
ISSN1177-9322
DOIs
Publication statusPublished - 2021

Fingerprint

Dive into the research topics of 'Performance of Regression Models as a Function of Experiment Noise'. Together they form a unique fingerprint.

Cite this