Joint Feature and Differentiable $ k $-NN Graph Learning using Dirichlet Energy: Paper and Code

Get our free extension to see links to code for papers anywhere online!Free add-on: code for papers everywhere!Free add-on: See code for papers anywhere!

Add to Chrome

Add to Firefox

Add to Edge

Joint Feature and Differentiable $ k $-NN Graph Learning using Dirichlet Energy

May 21, 2023
Lei Xu, Lei Chen, Rong Wang, Feiping Nie, Xuelong Li

Figure 1 for Joint Feature and Differentiable $ k $-NN Graph Learning using Dirichlet Energy

Figure 2 for Joint Feature and Differentiable $ k $-NN Graph Learning using Dirichlet Energy

Figure 3 for Joint Feature and Differentiable $ k $-NN Graph Learning using Dirichlet Energy

Figure 4 for Joint Feature and Differentiable $ k $-NN Graph Learning using Dirichlet Energy

Share this with someone who'll enjoy it:

Feature selection (FS) plays an important role in machine learning, which extracts important features and accelerates the learning process. In this paper, we propose a deep FS method that simultaneously conducts feature selection and differentiable $ k $-NN graph learning based on the Dirichlet Energy. The Dirichlet Energy identifies important features by measuring their smoothness on the graph structure, and facilitates the learning of a new graph that reflects the inherent structure in the new feature subspace during the training process using selected features. We employ the Gumbel Softmax technique and the Optimal Transport theory to address the non-differentiability issues of learning discrete FS results and learning $ k $-NN graphs in neural networks, which theoretically makes our model applicable to other graph neural networks. Furthermore, the proposed framework is interpretable, since all modules are designed algorithmically. We validate the effectiveness of our model with extensive experiments on both synthetic and real-world datasets.

View paper on

Share this with someone who'll enjoy it: