3D WaveUNet: 3D Wavelet Integrated Encoder-Decoder Network for Neuron Segmentation

by   Qiufu Li, et al.
Shenzhen University
NetEase, Inc

3D neuron segmentation is a key step for the neuron digital reconstruction, which is essential for exploring brain circuits and understanding brain functions. However, the fine line-shaped nerve fibers of neuron could spread in a large region, which brings great computational cost to the segmentation in 3D neuronal images. Meanwhile, the strong noises and disconnected nerve fibers in the image bring great challenges to the task. In this paper, we propose a 3D wavelet and deep learning based 3D neuron segmentation method. The neuronal image is first partitioned into neuronal cubes to simplify the segmentation task. Then, we design 3D WaveUNet, the first 3D wavelet integrated encoder-decoder network, to segment the nerve fibers in the cubes; the wavelets could assist the deep networks in suppressing data noise and connecting the broken fibers. We also produce a Neuronal Cube Dataset (NeuCuDa) using the biggest available annotated neuronal image dataset, BigNeuron, to train 3D WaveUNet. Finally, the nerve fibers segmented in cubes are assembled to generate the complete neuron, which is digitally reconstructed using an available automatic tracing algorithm. The experimental results show that our neuron segmentation method could completely extract the target neuron in noisy neuronal images. The integrated 3D wavelets can efficiently improve the performance of 3D neuron segmentation and reconstruction. The code and pre-trained models for this work will be available at https://github.com/LiQiufu/3D-WaveUNet.


page 1

page 3

page 4

page 5

page 6

page 8

page 9


PointNeuron: 3D Neuron Reconstruction via Geometry and Topology Learning of Point Clouds

Digital neuron reconstruction from 3D microscopy images is an essential ...

Quality Control of Neuron Reconstruction Based on Deep Learning

Neuron reconstruction is essential to generate exquisite neuron connecti...

WaveSNet: Wavelet Integrated Deep Networks for Image Segmentation

In deep networks, the lost data details significantly degrade the perfor...

NRTR: Neuron Reconstruction with Transformer from 3D Optical Microscopy Images

The neuron reconstruction from raw Optical Microscopy (OM) image stacks ...

Hidden Markov Modeling for Maximum Likelihood Neuron Reconstruction

Recent advances in brain clearing and imaging have made it possible to i...

Efficient 2D neuron boundary segmentation with local topological constraints

We present a method for segmenting neuron membranes in 2D electron micro...

SANTIAGO: Spine Association for Neuron Topology Improvement and Graph Optimization

Developing automated and semi-automated solutions for reconstructing wir...

Please sign up or login with your details

Forgot password? Click here to reset