Effect of clustering data in improving machine learning model accuracy

Samih M. Mostafa, Hirofumi Amano

    Research output: Contribution to journalArticlepeer-review

    2 Citations (Scopus)

    Abstract

    Supervised machine learning algorithms consider the relationship between dependent and independent variables rather than the relationship between the instances. Machine learning algorithms try to learn the relationship between the input and output from the historical data in order to attain precise predictions about unseen future. Conventional foretelling algorithms are usually based on a model learned and trained from historical data. The instances in the historical data may vary in its characteristics. The variation may be a result of difference in case's pertinence degree to some cases compared to others. However, the problem with such machine learning algorithms is their dealing with the whole data without considering this variation. This paper presents a novel technique to the trained model to improve the prediction accuracy. The proposed method clusters the data using K-means clustering algorithm, and then applies the prediction algorithm to every cluster. The value of K which gives the highest accuracy is selected. The authors performed comparative study of the proposed technique and popular prediction methods namely Linear Regression, Ridge, Lasso, and Elastic. On analysing on five datasets with different sizes and different number of clusters, it was observed that the accuracy of the proposed technique is better from the point of view of Root Mean Square Error (RMSE), and coefficient of determination (Rz).

    Original languageEnglish
    Pages (from-to)2973-2981
    Number of pages9
    JournalJournal of Theoretical and Applied Information Technology
    Volume97
    Issue number21
    Publication statusPublished - Nov 15 2019

    All Science Journal Classification (ASJC) codes

    • Theoretical Computer Science
    • Computer Science(all)

    Fingerprint Dive into the research topics of 'Effect of clustering data in improving machine learning model accuracy'. Together they form a unique fingerprint.

    Cite this