MLP4ML: Machine Learning Service Recommendation System using MLP

Abstract

In this work, we propose a unique approach for Machine Learning (ML) service recommendation using multilayer perceptron architecture. A service is recommended based on its predicted performance on the input dataset. We take Quality of Services (QoS) as the performance indicator. Depending on the application domain and user requirements, the importance level of different QoS attributes could be different. For ML services, their QoS values are affected by both the input dataset and the service. It would be helpful if we can include their features into the recommendation model. In this work, we consider two types of side information: features of the services and of the user (in our case the dataset given by the user). In the experiment, we take OpenML as our data source and extract QoS values of multiple classification services running on 390 datasets. The result shows that dataset-service interactions can be used to predict the performance of a service on a given dataset. When we integrate all the side information, the performance is better than using the interaction data alone in terms of both prediction and recommendation accuracy.