arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2407.20600cs.CV

类别知识融合识别:通过对齐与深度度量学习将层级知识与图像分类相融合

Categorical Knowledge Fused Recognition: Fusing Hierarchical Knowledge with Image Classification through Aligning and Deep Metric Learning

Zhejiang University(浙江大学);University of Leicester(莱斯特大学)

Yunfeng Zhao, Huiyu Zhou, Fei Wu, Xifeng Wu

AI总结:

针对图像分类模型依赖背景等非目标因素推理的问题,提出融合类别层级知识的深度度量学习方法,通过对齐潜在空间与知识空间距离的三元组损失实现端到端融合,在多个数据集上验证了其提升识别推理能力的有效性。

摘要:

图像分类是计算机视觉的基础任务,也是深度度量学习的重要基准。数十年来,研究人员一直致力于通过深度学习模型提升图像分类准确率,但对识别过程中的推理层面关注较少——也就是说,模型可能是依据背景或其他周围物体而非目标物体做出预测。图像类别的层级知识描述了类间的相似性与差异性,将这类知识与深度学习图像分类模型有效融合,有望提升目标物体识别能力,增强识别的推理层面。本文提出了一种基于深度度量学习的新方法,可将图像类别的先验知识与主流骨干图像分类模型有效融合,并以端到端的方式增强识别的推理层面。现有结合深度度量学习的图像分类方法主要关注采样图像是否属于同一类别,本文提出了一种新的三元组损失函数项,可将模型潜在空间中的距离与知识空间中的距离进行对齐,并将其融入所提方法以促进双模态融合。研究在CIFAR-10、CIFAR-100、Mini-ImageNet和ImageNet-1K数据集上开展了大量实验对所提方法进行评估,结果表明,该方法在弱监督目标定位性能方面,可有效增强图像识别的推理层面。

英文摘要:

Image classification is a fundamental computer vision task and an important baseline for deep metric learning. In decades efforts have been made on enhancing image classification accuracy by using deep learning models while less attention has been paid on the reasoning aspect of the recognition, i.e., predictions could be made because of background or other surrounding objects rather than the target object. Hierarchical knowledge about image categories depicts inter-class similarities or dissimilarities. Effective fusion of such knowledge with deep learning image classification models is promising in improving target object identification and enhancing the reasoning aspect of the recognition. In this paper, we propose a novel deep metric learning based method to effectively fuse prior knowledge about image categories with mainstream backbone image classification models and enhance the reasoning aspect of the recognition in an end-to-end manner. Existing deep metric learning incorporated image classification methods mainly focus on whether sampled images are from the same class. A new triplet loss function term that aligns distances in the model latent space with those in knowledge space is presented and incorporated in the proposed method to facilitate the dual-modality fusion. Extensive experiments on the CIFAR-10, CIFAR-100, Mini-ImageNet, and ImageNet-1K datasets evaluated the proposed method, and results indicate that the proposed method is effective in enhancing the reasoning aspect of image recognition in terms of weakly-supervised object localization performance.

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

↑