HyDI: A hybrid Deep Learning-Inductive Logic Programming ensemble for multi-label classification
Organizations: Institute for Computer Science, University of Osnabrück, Neuer Graben 29, 49074 Osnabrück, Germany
Abstract
While attaining remarkable results for many applications, Deep Learning models are notoriously difficult to explain. This work introduces HyDI, a hybrid ensemble architecture for hierarchical multi-label classification. It combines a Deep Learning (DL) model with rule-based classifiers generated by Inductive Logic Programming (ILP). For leaf classes of the label hierarchy, the rule-based classifiers replace the DL model, leading to more transparent classification results. HyDI is applied to the Chemical Entities of Biological Interest (ChEBI) ontology, providing ILP-generated rules for 314 classes. For these classes, HyDI can generate global explanations as well as local explanations that combine visual and text-based descriptions.
Figures & tables
| Predicate | Description |
|---|---|
| chebi_{class_id}/1 | molecule belongs to ChEBI class |
| has_atom/2 | atom belongs to molecule |
| c/1 , n/1 , … | element symbols |
| charge_p/1 , charge_n/1 | positive / negative atom charge |
| charge0/1 , charge_m1/1 , charge1/1 , … | specific atom charge |
| has_0_hs/1 , has_1_hs/1 , … | atom has this number of attached hydrogen atoms |
| All classes with rules | Shared classes | ||||||
|---|---|---|---|---|---|---|---|
| ILP configuration | #rules | Precision | Recall | F1 score | Precision | Recall | F1 score |
| ILP-FN | 666 | 0.677 | 0.809 | 0.680 | 0.712 | 0.877 | 0.730 |
| ILP-FP | 520 | 0.732 | 0.666 | 0.640 | 0.734 | 0.667 | 0.642 |
| ILP-REG | 636 | 0.699 | 0.774 | 0.681 | 0.731 | 0.814 | 0.716 |
| Micro-aggregated | Macro-aggregated | |||||
|---|---|---|---|---|---|---|
| Model | Precision | Recall | F1 score | Precision | Recall | F1 score |
| DL | 0.929 | 0.910 | 0.920 | 0.784 | 0.734 | 0.737 |
| HyDI | 0.919 | 0.910 | 0.914 | 0.782 | 0.729 | 0.731 |
| Precision | Recall | F1 score | |||||
|---|---|---|---|---|---|---|---|
| Class subset | # classes | DL | HyDI | DL | HyDI | DL | HyDI |
| DL | 1,494 | 0.794 | 0.794 | 0.741 | 0.741 | 0.746 | 0.746 |
| ILP-FN | 260 | 0.730 | 0.735 | 0.712 | 0.700 | 0.698 | 0.680 |
| ILP-REG | 24 | 0.766 | 0.558 | 0.640 | 0.524 | 0.674 | 0.481 |
| ILP-FP | 30 | 0.738 | 0.732 | 0.664 | 0.602 | 0.668 | 0.613 |
Multi-Label Classification
Multi-label classification assigns several potentially related tags to each example, from medical findings and topics to objects and user interests.
See all 132 papers