arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2001.05858cs.CVcs.LG

使用STN对齐CNN特征图的问题

The problems with using STNs to align CNN feature maps

KTH Royal Institute of Technology(皇家理工学院)

Lukas Finnveden, Ylva Jansson, Tony Lindeberg

AI总结:

本文指出STN无法对齐变换前后图像的特征图,导致分类准确率下降,并提出在分类与定位网络间共享参数以利用深层复杂特征。

摘要:

空间变换网络(STN)旨在使卷积神经网络(CNN)能够学习对图像变换的不变性。STN最初被提出用于变换CNN特征图以及输入图像。这使得在预测变换参数时能够利用更复杂的特征。然而,由于STN执行的是纯空间变换,在一般情况下,它们不具备将变换后的图像及其原始图像的特征图对齐的能力。我们对此提出了理论论证,并研究了实际影响,表明这种无能伴随着分类准确率的下降。我们主张通过替代性地在分类网络和定位网络之间共享参数,来利用更深层中更复杂的特征。

英文摘要:

Spatial transformer networks (STNs) were designed to enable CNNs to learn invariance to image transformations. STNs were originally proposed to transform CNN feature maps as well as input images. This enables the use of more complex features when predicting transformation parameters. However, since STNs perform a purely spatial transformation, they do not, in the general case, have the ability to align the feature maps of a transformed image and its original. We present a theoretical argument for this and investigate the practical implications, showing that this inability is coupled with decreased classification accuracy. We advocate taking advantage of more complex features in deeper layers by instead sharing parameters between the classification and the localisation network.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑