Bioengineering (Mar 2024)

Utilizing Nearest-Neighbor Clustering for Addressing Imbalanced Datasets in Bioengineering

  • Chih-Ming Huang,
  • Chun-Hung Lin,
  • Chuan-Sheng Hung,
  • Wun-Hui Zeng,
  • You-Cheng Zheng,
  • Chih-Min Tsai

DOI
https://doi.org/10.3390/bioengineering11040345
Journal volume & issue
Vol. 11, no. 4
p. 345

Abstract

Read online

Imbalance classification is common in scenarios like fault diagnosis, intrusion detection, and medical diagnosis, where obtaining abnormal data is difficult. This article addresses a one-class problem, implementing and refining the One-Class Nearest-Neighbor (OCNN) algorithm. The original inter-quartile range mechanism is replaced with the K-means with outlier removal (KMOR) algorithm for efficient outlier identification in the target class. Parameters are optimized by treating these outliers as non-target-class samples. A new algorithm, the Location-based Nearest-Neighbor (LBNN) algorithm, clusters one-class training data using KMOR and calculates the farthest distance and percentile for each test data point to determine if it belongs to the target class. Experiments cover parameter studies, validation on eight standard imbalanced datasets from KEEL, and three applications on real medical imbalanced datasets. Results show superior performance in precision, recall, and G-means compared to traditional classification models, making it effective for handling imbalanced data challenges.

Keywords