Salt & Pepper Heatmaps: Diffusion-informed Landmark Detection Strategy

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Wyatt, Julian, Voiculescu, Irina
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913428586627072
author Wyatt, Julian
Voiculescu, Irina
author_facet Wyatt, Julian
Voiculescu, Irina
contents Anatomical Landmark Detection is the process of identifying key areas of an image for clinical measurements. Each landmark is a single ground truth point labelled by a clinician. A machine learning model predicts the locus of a landmark as a probability region represented by a heatmap. Diffusion models have increased in popularity for generative modelling due to their high quality sampling and mode coverage, leading to their adoption in medical image processing for semantic segmentation. Diffusion modelling can be further adapted to learn a distribution over landmarks. The stochastic nature of diffusion models captures fluctuations in the landmark prediction, which we leverage by blurring into meaningful probability regions. In this paper, we reformulate automatic Anatomical Landmark Detection as a precise generative modelling task, producing a few-hot pixel heatmap. Our method achieves state-of-the-art MRE and comparable SDR performance with existing work.
format Preprint
id arxiv_https___arxiv_org_abs_2407_09192
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Salt & Pepper Heatmaps: Diffusion-informed Landmark Detection Strategy
Wyatt, Julian
Voiculescu, Irina
Computer Vision and Pattern Recognition
Anatomical Landmark Detection is the process of identifying key areas of an image for clinical measurements. Each landmark is a single ground truth point labelled by a clinician. A machine learning model predicts the locus of a landmark as a probability region represented by a heatmap. Diffusion models have increased in popularity for generative modelling due to their high quality sampling and mode coverage, leading to their adoption in medical image processing for semantic segmentation. Diffusion modelling can be further adapted to learn a distribution over landmarks. The stochastic nature of diffusion models captures fluctuations in the landmark prediction, which we leverage by blurring into meaningful probability regions. In this paper, we reformulate automatic Anatomical Landmark Detection as a precise generative modelling task, producing a few-hot pixel heatmap. Our method achieves state-of-the-art MRE and comparable SDR performance with existing work.
title Salt & Pepper Heatmaps: Diffusion-informed Landmark Detection Strategy
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2407.09192