2 code implementations • 13 Mar 2024 • Rikab Gambhir, Athis Osathapan, Jesse Thaler
Moment Pooling generalizes the summation in Deep Sets to arbitrary multivariate moments, which enables the model to achieve a much higher effective latent dimensionality for a fixed latent dimension.