Skip to Main content Skip to Navigation
Conference papers

An Adjusted Nearest Neighbor Algorithm Maximizing the F-Measure from Imbalanced Data

Abstract : In this paper, we address the challenging problem of learning from imbalanced data using a Nearest-Neighbor (NN) algorithm. In this setting, the minority examples typically belong to the class of interest requiring the optimization of specific criteria, like the F-Measure. Based on simple geometrical ideas, we introduce an algorithm that reweights the distance between a query sample and any positive training example. This leads to a modification of the Voronoi regions and thus of the decision boundaries of the NN algorithm. We provide a theoretical justification about the weighting scheme needed to reduce the False Negative rate while controlling the number of False Positives. We perform an extensive experimental study on many public imbalanced datasets, but also on large scale non public data from the French Ministry of Economy and Finance on a tax fraud detection task, showing that our method is very effective and, interestingly, yields the best performance when combined with state of the art sampling methods.
Document type :
Conference papers
Complete list of metadata

Cited literature [29 references]  Display  Hide  Download

https://hal.archives-ouvertes.fr/hal-02318868
Contributor : Guillaume Metzler Connect in order to contact the contributor
Submitted on : Thursday, October 17, 2019 - 2:54:12 PM
Last modification on : Sunday, November 29, 2020 - 10:48:02 AM
Long-term archiving on: : Saturday, January 18, 2020 - 2:12:32 PM

File

ICTAI2019.pdf
Files produced by the author(s)

Identifiers

  • HAL Id : hal-02318868, version 1

Collections

Citation

Rémi Viola, Rémi Emonet, Amaury Habrard, Guillaume Metzler, Sébastien Riou, et al.. An Adjusted Nearest Neighbor Algorithm Maximizing the F-Measure from Imbalanced Data. International Conference on Tools with Artificial Intelligence (ICTAI), Nov 2019, Portland, Oregon, United States. ⟨hal-02318868⟩

Share

Metrics

Record views

98

Files downloads

321