Leveraging Frequency Domain Learning in 3D Vessel Segmentation
- URL: http://arxiv.org/abs/2401.06224v1
- Date: Thu, 11 Jan 2024 19:07:58 GMT
- Title: Leveraging Frequency Domain Learning in 3D Vessel Segmentation
- Authors: Xinyuan Wang, Chengwei Pan, Hongming Dai, Gangming Zhao, Jinpeng Li,
Xiao Zhang, Yizhou Yu
- Abstract summary: In this study, we leverage Fourier domain learning as a substitute for multi-scale convolutional kernels in 3D hierarchical segmentation models.
We show that our novel network achieves remarkable dice performance (84.37% on ASACA500 and 80.32% on ImageCAS) in tubular vessel segmentation tasks.
- Score: 50.54833091336862
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Coronary microvascular disease constitutes a substantial risk to human
health. Employing computer-aided analysis and diagnostic systems, medical
professionals can intervene early in disease progression, with 3D vessel
segmentation serving as a crucial component. Nevertheless, conventional U-Net
architectures tend to yield incoherent and imprecise segmentation outcomes,
particularly for small vessel structures. While models with attention
mechanisms, such as Transformers and large convolutional kernels, demonstrate
superior performance, their extensive computational demands during training and
inference lead to increased time complexity. In this study, we leverage Fourier
domain learning as a substitute for multi-scale convolutional kernels in 3D
hierarchical segmentation models, which can reduce computational expenses while
preserving global receptive fields within the network. Furthermore, a
zero-parameter frequency domain fusion method is designed to improve the skip
connections in U-Net architecture. Experimental results on a public dataset and
an in-house dataset indicate that our novel Fourier transformation-based
network achieves remarkable dice performance (84.37\% on ASACA500 and 80.32\%
on ImageCAS) in tubular vessel segmentation tasks and substantially reduces
computational requirements without compromising global receptive fields.
Related papers
- A generalised novel loss function for computational fluid dynamics [0.9999629695552196]
A novel loss function is proposed called Gradient Mean Squared Error (GMSE)
It automatically and dynamically identifies the regions of importance on a field-by-field basis, assigning appropriate weights according to the local variance.
It is hoped that this loss function will enable accelerated machine learning within computational fluid dynamics.
arXiv Detail & Related papers (2024-11-26T02:57:31Z) - MambaClinix: Hierarchical Gated Convolution and Mamba-Based U-Net for Enhanced 3D Medical Image Segmentation [6.673169053236727]
We propose MambaClinix, a novel U-shaped architecture for medical image segmentation.
MambaClinix integrates a hierarchical gated convolutional network with Mamba in an adaptive stage-wise framework.
Our results show that MambaClinix achieves high segmentation accuracy while maintaining low model complexity.
arXiv Detail & Related papers (2024-09-19T07:51:14Z) - Skeleton Recall Loss for Connectivity Conserving and Resource Efficient Segmentation of Thin Tubular Structures [2.0710692923459804]
Accurately segmenting thin tubular structures, such as vessels, nerves, roads or concrete cracks, is a crucial task in computer vision.
Standard deep learning-based segmentation loss functions, such as Dice or Cross-Entropy, focus on overlap, often at the expense of preserving structural connectivity or topology.
We propose a novel Skeleton Recall Loss, which effectively addresses these challenges by circumventing intensive GPU-based calculations with inexpensive CPU operations.
arXiv Detail & Related papers (2024-04-03T18:42:19Z) - Predicting Infant Brain Connectivity with Federated Multi-Trajectory
GNNs using Scarce Data [54.55126643084341]
Existing deep learning solutions suffer from three major limitations.
We introduce FedGmTE-Net++, a federated graph-based multi-trajectory evolution network.
Using the power of federation, we aggregate local learnings among diverse hospitals with limited datasets.
arXiv Detail & Related papers (2024-01-01T10:20:01Z) - 3D Mitochondria Instance Segmentation with Spatio-Temporal Transformers [101.44668514239959]
We propose a hybrid encoder-decoder framework that efficiently computes spatial and temporal attentions in parallel.
We also introduce a semantic clutter-background adversarial loss during training that aids in the region of mitochondria instances from the background.
arXiv Detail & Related papers (2023-03-21T17:58:49Z) - NAF: Neural Attenuation Fields for Sparse-View CBCT Reconstruction [79.13750275141139]
This paper proposes a novel and fast self-supervised solution for sparse-view CBCT reconstruction.
The desired attenuation coefficients are represented as a continuous function of 3D spatial coordinates, parameterized by a fully-connected deep neural network.
A learning-based encoder entailing hash coding is adopted to help the network capture high-frequency details.
arXiv Detail & Related papers (2022-09-29T04:06:00Z) - Multi-organ Segmentation Network with Adversarial Performance Validator [10.775440368500416]
This paper introduces an adversarial performance validation network into a 2D-to-3D segmentation framework.
The proposed network converts the 2D-coarse result to 3D high-quality segmentation masks in a coarse-to-fine manner, allowing joint optimization to improve segmentation accuracy.
Experiments on the NIH pancreas segmentation dataset demonstrate the proposed network achieves state-of-the-art accuracy on small organ segmentation and outperforms the previous best.
arXiv Detail & Related papers (2022-04-16T18:00:29Z) - Convolutional generative adversarial imputation networks for
spatio-temporal missing data in storm surge simulations [86.5302150777089]
Generative Adversarial Imputation Nets (GANs) and GAN-based techniques have attracted attention as unsupervised machine learning methods.
We name our proposed method as Con Conval Generative Adversarial Imputation Nets (Conv-GAIN)
arXiv Detail & Related papers (2021-11-03T03:50:48Z) - Deep Cellular Recurrent Network for Efficient Analysis of Time-Series
Data with Spatial Information [52.635997570873194]
This work proposes a novel deep cellular recurrent neural network (DCRNN) architecture to process complex multi-dimensional time series data with spatial information.
The proposed architecture achieves state-of-the-art performance while utilizing substantially less trainable parameters when compared to comparable methods in the literature.
arXiv Detail & Related papers (2021-01-12T20:08:18Z) - DVNet: A Memory-Efficient Three-Dimensional CNN for Large-Scale
Neurovascular Reconstruction [1.9199289015460215]
We present a fully-convolutional, deep, and densely-connected encoder-decoder for pixel-wise semantic segmentation.
The proposed network provides superior performance for semantic segmentation problems applied to open-source benchmarks.
arXiv Detail & Related papers (2020-02-04T22:39:58Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.