A machine learning model with human cognitive biases capable of learning from small and biased datasets

Sci Rep. 2018 May 9;8(1):7397. doi: 10.1038/s41598-018-25679-z.

Abstract

Human learners can generalize a new concept from a small number of samples. In contrast, conventional machine learning methods require large amounts of data to address the same types of problems. Humans have cognitive biases that promote fast learning. Here, we developed a method to reduce the gap between human beings and machines in this type of inference by utilizing cognitive biases. We implemented a human cognitive model into machine learning algorithms and compared their performance with the currently most popular methods, naïve Bayes, support vector machine, neural networks, logistic regression and random forests. We focused on the task of spam classification, which has been studied for a long time in the field of machine learning and often requires a large amount of data to obtain high accuracy. Our models achieved superior performance with small and biased samples in comparison with other representative machine learning methods.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Cognition*
  • Humans
  • Machine Learning*
  • Neural Networks, Computer
  • Support Vector Machine