TY - JOUR
T1 - NiaAML
T2 - AutoML for classification and regression pipelines
AU - Fister, Iztok
AU - Farthofer, Laurenz A.
AU - Pečnik, Luka
AU - Holzinger, Andreas
N1 - Publisher Copyright:
© 2024 The Authors
PY - 2025/2
Y1 - 2025/2
N2 - 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.
AB - 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.
KW - AutoML
KW - Classification
KW - Nature-inspired algorithms
KW - Optimization
UR - http://www.scopus.com/inward/record.url?scp=85209662857&partnerID=8YFLogxK
U2 - 10.1016/j.softx.2024.101974
DO - 10.1016/j.softx.2024.101974
M3 - Article
AN - SCOPUS:85209662857
SN - 2352-7110
VL - 29
JO - SoftwareX
JF - SoftwareX
M1 - 101974
ER -