Current browse context:
cs.LG
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: Leveraging Class Hierarchies with Metric-Guided Prototype Learning
(Submitted on 6 Jul 2020 (v1), last revised 29 Nov 2021 (this version, v3))
Abstract: 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.
Submission history
From: Vivien Sainte Fare Garnot [view email][v1] Mon, 6 Jul 2020 20:22:08 GMT (2942kb,D)
[v2] Fri, 2 Oct 2020 15:08:46 GMT (17064kb,D)
[v3] Mon, 29 Nov 2021 14:07:06 GMT (17939kb,D)
Link back to: arXiv, form interface, contact.