Utilizing Nearest-Neighbor Clustering for Addressing Imbalanced Datasets in Bioengineering

Bioengineering (Basel). 2024 Mar 31;11(4):345. doi: 10.3390/bioengineering11040345.

Abstract

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: K-means with outlier removal (KMOR); Location-based Nearest Neighbor (LBNN); One-Class Nearest-Neighbor (OCNN).

Grants and funding

This study received funding from the grants of National Science and Technology Council of TAIWAN: NSTC 112-2221-E-182A-005 and NSTC 112-2622-E-110-013.