An Improved Multi-label Relief Feature Selection Algorithm for Unbalanced Datasets

Yonghong Xie,Daole Li,Dezheng Zhang,Ha Shuang
DOI: https://doi.org/10.1007/978-3-319-69096-4_21
2017-01-01
Abstract:RelieF algorithm is a series of feature selection methods, including the first proposed RelifF algorithm and later extended ReliefF algorithm. The core idea is to give greater weight to the features which have great contribution to the classification problem. The algorithm is simple and has high efficiency. However, it is not ideal to apply the Relief algorithm directly to unbalanced multi-label datasets. Based on the Relief algorithm, this paper proposes an improved multi-label ReliefF feature selection algorithm for unbalanced datasets, called UBML-ReliefF algorithm, which effectively eliminates the problem of datasets unbalance. Finally, we use the stroke cases to do experiments, and the experimental results show that the multi-label classification effect in UBML-ReliefF algorithm is better than the original ReliefF algorithm.
What problem does this paper attempt to address?