Interpretable and Interactive Deep Multiple Instance Learning for Dental Caries Classification in Bitewing X-rays

Autor:	Bergner, Benjamin, Rohrer, Csaba, Taleb, Aiham, Duchrau, Martha, De Leon, Guilherme, Rodrigues, Jonas Almeida, Schwendicke, Falk, Krois, Joachim, Lippert, Christoph
Rok vydání:	2021
Předmět:	Electrical Engineering and Systems Science - Image and Video Processing Computer Science - Artificial Intelligence Computer Science - Computer Vision and Pattern Recognition Computer Science - Machine Learning
Druh dokumentu:	Working Paper
Popis:	We propose a simple and efficient image classification architecture based on deep multiple instance learning, and apply it to the challenging task of caries detection in dental radiographs. Technically, our approach contributes in two ways: First, it outputs a heatmap of local patch classification probabilities despite being trained with weak image-level labels. Second, it is amenable to learning from segmentation labels to guide training. In contrast to existing methods, the human user can faithfully interpret predictions and interact with the model to decide which regions to attend to. Experiments are conducted on a large clinical dataset of $\sim$38k bitewings ($\sim$316k teeth), where we achieve competitive performance compared to various baselines. When guided by an external caries segmentation model, a significant improvement in classification and localization performance is observed. Comment: 19 pages, 10 figures, Full Paper, MIDL 2022
Databáze:	arXiv
Externí odkaz:	http://arxiv.org/abs/2112.09694 Zobrazit plný text záznamu View this record from Arxiv