Scientific Reports (May 2018)

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

  • Hidetaka Taniguchi,
  • Hiroshi Sato,
  • Tomohiro Shirakawa

DOI
https://doi.org/10.1038/s41598-018-25679-z
Journal volume & issue
Vol. 8, no. 1
pp. 1 – 13

Abstract

Read online

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.