arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2209.05723cs.CVcs.LG

A Capsule Network for Hierarchical Multi-Label Image Classification

Deakin University(迪肯大学);CSIRO Data 61(联邦科学与工业研究组织数据61所)

Khondaker Tasrif Noor, Antonio Robles-Kelly, Brano Kusy

英文摘要:

Image classification is one of the most important areas in computer vision. Hierarchical multi-label classification applies when a multi-class image classification problem is arranged into smaller ones based upon a hierarchy or taxonomy. Thus, hierarchical classification modes generally provide multiple class predictions on each instance, whereby these are expected to reflect the structure of image classes as related to one another. In this paper, we propose a multi-label capsule network (ML-CapsNet) for hierarchical classification. Our ML-CapsNet predicts multiple image classes based on a hierarchical class-label tree structure. To this end, we present a loss function that takes into account the multi-label predictions of the network. As a result, the training approach for our ML-CapsNet uses a coarse to fine paradigm while maintaining consistency with the structure in the classification levels in the label-hierarchy. We also perform experiments using widely available datasets and compare the model with alternatives elsewhere in the literature. In our experiments, our ML-CapsNet yields a margin of improvement with respect to these alternative methods.

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

↑