【24h】

Set Aggregation Network as a Trainable Pooling Layer

机译:将聚合网络设置为可训练的池层

获取原文

摘要

Global pooling, such as max- or sum-pooling, is one of the key ingredients in deep neural networks used for processing images, texts, graphs and other types of structured data. Based on the recent DeepSets architecture proposed by Zaheer et al. (NIPS 2017), we introduce a Set Aggregation Network (SAN) as an alternative global pooling layer. In contrast to typical pooling operators, SAN allows to embed a given set of features to a vector representation of arbitrary size. We show that by adjusting the size of embedding, SAN is capable of preserving the whole information from the input. In experiments, we demonstrate that replacing global pooling layer by SAN leads to the improvement of classification accuracy. Moreover, it is less prone to overfitting and can be used as a regularizer.
机译:全局池(例如最大池或总池)是深度神经网络中用于处理图像,文本,图形和其他类型的结构化数据的关键要素之一。基于Zaheer等人最近提出的DeepSets体系结构。 (NIPS 2017),我们引入了集合聚合网络(SAN)作为替代的全局池化层。与典型的池化运算符相比,SAN允许将给定的功能集嵌入到任意大小的矢量表示中。我们表明,通过调整嵌入的大小,SAN能够保留来自输入的全部信息。在实验中,我们证明了用SAN替换全局池层可以提高分类准确性。此外,它不太容易过度拟合,可以用作正则化器。

著录项

相似文献

  • 外文文献
  • 中文文献
  • 专利
获取原文

客服邮箱:kefu@zhangqiaokeyan.com

京公网安备:11010802029741号 ICP备案号:京ICP备15016152号-6 六维联合信息科技 (北京) 有限公司©版权所有
  • 客服微信

  • 服务号