vesselFM: A Foundation Model for Universal 3D Blood Vessel Segmentation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wittmann, Bastian, Wattenberg, Yannick, Amiranashvili, Tamaz, Shit, Suprosanna, Menze, Bjoern
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910876442820608
author Wittmann, Bastian
Wattenberg, Yannick
Amiranashvili, Tamaz
Shit, Suprosanna
Menze, Bjoern
author_facet Wittmann, Bastian
Wattenberg, Yannick
Amiranashvili, Tamaz
Shit, Suprosanna
Menze, Bjoern
contents Segmenting 3D blood vessels is a critical yet challenging task in medical image analysis. This is due to significant imaging modality-specific variations in artifacts, vascular patterns and scales, signal-to-noise ratios, and background tissues. These variations, along with domain gaps arising from varying imaging protocols, limit the generalization of existing supervised learning-based methods, requiring tedious voxel-level annotations for each dataset separately. While foundation models promise to alleviate this limitation, they typically fail to generalize to the task of blood vessel segmentation, posing a unique, complex problem. In this work, we present vesselFM, a foundation model designed specifically for the broad task of 3D blood vessel segmentation. Unlike previous models, vesselFM can effortlessly generalize to unseen domains. To achieve zero-shot generalization, we train vesselFM on three heterogeneous data sources: a large, curated annotated dataset, data generated by a domain randomization scheme, and data sampled from a flow matching-based generative model. Extensive evaluations show that vesselFM outperforms state-of-the-art medical image segmentation foundation models across four (pre-)clinically relevant imaging modalities in zero-, one-, and few-shot scenarios, therefore providing a universal solution for 3D blood vessel segmentation.
format Preprint
id arxiv_https___arxiv_org_abs_2411_17386
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle vesselFM: A Foundation Model for Universal 3D Blood Vessel Segmentation
Wittmann, Bastian
Wattenberg, Yannick
Amiranashvili, Tamaz
Shit, Suprosanna
Menze, Bjoern
Image and Video Processing
Computer Vision and Pattern Recognition
Segmenting 3D blood vessels is a critical yet challenging task in medical image analysis. This is due to significant imaging modality-specific variations in artifacts, vascular patterns and scales, signal-to-noise ratios, and background tissues. These variations, along with domain gaps arising from varying imaging protocols, limit the generalization of existing supervised learning-based methods, requiring tedious voxel-level annotations for each dataset separately. While foundation models promise to alleviate this limitation, they typically fail to generalize to the task of blood vessel segmentation, posing a unique, complex problem. In this work, we present vesselFM, a foundation model designed specifically for the broad task of 3D blood vessel segmentation. Unlike previous models, vesselFM can effortlessly generalize to unseen domains. To achieve zero-shot generalization, we train vesselFM on three heterogeneous data sources: a large, curated annotated dataset, data generated by a domain randomization scheme, and data sampled from a flow matching-based generative model. Extensive evaluations show that vesselFM outperforms state-of-the-art medical image segmentation foundation models across four (pre-)clinically relevant imaging modalities in zero-, one-, and few-shot scenarios, therefore providing a universal solution for 3D blood vessel segmentation.
title vesselFM: A Foundation Model for Universal 3D Blood Vessel Segmentation
topic Image and Video Processing
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2411.17386