Global Pooling via EquilibriumAggregation

[rusty1s]

🚀 The feature, motivation and pitch

Support the recently proposed "Equilibrium Aggregation" scheme which defines an implicitly given global readout function. The blog post goes over it in detail.

• Paper: The code is given in the appendix.

Alternatives

No response

Additional context

No response

[Padarn]

This paper is pretty cool. I'm happy to start on this.

I wonder if its worth also trying to implement it as a possible aggregation function in message passing, rather than just as a global readout. That way the global readout version could be implemented just by passing a different adjacency matrix to the layer?

[rusty1s]

I think providing it as a global readout layer is sufficient for now. I don‘t think we need to add it as a general option for aggregation in any message passing layer for now.

[Padarn]

Fair enough, worth doing separately

[Padarn]

Have opened a PR, and assigned this to myself - still a number of open questions to discuss and a lot of cleaning to do.