HiSwin UT: Hybrid Swin Transformers Network for Segmenting 3D in Vivo Two-photon Images of Rodent Cerebrovasculature

Jie Li,Chong Wang,Zizhen Li,Hao Li,Mingzhu Sun,Xin Zhao
DOI: https://doi.org/10.1109/bibm58861.2023.10385547
2023-01-01
Abstract:Two-photon microscopy is currently the preferred technique for in vivo vasculature imaging. The ability to generate high-resolution 3D images of blood vessels could improve our understanding of normal vascular physiology and disease pathogenesis. Automatic segmenting and mapping of these 3D blood vessel networks remain challenging due to vascular geometries' complexity and signal-to-noise ratio limitations. In this paper, we propose a novel segmentation network named HYbrid Swin Unet Transformers (HiSwin UT) to address this problem. The network explores the use of the Swin transformer to segment 3D volumetric vessels in vivo images acquired by two-photon microscopy. We propose to use a hybrid of hierarchical Swin transformers with 3D convolutional neural networks (3D CNNs) as both the encoder and the decoder. The Swin transformer guards guide the 3D CNN laborers at every stage in both the encoding and decoding paths to enlarge the receptive field of the network. Specifically, the 3D vasculature images are projected into a sequence of embedding and fed into the network. The encoders extract features at five different resolutions by using a combination of sifted windows self-attention layers and residual convolutional layers. Then these latent multi-scale features are connected to each decoder via skip connection to cater to the diverse range of vessels in terms of their sizes and shapes. We evaluate this HiSwin UT model on the public two-photon vasculature dataset, extensive experimental results demonstrate the superiority of our network compared with the state-of-the-art models. The source code of the HiSwin UT will be publicly available at https://github.com/UTINK/HiSwinUT.
What problem does this paper attempt to address?