Implement Stochastic Pooling Neural Network

bayesmech / research-gnn-pooling

Research on Heirarchical pooling for Graph Neural Networks

MIT License

0 stars 0 forks source link

Implement Stochastic Pooling Neural Network #7

Closed AnimeshSinha1309 closed 2 years ago

AnimeshSinha1309 commented 2 years ago

Make a layer that is equivalent to diff-pool, but works stochastically using Gumbel softmax. This should always reduce the number of edges in the graph, ensuring our point of memory efficiency while paying in time, but not too much in accuracy.