PHOTONAI-A Python API for rapid machine learning model development.

Journal: PloS one
Published Date:

Abstract

PHOTONAI is a high-level Python API designed to simplify and accelerate machine learning model development. It functions as a unifying framework allowing the user to easily access and combine algorithms from different toolboxes into custom algorithm sequences. It is especially designed to support the iterative model development process and automates the repetitive training, hyperparameter optimization and evaluation tasks. Importantly, the workflow ensures unbiased performance estimates while still allowing the user to fully customize the machine learning analysis. PHOTONAI extends existing solutions with a novel pipeline implementation supporting more complex data streams, feature combinations, and algorithm selection. Metrics and results can be conveniently visualized using the PHOTONAI Explorer and predictive models are shareable in a standardized format for further external validation or application. A growing add-on ecosystem allows researchers to offer data modality specific algorithms to the community and enhance machine learning in the areas of the life sciences. Its practical utility is demonstrated on an exemplary medical machine learning problem, achieving a state-of-the-art solution in few lines of code. Source code is publicly available on Github, while examples and documentation can be found at www.photon-ai.com.

Authors

  • Ramona Leenings
    Institute for Translational Psychiatry, University of Münster, Münster, Germany.
  • Nils Ralf Winter
    Universitätsklinikum Münster Klinik für Psychiatrie und Psychotherapie.
  • Lucas Plagwitz
    Institute for Translational Psychiatry, University of Münster, Münster, Germany.
  • Vincent Holstein
    Institute for Translational Psychiatry, University of Münster, Münster, Germany.
  • Jan Ernsting
    Institute for Translational Psychiatry, University of Münster, Münster, Germany.
  • Kelvin Sarink
    Institute for Translational Psychiatry, University of Münster, Münster, Germany.
  • Lukas Fisch
    Institute for Translational Psychiatry, University of Münster, Münster, Germany.
  • Jakob Steenweg
    Institute for Translational Psychiatry, University of Münster, Münster, Germany.
  • Leon Kleine-Vennekate
    Institute for Translational Psychiatry, University of Münster, Münster, Germany.
  • Julian Gebker
    Institute for Translational Psychiatry, University of Münster, Münster, Germany.
  • Daniel Emden
    Institute for Translational Psychiatry, University of Münster, Münster, Germany.
  • Dominik Grotegerd
    Department of Psychiatry and Psychotherapy, University of Münster, Germany.
  • Nils Opel
    Department of Psychiatry, University of Muenster, Muenster, Germany.
  • Benjamin Risse
    Faculty of Mathematics and Computer Science, University of Münster, Münster, Germany.
  • Xiaoyi Jiang
    Faculty of Mathematics and Computer Science, University of Münster, Münster, Germany.
  • Udo Dannlowski
    Department of Psychiatry and Psychotherapy, University of Münster, Germany.
  • Tim Hahn