Published November 14, 2024 | Version v1
Journal Open

SBNNR: Small-size Bat-optimised KNN Regression

  • 1. ROR icon Universitat Politècnica de Catalunya
  • 2. UPC BarcelonaTECH
  • 3. Universitat Politecnica de Catalunya

Description

Small datasets are frequent in some scientific fields. Such datasets are usually created due to the difficulty or cost of producing laboratory and experimental data. On the other hand, researchers are interested in using machine learning methods to analyze this scale of data. For this reason, in some cases, low-performance, overfitting models are developed for small-scale data. As a result, it appears necessary to develop methods for dealing with this type of data. In this research, we provide a new and innovative framework for regression problems with a small sample size. The base of our proposed method is the K-nearest neighbors (KNN) algorithm. For feature selection, instance selection, and hyperparameter tuning, we use the bat optimization algorithm (BA). Generative Adversarial Networks (GANs) are employed to generate synthetic data, effectively addressing the challenges associated with data sparsity. Concurrently, Deep Neural Networks (DNNs), as a deep learning approach, are utilized for feature extraction from both synthetic and real datasets. This hybrid framework integrates KNN, DNN, and GAN as foundational components and is optimized in multiple aspects (features, instances, and hyperparameters) using BA. The outcomes exhibit an enhancement of up to 5% in the coefficient of determination (R2 score) using the proposed method compared to the standard KNN method optimized through grid search.

Files

futureinternet-16-00422.pdf

Files (565.3 kB)

Name Size Download all
md5:56ed7c3ae5ea3bb99626ea555211e1e0
565.3 kB Preview Download