NiaAML: AutoML for classification and regression pipelines

Iztok Fister, Laurenz A. Farthofer, Luka Pečnik, Andreas Holzinger*

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

Abstract

In this paper we present NiaAML, an AutoML framework that we have developed for creating machine learning pipelines and hyperparameter tuning. The composition of machine learning pipelines is presented as an optimization problem that can be solved using various stochastic, population-based, nature-inspired algorithms. Nature-inspired algorithms are powerful tools for solving real-world optimization problems, especially those that are highly complex, nonlinear, and involve large search spaces where traditional algorithms may struggle. They are applied widely in various fields, including robotics, operations research, and bioinformatics. This paper provides a comprehensive overview of the software architecture, and describes the main tasks of NiaAML, including the automatic composition of classification and regression pipelines. The overview is supported by an practical illustrative example.

Original languageEnglish
Article number101974
JournalSoftwareX
Volume29
DOIs
Publication statusPublished - Feb 2025

Keywords

  • AutoML
  • Classification
  • Nature-inspired algorithms
  • Optimization

ASJC Scopus subject areas

  • Software
  • Computer Science Applications

Cite this