Towards ML Models’ Recommendations: Towards ML Models’ Recommendations: L. Kallab et al.

Abstract

Artificial Intelligence encompasses a range of technologies that replicate human-like cognitive abilities through computer systems, enabling the execution of tasks associated with intelligent beings. A prominent way to achieve this is machine learning (ML), which optimizes system performance by employing learning algorithms to create models based on data and its inherent patterns. Today, a multitude of ML models exist having diverse characteristics, including the algorithm type, training dataset, and resultant performance. Such diversity complicates the selection of an appropriate model for a specific use case, answering user demands. This paper presents an approach for ML models retrieval based on the matching between user inputs and ML models criteria, all described in a semantic ML ontology named SML model (Semantic Machine Learning model), which facilitates the process of ML models selection. Our approach is based on similarities measures that we tested and experimented to score the ML models and retrieve the ones matching, at best, user inputs.