MED-TEX: transfer and explain knowledge with less data from pretrained medical imaging models

Thanh Nguyen-Duc, He Zhao, Jianfei Cai, Dinh Phung

Research output: Chapter in Book/Report/Conference proceedingConference PaperResearchpeer-review

1 Citation (Scopus)

Abstract

Deep learning methods usually require a large amount of training data and lack interpretability. In this paper, we pro-pose a novel knowledge distillation and model interpretation framework for medical image classification that jointly solves the above two issues. Specifically, to address the data-hungry issue, a small student model is learned with less data by distilling knowledge from a cumbersome pretrained teacher model. To interpret the teacher model and assist the learning of the student, an explainer module is introduced to highlight the regions of an input that are important for the predictions of the teacher model. Furthermore, the joint framework is trained by a principled way derived from the information-theoretic perspective. Our framework outperforms on the knowledge distillation and model interpretation tasks com-pared to state-of-the-art methods on a fundus dataset.

Original languageEnglish
Title of host publicationISBI 2022 - Symposium Proceedings
EditorsSuyash P. Awate, Tanveer F. Syeda-Mahmood
Place of PublicationPiscataway NJ USA
PublisherIEEE, Institute of Electrical and Electronics Engineers
Number of pages4
ISBN (Electronic)9781665429238
ISBN (Print)9781665429245
DOIs
Publication statusPublished - 2022
EventIEEE International Symposium on Biomedical Imaging (ISBI) 2022 - Kolkata, India
Duration: 28 Mar 202231 Mar 2022
Conference number: 19th
https://ieeexplore.ieee.org/xpl/conhome/9761376/proceeding (Proceedings)

Publication series

NameProceedings - International Symposium on Biomedical Imaging
Publisher IEEE, Institute of Electrical and Electronics Engineers
Volume2022-March
ISSN (Print)1945-7928
ISSN (Electronic)1945-8452

Conference

ConferenceIEEE International Symposium on Biomedical Imaging (ISBI) 2022
Abbreviated titleISBI 2022
Country/TerritoryIndia
CityKolkata
Period28/03/2231/03/22
Internet address

Keywords

  • Knowledge Distillation
  • Model Interpretation
  • Mutual Information

Cite this