Hierarchical skin lesion image classification with prototypical decision tree.

NPJ Digit Med

AIM for Health Lab, Faculty of Information Technology, Monash University, Melbourne, VIC, Australia.

Published: January 2025

Traditional disease classification models often disregard the clinical significance of misclassifications and lack interpretability. To overcome these challenges, we propose a hierarchical prototypical decision tree (HPDT) for skin lesion classification. HPDT combines prototypical networks and decision trees, leveraging a class hierarchy to guide interpretable predictions from general to specific categories. By incorporating a hierarchy-based distance matrix, the model prioritizes less severe misclassifications while maintaining diagnostic accuracy. Evaluated on a dataset of 235,268 dermoscopic images across 65 conditions, HPDT outperforms flat classifiers and existing hierarchical methods in accuracy, error severity reduction, and interpretability. It also generalizes effectively to unseen classes. These results highlight the value of integrating clinical hierarchies into model design and training to improve diagnostic reliability and decision transparency, demonstrating HPDT's potential for clinical decision support.

Download full-text PDF

Source
http://www.ncbi.nlm.nih.gov/pmc/articles/PMC11729859PMC
http://dx.doi.org/10.1038/s41746-024-01395-zDOI Listing

Publication Analysis

Top Keywords

skin lesion
8
prototypical decision
8
decision tree
8
decision
5
hierarchical skin
4
lesion image
4
image classification
4
classification prototypical
4
tree traditional
4
traditional disease
4

Similar Publications

Want AI Summaries of new PubMed Abstracts delivered to your In-box?

Enter search terms and have AI summaries delivered each week - change queries or unsubscribe any time!