Efficient approximate top-k mutual information based feature selection
Efficient approximate top-k mutual information based feature selection
复制标题
DOI:
10.1007/s10844-022-00750-4
复制
发表时间:
2022-10
影响因子:
3.4
通讯作者:
Md. Abdus Salam;Senjuti Basu Roy;Gautam Das
中科院分区:
文献类型:
--
作者:
Md. Abdus Salam;Senjuti Basu Roy;Gautam Das
Feature selection is an important step in the data science pipeline, and it is critical to develop efficient algorithms for this step.Mutual Information(MI) is one of the important measures used for feature selection, where attributes are sorted according to descending score of MI, and top-k attributes are retained. The goal of this work is to develop a new measureAttribute Average Conflictto effectively approximate top-kattributes, without actually calculating MI. Our proposed method is based on using the database concept ofapproximate functional dependencyto quantify MI rank of attributes which to our knowledge has not been studied before. We demonstrate the effectiveness of our proposed measure with a Monte-Carlo simulation. We also perform extensive experiments using high dimensional synthetic and real datasets with millions of records. Our results show that our proposed method demonstratesperfectaccuracy in selecting the top-kattributes, yet is significantly more efficient than state-of-art baselines, including exact methods for computing Mutual Information based feature selection, as well as adaptive random- sampling based approaches. We also investigate the upper and lower bounds of the proposed new measure and show that tighter bounds can be derived by using marginal frequency of attributes in specific arrangements. The bounds on the proposed measure can be used to select top-kattributes without full scan of the dataset in a single pass. We perform experimental evaluation on real datasets to show the accuracy and effectiveness of this approach.