Sep, 2023

可微的算术分布模块实现的仿射变换不变的图像分类

TL;DR本研究通过结合分布学习技术设计了一个更稳健的模块,重点关注图像像素的空间分布信息,以解决卷积神经网络在旋转、平移、翻转和混洗等仿射变换下容易受到影响的问题。通过采用核密度估计构建可微分的直方图,提出了一种新颖的可微分算术分布模块 (DADM),能够从图像中提取内在的概率分布。该方法能够增强模型对仿射变换的鲁棒性,同时保持其特征提取能力,从而弥合传统卷积神经网络与基于分布学习的方法之间的差距。通过消融实验和与 LeNet 的对比实验证明了所提方法的有效性。