PK-GCN：基于图卷积网络的先验知识辅助图像分类
计算机视觉与模式识别
2020-09-28 v1 机器学习
图像与视频处理
摘要
深度学习已在各种分类任务中取得巨大成功。典型地，深度学习模型直接从数据学习潜在特征，而不包含类之间的潜在关系。类之间的相似性会影响分类性能。本文中，我们提出一种方法，使用图卷积层将类相似性知识融入卷积神经网络模型。我们在两个基准图像数据集 MNIST 与 CIFAR10 上评估我们的方法，并分析不同数据与模型规模下的结果。实验结果表明，我们的模型可提升分类准确率，尤其在可用数据量较小时。
引用
@article{arxiv.2009.11892,
title = {PK-GCN: Prior Knowledge Assisted Image Classification using Graph Convolution Networks},
author = {Xueli Xiao and Chunyan Ji and Thosini Bamunu Mudiyanselage and Yi Pan},
journal= {arXiv preprint arXiv:2009.11892},
year = {2020}
}