arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2001.09654cs.LGcs.ITmath.ITstat.ML

Feature selection in machine learning: Rényi min-entropy vs Shannon entropy

Inria(法国国家信息与自动化研究所);École Polytechnique(巴黎综合理工学院);Università di Siena(锡耶纳大学)

Catuscia Palamidessi, Marco Romanelli

英文摘要:

Feature selection, in the context of machine learning, is the process of separating the highly predictive feature from those that might be irrelevant or redundant. Information theory has been recognized as a useful concept for this task, as the prediction power stems from the correlation, i.e., the mutual information, between features and labels. Many algorithms for feature selection in the literature have adopted the Shannon-entropy-based mutual information. In this paper, we explore the possibility of using Rényi min-entropy instead. In particular, we propose an algorithm based on a notion of conditional Rényi min-entropy that has been recently adopted in the field of security and privacy, and which is strictly related to the Bayes error. We prove that in general the two approaches are incomparable, in the sense that we show that we can construct datasets on which the Rényi-based algorithm performs better than the corresponding Shannon-based one, and datasets on which the situation is reversed. In practice, however, when considering datasets of real data, it seems that the Rényi-based algorithm tends to outperform the other one. We have effectuate several experiments on the BASEHOCK, SEMEION, and GISETTE datasets, and in all of them we have indeed observed that the Rényi-based algorithm gives better results.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑