Skip to Main content Skip to Navigation
New interface
Conference papers

GH-CNN: A new CNN for coherent hierarchical classification

Abstract : Hierarchical multi-label classification is a challenging task implying the encoding of a high level constraint in the neural network model. Before the rise of this field, the classification was done without paying attention to the hierarchical links existing between data. Nevertheless, information relating the classes and subclasses may be very useful for improving the network performances. Recently, some works have integrated the hierarchy information by proposing new neural network architectures (called B-CNN or H-CNN), achieving promising results. However with these architectures, the network is separated into blocks where each block is responsible for predicting only the classes of a given level in the hierarchy. In this paper, we propose a novel architecture such that the whole network layers are involved in the prediction of the entire labels of a sample, i.e., from its class in the top level of the hierarchy to its class in the bottom level. The proposed solution is based on a Bayesian adjustment encoding the hierarchy in terms of conditional probabilities, together with a customized semantic loss function that penalizes drastically the hierarchy violation. A teacher forcing strategy learning is used to enhance the learning quality. Thanks to this approach, we could outperform the state of the art results in terms of accuracy (improved for all levels) and also in terms of hierarchy coherence.
Document type :
Conference papers
Complete list of metadata

https://hal.archives-ouvertes.fr/hal-03768304
Contributor : Florence Dupin de Saint-Cyr Connect in order to contact the contributor
Submitted on : Sunday, September 18, 2022 - 3:44:33 PM
Last modification on : Tuesday, September 20, 2022 - 4:12:09 AM

File

ICANN_Final.pdf
Files produced by the author(s)

Identifiers

  • HAL Id : hal-03768304, version 1

Citation

Mouna Sabrine Mayouf, Florence Dupin de Saint Cyr - Bannay. GH-CNN: A new CNN for coherent hierarchical classification. 31st International Conference on Artificial Neural Networks (ICANN 2022), European Neural Network Society; University of the West of England, Bristol, UK, Sep 2022, Bristol, United Kingdom. à paraître. ⟨hal-03768304⟩

Share

Metrics

Record views

17

Files downloads

11