Towards Out-of-Distribution Detection for breast cancer classification in Point-of-Care Ultrasound Imaging

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Karlsson, Jennie, Wodrich, Marisa, Overgaard, Niels Christian, Sahlin, Freja, Lång, Kristina, Heyden, Anders, Arvidsson, Ida
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866929259298160640
author Karlsson, Jennie
Wodrich, Marisa
Overgaard, Niels Christian
Sahlin, Freja
Lång, Kristina
Heyden, Anders
Arvidsson, Ida
author_facet Karlsson, Jennie
Wodrich, Marisa
Overgaard, Niels Christian
Sahlin, Freja
Lång, Kristina
Heyden, Anders
Arvidsson, Ida
contents Deep learning has shown to have great potential in medical applications. In critical domains as such, it is of high interest to have trustworthy algorithms which are able to tell when reliable assessments cannot be guaranteed. Detecting out-of-distribution (OOD) samples is a crucial step towards building a safe classifier. Following a previous study, showing that it is possible to classify breast cancer in point-of-care ultrasound images, this study investigates OOD detection using three different methods: softmax, energy score and deep ensembles. All methods are tested on three different OOD data sets. The results show that the energy score method outperforms the softmax method, performing well on two of the data sets. The ensemble method is the most robust, performing the best at detecting OOD samples for all three OOD data sets.
format Preprint
id arxiv_https___arxiv_org_abs_2402_18960
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Towards Out-of-Distribution Detection for breast cancer classification in Point-of-Care Ultrasound Imaging
Karlsson, Jennie
Wodrich, Marisa
Overgaard, Niels Christian
Sahlin, Freja
Lång, Kristina
Heyden, Anders
Arvidsson, Ida
Computer Vision and Pattern Recognition
Artificial Intelligence
Deep learning has shown to have great potential in medical applications. In critical domains as such, it is of high interest to have trustworthy algorithms which are able to tell when reliable assessments cannot be guaranteed. Detecting out-of-distribution (OOD) samples is a crucial step towards building a safe classifier. Following a previous study, showing that it is possible to classify breast cancer in point-of-care ultrasound images, this study investigates OOD detection using three different methods: softmax, energy score and deep ensembles. All methods are tested on three different OOD data sets. The results show that the energy score method outperforms the softmax method, performing well on two of the data sets. The ensemble method is the most robust, performing the best at detecting OOD samples for all three OOD data sets.
title Towards Out-of-Distribution Detection for breast cancer classification in Point-of-Care Ultrasound Imaging
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2402.18960