Alert button

Parallel Multi-Dimensional LSTM, With Application to Fast Biomedical Volumetric Image Segmentation

Add code
Bookmark button
Alert button
Jun 24, 2015
Marijn F. Stollenga, Wonmin Byeon, Marcus Liwicki, Juergen Schmidhuber

Figure 1 for Parallel Multi-Dimensional LSTM, With Application to Fast Biomedical Volumetric Image Segmentation
Figure 2 for Parallel Multi-Dimensional LSTM, With Application to Fast Biomedical Volumetric Image Segmentation
Figure 3 for Parallel Multi-Dimensional LSTM, With Application to Fast Biomedical Volumetric Image Segmentation
Figure 4 for Parallel Multi-Dimensional LSTM, With Application to Fast Biomedical Volumetric Image Segmentation

Share this with someone who'll enjoy it:

Twitter IconFacebook IconLinkedin IconWhatsapp IconMessenger Icon

Convolutional Neural Networks (CNNs) can be shifted across 2D images or 3D videos to segment them. They have a fixed input size and typically perceive only small local contexts of the pixels to be classified as foreground or background. In contrast, Multi-Dimensional Recurrent NNs (MD-RNNs) can perceive the entire spatio-temporal context of each pixel in a few sweeps through all pixels, especially when the RNN is a Long Short-Term Memory (LSTM). Despite these theoretical advantages, however, unlike CNNs, previous MD-LSTM variants were hard to parallelize on GPUs. Here we re-arrange the traditional cuboid order of computations in MD-LSTM in pyramidal fashion. The resulting PyraMiD-LSTM is easy to parallelize, especially for 3D data such as stacks of brain slice images. PyraMiD-LSTM achieved best known pixel-wise brain image segmentation results on MRBrainS13 (and competitive results on EM-ISBI12).

* Marijn F. Stollenga and Wonmin Byeon are the shared first authors, both authors contributed equally to this work  
View paper onarxiv icon

Share this with someone who'll enjoy it:

Twitter IconFacebook IconLinkedin IconWhatsapp IconMessenger Icon