Eleonora Giunchiglia, Thomas Lukasiewicz
Hierarchical multi-label classification (HMC) is a challenging classification task extending standard multi-label classification problems by imposing a hierarchy constraint on the classes. In this paper, we propose C-HMCNN(h), a novel approach for HMC problems, which, given a network h for the underlying multi-label classification problem, exploits the hierarchy information in order to produce predictions coherent with the constraint and improve performance. We conduct an extensive experimental analysis showing the superior performance of C-HMCNN(h) when compared to state-of-the-art models.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Multi-Label Classification | Derisi Funcat | AU(PRC) | 0.195 | C-HMCNN |
| Multi-Label Classification | Spo Funcat | AU(PRC) | 0.215 | C-HMCNN |
| Multi-Label Classification | Cellcycle Funcat | AU(PRC) | 0.255 | C-HMCNN |
| Multi-Label Classification | Expr Funcat | AU(PRC) | 0.302 | C-HMCNN |
| Multi-Label Classification | Seq Funcat | AU(PRC) | 0.292 | C-HMCNN |
| Multi-Label Classification | Gasch1 Funcat | AU(PRC) | 0.286 | C-HMCNN |
| Multi-Label Classification | Gasch2 Funcat | AU(PRC) | 0.258 | C-HMCNN |
| Multi-Label Classification | Expr GO | AU(PRC) | 0.447 | C-HMCNN |
| Multi-Label Classification | Eisen Funcat | AU(PRC) | 0.306 | C-HMCNN |
| Multi-Label Classification | Spo GO | AU(PRC) | 0.382 | C-HMCNN |
| Multi-Label Classification | Eisen GO | AU(PRC) | 0.455 | C-HMCNN |
| Multi-Label Classification | Gasch1 GO | AU(PRC) | 0.436 | C-HMCNN |
| Multi-Label Classification | Cellcycle GO | AU(PRC) | 0.413 | C-HMCNN |
| Multi-Label Classification | Gasch2 GO | AU(PRC) | 0.414 | C-HMCNN |
| Multi-Label Classification | Derisi GO | AU(PRC) | 0.37 | C-HMCNN |
| Multi-Label Classification | Seq GO | AU(PRC) | 0.446 | C-HMCNN |