LogAvgExp Provides a Principled and Performant Global Pooling Operator
Lowe, Scott C., Trappenberg, Thomas, Oore, Sageev
–arXiv.org Artificial Intelligence
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 $\text{LogAvgExp} := \log(\text{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 \to 0^+$ and $t \to +\infty$). We experimentally tested LogAvgExp, both with and without a learnable temperature parameter, in a variety of deep neural network architectures for computer vision.
arXiv.org Artificial Intelligence
Nov-2-2021
- Country:
- North America > Canada
- Nova Scotia (0.14)
- Ontario (0.14)
- North America > Canada
- Genre:
- Research Report (0.83)
- Technology: