Continuous and complete liver vessel segmentation with graph-attention guided diffusion

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhang, Xiaotong, Broersen, Alexander, van Erp, Gonnie CM, Pintea, Silvia L., Dijkstra, Jouke
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915575198908416
author Zhang, Xiaotong
Broersen, Alexander
van Erp, Gonnie CM
Pintea, Silvia L.
Dijkstra, Jouke
author_facet Zhang, Xiaotong
Broersen, Alexander
van Erp, Gonnie CM
Pintea, Silvia L.
Dijkstra, Jouke
contents Improving connectivity and completeness are the most challenging aspects of liver vessel segmentation, especially for small vessels. These challenges require both learning the continuous vessel geometry, and focusing on small vessel detection. However, current methods do not explicitly address these two aspects and cannot generalize well when constrained by inconsistent annotations. Here, we take advantage of the generalization of the diffusion model and explicitly integrate connectivity and completeness in our diffusion-based segmentation model. Specifically, we use a graph-attention module that adds knowledge about vessel geometry, and thus adds continuity. Additionally, we perform the graph-attention at multiple-scales, thus focusing on small liver vessels. Our method outperforms eight state-of-the-art medical segmentation methods on two public datasets: 3D-ircadb-01 and LiVS. Our code is available at https://github.com/ZhangXiaotong015/GATSegDiff.
format Preprint
id arxiv_https___arxiv_org_abs_2411_00617
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Continuous and complete liver vessel segmentation with graph-attention guided diffusion
Zhang, Xiaotong
Broersen, Alexander
van Erp, Gonnie CM
Pintea, Silvia L.
Dijkstra, Jouke
Image and Video Processing
Computer Vision and Pattern Recognition
Improving connectivity and completeness are the most challenging aspects of liver vessel segmentation, especially for small vessels. These challenges require both learning the continuous vessel geometry, and focusing on small vessel detection. However, current methods do not explicitly address these two aspects and cannot generalize well when constrained by inconsistent annotations. Here, we take advantage of the generalization of the diffusion model and explicitly integrate connectivity and completeness in our diffusion-based segmentation model. Specifically, we use a graph-attention module that adds knowledge about vessel geometry, and thus adds continuity. Additionally, we perform the graph-attention at multiple-scales, thus focusing on small liver vessels. Our method outperforms eight state-of-the-art medical segmentation methods on two public datasets: 3D-ircadb-01 and LiVS. Our code is available at https://github.com/ZhangXiaotong015/GATSegDiff.
title Continuous and complete liver vessel segmentation with graph-attention guided diffusion
topic Image and Video Processing
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2411.00617