Skip to content

Latest commit

 

History

History
12 lines (9 loc) · 1.13 KB

File metadata and controls

12 lines (9 loc) · 1.13 KB

Graph-regularized Neural Networks

The repo contains application of my extention of the method from Interpretable Neuron Structuring with Graph Spectral Regularization by Tong, et. al to a few other types of networks (more realistic in my opinion).

The main results are:

  • w2v_regularized.ipynb
    • Enforcing grid structure on the hidden layer of word2vec allows visual interpretation of the word embedding without loss in quality ("Graph-regularized (grid)" section)
    • To get actual clustering structure on the word2vec I had to enforce fixed number of modules into the learned graph using spectral clustering. It allowed to get a nice clustered structure of the hidden layer ("Grap-regularized: 6 clusters spectral" part), though my quick attempt to interpret it weren't succesful.
  • mlp_coactivation_reg_v2.ipynb: trying to add modular structure across several layers for an MLP on FashionMNIST reduced both its performance and its interpretability.