arXiv · 2312.01991
Shapley-Based Data Valuation with Mutual Information: A Key to Modified K-Nearest Neighbors
Abstract
The K-Nearest Neighbors (KNN) algorithm is widely used for classification and regression; however, it suffers from limitations, including the equal treatment of all samples. We propose Information-Modified KNN (IM-KNN), a novel approach that leverages Mutual Information ($I$) and Shapley values to assign weighted values to neighbors, thereby bridging the gap in treating all samples with the same value and weight. On average, IM-KNN improves the accuracy, precision, and recall of traditional KNN by 16.80%, 17.08%, and 16.98%, respectively, across 12 benchmark datasets. Experiments on four large-scale datasets further highlight IM-KNN's robustness to noise, imbalanced data, and skewed distributions.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Mohammad Ali Vahedifar, Azim Akhtarshenas, Mohammad Mohammadi Rafatpanah, Maryam Sabbaghian. 2023-12-04. Shapley-Based Data Valuation with Mutual Information: A Key to Modified K-Nearest Neighbors. https://arxiv.org/abs/2312.01991
Cite the original work for its findings. Save a collection to share your selection of sources.