Prediction of Research Project Execution using Data Augmentation and Deep Learning

Flores A.; Tito-Chura H.; Zea-Rospigliosi L. · Inteligencia Artificial (2023) · DOI: 10.4114/intartif.vol26iss71pp46-58

Resumen

This paper presents the results of seven deep learning models for prediction of research project execution in graduates from a public university in Peru. The deep learning models implemented are non-hybrid: Deep Neural Networks (DNN), Long Short-Term Memory (LSTM), Gated Recurrent Unit (GRU), Convolutional Neural Networks (CNN) and, hybrid: CNN+GRU, CNN+ LSTM and LSTM+GRU. Since most of the dataset prediction features are of the nominal type (true or false), this paper proposes a simple novel data augmentation technique for this type of features. Taking as inspiration the input data type of a neural network, the proposal data augmentation technique considers nominal features as numeric, and obtain random values close to them to generate synthetic records. The results show that most of deep learning models with data augmentation significantly outperform models with just class balancing in terms of accuracy, precision, f1-score and specificity, being the main improvements of 17.39%, 80.00%, 25.00% and 20.00% respectively. © IBERAMIA and the authors.

Instituciones: Universidad Nacional de Moquegua; Universidad Nacional Jorge Basadre Grohmann