用于鲁棒图像分类的小批量图
计算机视觉与模式识别
2021-05-10 v1 人工智能
摘要
当前计算机视觉中用于分类任务的深度学习模型使用小批量进行训练。在本文中，我们利用小批量中样本之间的关系，使用图神经网络从相似图像中聚合信息。这有助于减轻输入图像扰动对分类性能的不利影响。在基于图像的目标与场景分类上的多种实验表明，该方法不仅提升了分类器的性能，还增强了对图像扰动和对抗攻击的鲁棒性。此外，我们还表明小批量图神经网络有助于缓解生成对抗网络（GAN）中的模式崩溃问题。
引用
@article{arxiv.2105.03237,
title = {Mini-batch graphs for robust image classification},
author = {Arnab Kumar Mondal and Vineet Jain and Kaleem Siddiqi},
journal= {arXiv preprint arXiv:2105.03237},
year = {2021}
}