LogAvgExp Provides a Principled and Performant Global Pooling Operator
Machine Learning
2021-11-03 v1 Artificial Intelligence
Computer Vision and Pattern Recognition
Abstract
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 . By introducing a single temperature parameter, LogAvgExp smoothly transitions from the max of its operands to the mean (found at the limiting cases and ). We experimentally tested LogAvgExp, both with and without a learnable temperature parameter, in a variety of deep neural network architectures for computer vision.
Cite
@article{arxiv.2111.01742,
title = {LogAvgExp Provides a Principled and Performant Global Pooling Operator},
author = {Scott C. Lowe and Thomas Trappenberg and Sageev Oore},
journal= {arXiv preprint arXiv:2111.01742},
year = {2021}
}