Neural Greedy Pursuit for Feature SelectionVisa övriga samt affilieringar
2022 (Engelska)Ingår i: 2022 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), Institute of Electrical and Electronics Engineers (IEEE) , 2022Konferensbidrag, Publicerat paper (Refereegranskat)
Abstract [en]
We propose a greedy algorithm to select N important features among P input features for a non-linear prediction problem. The features are selected one by one sequentially, in an iterative loss minimization procedure. We use neural networks as predictors in the algorithm to compute the loss and hence, we refer to our method as neural greedy pursuit (NGP). NGP is efficient in selecting N features when N << P, and it provides a notion of feature importance in a descending order following the sequential selection procedure. We experimentally show that NGP provides better performance than several feature selection methods such as DeepLIFT and Drop-one-out loss. In addition, we experimentally show a phase transition behavior in which perfect selection of all N features without false positives is possible when the training data size exceeds a threshold.
Ort, förlag, år, upplaga, sidor
Institute of Electrical and Electronics Engineers (IEEE) , 2022.
Serie
IEEE International Joint Conference on Neural Networks (IJCNN), ISSN 2161-4393
Nyckelord [en]
Feature selection, Deep learning
Nationell ämneskategori
Datavetenskap (datalogi)
Identifikatorer
URN: urn:nbn:se:kth:diva-323022DOI: 10.1109/IJCNN55064.2022.9892946ISI: 000867070908056Scopus ID: 2-s2.0-85140774694OAI: oai:DiVA.org:kth-323022DiVA, id: diva2:1725934
Konferens
IEEE International Conference on Fuzzy Systems (FUZZ-IEEE) / IEEE World Congress on Computational Intelligence (IEEE WCCI) / International Joint Conference on Neural Networks (IJCNN) / IEEE Congress on Evolutionary Computation (IEEE CEC), JUL 18-23, 2022, Padua, ITALY
Anmärkning
Part of proceedings: ISBN 978-1-7281-8671-9
QC 20230112
2023-01-122023-01-122023-01-12Bibliografiskt granskad