LogAvgExp Provides a Principled and Performant Global Pooling Operator

11/02/2021
by   Scott C. Lowe, et al.
0

We seek to improve the pooling operation in neural networks, by applying a more theoretically justified operator. We demonstrate that LogSumExp provides a natural OR operator for logits. When one corrects for the number of elements inside the pooling operator, this becomes LogAvgExp := log(mean(exp(x))). By introducing a single temperature parameter, LogAvgExp smoothly transitions from the max of its operands to the mean (found at the limiting cases t → 0^+ and t → +∞). We experimentally tested LogAvgExp, both with and without a learnable temperature parameter, in a variety of deep neural network architectures for computer vision.

READ FULL TEXT

Please sign up or login with your details

Forgot password? Click here to reset