Optimal -k nearest neighbours based ensemble for classification and feature selection in chemometrics data

Inzamam ul Haq, Dost Muhammad Khan, Muhammad Hamraz, Nadeem Iqbal, Amjad Ali, Zardad Khan

Research output: Contribution to journalArticlepeer-review

7 Citations (Scopus)

Abstract

There are various machine-learning techniques available for classification and regression tasks. The k-nearest neighbours (k-NN) method is a well-recognized algorithm that is used for both regression and classification problems. It identifies a group of knearest observations to a given test point, reducing the impact of outliers in the training dataset. For regression, the mean value is calculated, while for classification, the majority value is determined. This study proposes a novel ensemble approach that constructs k-NN models using bootstrap samples from the training data and a randomly selected subset of features. Stepwise logistic regression is then applied to the nearest neighbours identified by each k-NN model to estimate the test observations. The final estimation for the test point's response is made through a majority voting approach using the estimates from different k-NN models. The performance of the proposed method is compared to other methods using five benchmark datasets, using Brier score, sensitivity, and accuracy as performance metrics. The results indicate that the proposed ensemble method outperforms the other methods across most of the datasets. Additionally, the proposed ensemble method is used for feature selection and compared with four other feature selection methods using 9 benchmark datasets. The results demonstrate that the proposed method exhibits superior performance compared to the other methods.

Original languageEnglish
Article number104882
JournalChemometrics and Intelligent Laboratory Systems
Volume240
DOIs
Publication statusPublished - Sept 15 2023

Keywords

  • Classification
  • Ensemble learning
  • K-nearest neighbours (k-NN)
  • Non-informative features
  • Random k-NN
  • Stepwise model selection

ASJC Scopus subject areas

  • Analytical Chemistry
  • Software
  • Computer Science Applications
  • Process Chemistry and Technology
  • Spectroscopy

Fingerprint

Dive into the research topics of 'Optimal -k nearest neighbours based ensemble for classification and feature selection in chemometrics data'. Together they form a unique fingerprint.

Cite this