arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.03047cs.LGcs.CVstat.ML
收藏

Leveraging Class Hierarchies with Metric-Guided Prototype Learning

LASTIG, ENSG, IGN(法国国家地理信息与林业信息研究所,国立地理科学学院,国家地理研究所);Univ Gustave Eiffel(古斯塔夫·埃菲尔大学)

Vivien Sainte Fare Garnot, Loic Landrieu

英文摘要:

In many classification tasks, the set of target classes can be organized into a hierarchy. This structure induces a semantic distance between classes, and can be summarised under the form of a cost matrix, which defines a finite metric on the class set. In this paper, we propose to model the hierarchical class structure by integrating this metric in the supervision of a prototypical network. Our method relies on jointly learning a feature-extracting network and a set of class prototypes whose relative arrangement in the embedding space follows an hierarchical metric. We show that this approach allows for a consistent improvement of the error rate weighted by the cost matrix when compared to traditional methods and other prototype-based strategies. Furthermore, when the induced metric contains insight on the data structure, our method improves the overall precision as well. Experiments on four different public datasets - from agricultural time series classification to depth image semantic segmentation - validate our approach.

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

↑