Automatic Input Feature Relevance via Spectral Neural Networks

Lorenzo Chicchi,Lorenzo Buffoni,Diego Febbe,Lorenzo Giambagli,Raffaele Marino,Duccio Fanelli
2024-06-03
Abstract:Working with high-dimensional data is a common practice, in the field of machine learning. Identifying relevant input features is thus crucial, so as to obtain compact dataset more prone for effective numerical handling. Further, by isolating pivotal elements that form the basis of decision making, one can contribute to elaborate on - ex post - models' interpretability, so far rather elusive. Here, we propose a novel method to estimate the relative importance of the input components for a Deep Neural Network. This is achieved by leveraging on a spectral re-parametrization of the optimization process. Eigenvalues associated to input nodes provide in fact a robust proxy to gauge the relevance of the supplied entry features. Unlike existing techniques, the spectral features ranking is carried out automatically, as a byproduct of the network training. The technique is successfully challenged against both synthetic and real data.
Machine Learning,Disordered Systems and Neural Networks,Statistical Mechanics,Artificial Intelligence
What problem does this paper attempt to address?