论文标题
随机特征采样和插值在空间自适应推理上
Spatially Adaptive Inference with Stochastic Feature Sampling and Interpolation
论文作者
论文摘要
在CNN的特征地图中,通常存在相当大的空间冗余,从而导致了许多重复的处理。为了减少这种多余的计算,我们建议仅在稀疏采样的位置计算特征,这些特征是根据激活响应概率地选择的,然后通过有效的插值过程密集地重建特征映射。通过这种采样中的交流方案,我们的网络避免在可以有效插值的空间位置进行计算,同时通过广泛分布的采样对激活预测误差进行稳健。这种基于抽样的方法的技术挑战是,用于表示离散抽样位置的二进制决策变量是不可差异的,这使得它们与反向传播不相容。为了解决这个问题,我们利用基于Gumbel-Softmax分布的重新聚集技巧,通过该技巧,反向传播可以将这些变量迭代到二进制值。实验表明,提出的网络可以节省大量计算,同时保持各种计算机视觉任务的准确性。
In the feature maps of CNNs, there commonly exists considerable spatial redundancy that leads to much repetitive processing. Towards reducing this superfluous computation, we propose to compute features only at sparsely sampled locations, which are probabilistically chosen according to activation responses, and then densely reconstruct the feature map with an efficient interpolation procedure. With this sampling-interpolation scheme, our network avoids expending computation on spatial locations that can be effectively interpolated, while being robust to activation prediction errors through broadly distributed sampling. A technical challenge of this sampling-based approach is that the binary decision variables for representing discrete sampling locations are non-differentiable, making them incompatible with backpropagation. To circumvent this issue, we make use of a reparameterization trick based on the Gumbel-Softmax distribution, with which backpropagation can iterate these variables towards binary values. The presented network is experimentally shown to save substantial computation while maintaining accuracy over a variety of computer vision tasks.