Reducing Annotation Need in Self-explanatory Models for Lung Nodule Diagnosis

Research output: Chapter in Book/Report/Conference proceedingArticle in proceedingsResearchpeer-review

Documents

  • Fulltext

    Submitted manuscript, 871 KB, PDF document

Feature-based self-explanatory methods explain their classification in terms of human-understandable features. In the medical imaging community, this semantic matching of clinical knowledge adds significantly to the trustworthiness of the AI. However, the cost of additional annotation of features remains a pressing issue. We address this problem by proposing cRedAnno, a data-/annotation-efficient self-explanatory approach for lung nodule diagnosis. cRedAnno considerably reduces the annotation need by introducing self-supervised contrastive learning to alleviate the burden of learning most parameters from annotation, replacing end-to-end training with two-stage training. When training with hundreds of nodule samples and only 1% of their annotations, cRedAnno achieves competitive accuracy in predicting malignancy, meanwhile significantly surpassing most previous works in predicting nodule attributes. Visualisation of the learned space further indicates that the correlation between the clustering of malignancy and nodule attributes coincides with clinical knowledge. Our complete code is open-source available: https://github.com/diku-dk/credanno.

Original languageEnglish
Title of host publicationInterpretability of Machine Intelligence in Medical Image Computing
EditorsM Reyes, PH Abreu, J Cardoso
PublisherSpringer
Publication date2022
Pages33-43
ISBN (Print)978-3-031-17975-4
DOIs
Publication statusPublished - 2022
Event5th International Workshop on Interpretability of Machine Intelligence in Medical Image Computing (IMIMIC) - Singapore, Singapore
Duration: 22 Sep 2022 → …

Conference

Conference5th International Workshop on Interpretability of Machine Intelligence in Medical Image Computing (IMIMIC)
LandSingapore
BySingapore
Periode22/09/2022 → …
SeriesLecture Notes in Computer Science
Volume13611
ISSN0302-9743

    Research areas

  • Explainable AI, Lung nodule diagnosis, Self-explanatory model, Intrinsic explanation, Self-supervised learning

ID: 324695516