Cargando…

mlf-core: a framework for deterministic machine learning

MOTIVATION: Machine learning has shown extensive growth in recent years and is now routinely applied to sensitive areas. To allow appropriate verification of predictive models before deployment, models must be deterministic. Solely fixing all random seeds is not sufficient for deterministic machine...

Descripción completa

Detalles Bibliográficos
Autores principales: Heumos, Lukas, Ehmele, Philipp, Kuhn Cuellar, Luis, Menden, Kevin, Miller, Edmund, Lemke, Steffen, Gabernet, Gisela, Nahnsen, Sven
Formato: Online Artículo Texto
Lenguaje:English
Publicado: Oxford University Press 2023
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC10089676/
https://www.ncbi.nlm.nih.gov/pubmed/37004171
http://dx.doi.org/10.1093/bioinformatics/btad164
_version_ 1785022811305148416
author Heumos, Lukas
Ehmele, Philipp
Kuhn Cuellar, Luis
Menden, Kevin
Miller, Edmund
Lemke, Steffen
Gabernet, Gisela
Nahnsen, Sven
author_facet Heumos, Lukas
Ehmele, Philipp
Kuhn Cuellar, Luis
Menden, Kevin
Miller, Edmund
Lemke, Steffen
Gabernet, Gisela
Nahnsen, Sven
author_sort Heumos, Lukas
collection PubMed
description MOTIVATION: Machine learning has shown extensive growth in recent years and is now routinely applied to sensitive areas. To allow appropriate verification of predictive models before deployment, models must be deterministic. Solely fixing all random seeds is not sufficient for deterministic machine learning, as major machine learning libraries default to the usage of nondeterministic algorithms based on atomic operations. RESULTS: Various machine learning libraries released deterministic counterparts to the nondeterministic algorithms. We evaluated the effect of these algorithms on determinism and runtime. Based on these results, we formulated a set of requirements for deterministic machine learning and developed a new software solution, the mlf-core ecosystem, which aids machine learning projects to meet and keep these requirements. We applied mlf-core to develop deterministic models in various biomedical fields including a single-cell autoencoder with TensorFlow, a PyTorch-based U-Net model for liver-tumor segmentation in computed tomography scans, and a liver cancer classifier based on gene expression profiles with XGBoost. AVAILABILITY AND IMPLEMENTATION: The complete data together with the implementations of the mlf-core ecosystem and use case models are available at https://github.com/mlf-core.
format Online
Article
Text
id pubmed-10089676
institution National Center for Biotechnology Information
language English
publishDate 2023
publisher Oxford University Press
record_format MEDLINE/PubMed
spelling pubmed-100896762023-04-12 mlf-core: a framework for deterministic machine learning Heumos, Lukas Ehmele, Philipp Kuhn Cuellar, Luis Menden, Kevin Miller, Edmund Lemke, Steffen Gabernet, Gisela Nahnsen, Sven Bioinformatics Original Paper MOTIVATION: Machine learning has shown extensive growth in recent years and is now routinely applied to sensitive areas. To allow appropriate verification of predictive models before deployment, models must be deterministic. Solely fixing all random seeds is not sufficient for deterministic machine learning, as major machine learning libraries default to the usage of nondeterministic algorithms based on atomic operations. RESULTS: Various machine learning libraries released deterministic counterparts to the nondeterministic algorithms. We evaluated the effect of these algorithms on determinism and runtime. Based on these results, we formulated a set of requirements for deterministic machine learning and developed a new software solution, the mlf-core ecosystem, which aids machine learning projects to meet and keep these requirements. We applied mlf-core to develop deterministic models in various biomedical fields including a single-cell autoencoder with TensorFlow, a PyTorch-based U-Net model for liver-tumor segmentation in computed tomography scans, and a liver cancer classifier based on gene expression profiles with XGBoost. AVAILABILITY AND IMPLEMENTATION: The complete data together with the implementations of the mlf-core ecosystem and use case models are available at https://github.com/mlf-core. Oxford University Press 2023-04-02 /pmc/articles/PMC10089676/ /pubmed/37004171 http://dx.doi.org/10.1093/bioinformatics/btad164 Text en © The Author(s) 2023. Published by Oxford University Press. https://creativecommons.org/licenses/by/4.0/This is an Open Access article distributed under the terms of the Creative Commons Attribution License (https://creativecommons.org/licenses/by/4.0/), which permits unrestricted reuse, distribution, and reproduction in any medium, provided the original work is properly cited.
spellingShingle Original Paper
Heumos, Lukas
Ehmele, Philipp
Kuhn Cuellar, Luis
Menden, Kevin
Miller, Edmund
Lemke, Steffen
Gabernet, Gisela
Nahnsen, Sven
mlf-core: a framework for deterministic machine learning
title mlf-core: a framework for deterministic machine learning
title_full mlf-core: a framework for deterministic machine learning
title_fullStr mlf-core: a framework for deterministic machine learning
title_full_unstemmed mlf-core: a framework for deterministic machine learning
title_short mlf-core: a framework for deterministic machine learning
title_sort mlf-core: a framework for deterministic machine learning
topic Original Paper
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC10089676/
https://www.ncbi.nlm.nih.gov/pubmed/37004171
http://dx.doi.org/10.1093/bioinformatics/btad164
work_keys_str_mv AT heumoslukas mlfcoreaframeworkfordeterministicmachinelearning
AT ehmelephilipp mlfcoreaframeworkfordeterministicmachinelearning
AT kuhncuellarluis mlfcoreaframeworkfordeterministicmachinelearning
AT mendenkevin mlfcoreaframeworkfordeterministicmachinelearning
AT milleredmund mlfcoreaframeworkfordeterministicmachinelearning
AT lemkesteffen mlfcoreaframeworkfordeterministicmachinelearning
AT gabernetgisela mlfcoreaframeworkfordeterministicmachinelearning
AT nahnsensven mlfcoreaframeworkfordeterministicmachinelearning