ExplainabilityTransparencySafetycs.CV
Explainable Knowledge Distillation for On-device Chest X-Ray Classification
Chakkrit Termritthikun, Ayaz Umer, Suwichaya Suwanwimolkul, Feng Xia, Ivan Lee
May 10, 2023
IEEE/ACM Transactions on Computational Biology & Bioinformatics
36 citations
2 influential
Key Takeaways
-
Smaller AI models can match larger ones' performance: The researchers used "knowledge distillation" to transfer expertise from large, complex AI models to smaller ones. Their compact model achieved strong diagnostic accuracy (83.7-88.7% AUC across datasets) while using 4.7 million parameters instead of millions more, making deployment on phones and portable devices feasible.
-
Explainability matters for medical AI: The team combined their efficient model with explainable AI (XAI) techniques to show why the model made specific diagnostic decisions on chest X-rays. This transparency is crucial for clinicians to trust and verify AI recommendations before using them in patient care.
-
Real-world deployment requires practical solutions: High-computational AI models are often impractical in clinical settings with limited hardware. This work addresses a genuine barrier to responsible AI adoption by creating models that work on resource-constrained devices while maintaining diagnostic reliability.
-
Multiple approaches were tested for robustness: The researchers compared different teacher architectures (CNNs and Transformers) to distill knowledge, suggesting their findings aren't dependent on a single model type—a responsible approach to validating that the method generalizes across different AI architectures.