Skip to content

maschimax/HPO-benchmarking-framework

Repository files navigation

HPO benchmarking framework

Overview

  • Software framework for benchmarking hyperparameter optimization (HPO) techniques for machine learning (ML) use cases in the production domain
  • The framework can be used to apply HPO techniques to user-defined use cases in production and benchmark their performance comparably
  • For each experiment, the framework provides several performance metrics, learning curves, and log-files
  • It is possible to extend the framework by further HPO techniques, ML techniques, and data sets

Usage

  1. Run the script run_benchmark.py to start a benchmarking experiment (specify positional and optional arguments via the command line to define the use case)
  2. The framework conducts the benchmarking experiments according to the setup and generates performance metrics, log-files, and learning curves
  3. Optional: Apply further scripts from ./analysis for a detailed performance analysis

Benchmarking process:

benchmarking framework

Integrated libraries and data sets

HPO libraries and techniques

  • HpBandSter: BOHB and Hyperband
  • Optuna: random search, TPE and CMA-ES
  • Scikit-Optimize: GPBO and SMAC
  • RoBO: FABOLAS and BOHAMIANN

ML libraries and techniques

  • Scikit-Learn: multilayer perceptron, random forest, support vector machine, AdaBoost, decision tree, linear regression, k-nearest neighbor, logistic regression, naive Bayes, and elastic net
  • Keras: multilayer perceptron
  • XGBoost: XGBRegressor and XGBClassifier
  • LightGBM: LGBRegressor and LGBClassifier

Data sets

  • Predictive maintenance
    • Air pressure system failures in Scania trucks (Scania)
    • Turbofan engine degradation simulation (Turbofan)
  • Predictive quality
    • Surface crack detection (Surface)
    • Sensor system production (Sensor)
  • Process parameter prediction
    • Blade integrated disk manufacturing (Blisk)

About

Framework for benchmarking Hyperparameter Tuning techniques for Machine Learning use cases in manufacturing

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages