A Comparison of Machine Learning Methods in a High-Dimensional Classification Problem

Authors

  • Marijana Zekić-Sušac University of Josip Juraj Strossmayer in Osijek, Faculty of Economics, Croatia
  • Sanja Pfeifer University of Josip Juraj Strossmayer in Osijek, Faculty of Economics, Croatia
  • Nataša Šarlija University of Josip Juraj Strossmayer in Osijek, Faculty of Economics, Croatia

Keywords:

machine learning, support vector machines, artificial neural networks, CART classification trees, k-nearest neighbour, large-dimensional data, cross-validation

Abstract

Background: Large-dimensional data modelling often relies on variable reduction methods in the pre-processing and in the post-processing stage. However, such a reduction usually provides less information and yields a lower accuracy of the model. Objectives: The aim of this paper is to assess the high-dimensional classification problem of recognizing entrepreneurial intentions of students by machine learning methods. Methods/Approach: Four methods were tested: artificial neural networks, CART classification trees, support vector machines, and k-nearest neighbour on the same dataset in order to compare their efficiency in the sense of classification accuracy. The performance of each method was compared on ten subsamples in a 10-fold cross-validation procedure in order to assess computing sensitivity and specificity of each model. Results: The artificial neural network model based on multilayer perceptron yielded a higher classification rate than the models produced by other methods. The pairwise t-test showed a statistical significance between the artificial neural network and the k-nearest neighbour model, while the difference among other methods was not statistically significant. Conclusions: Tested machine learning methods are able to learn fast and achieve high classification accuracy. However, further advancement can be assured by testing a few additional methodological refinements in machine learning methods.

Creative Commons License
This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.

Downloads

Published

2014-12-31