Publication detail

Upsampling Algorithms for Autoencoder Segmentation Neural Networks: A Comparison Study

KOLAŘÍK, M. BURGET, R. ŘÍHA, K.

Original Title

Upsampling Algorithms for Autoencoder Segmentation Neural Networks: A Comparison Study

Type

conference paper

Language

English

Original Abstract

This paper compares nine different upsampling methods used in convolutional neural networks in terms of accuracy and processing speed. The process of image segmentation using autoencoder neural networks consists of the image downsampling in the encoder and correspondingly of image upsampling in the decoder part of the network to achieve original image resolution. This paper focuses on the upsampling process in the decoder part of the standard U-Net neural network. Three different interpolations are compared with and without subsequent 1x1 convolution layers and three transpose convolution layers for image upsampling using different size convolutional cores. The experiment has shown that the best practical results were achieved using simple nearest neighbor interpolation upsampling taking into consideration the computational time needed. The network using nearest neighbor interpolation upsampling achieved pixel accuracy of 99.47\% and has shown fast training time and convergence in comparison with other networks using different upsampling methods. The data used in this work consist of a lumbar CT spine segmentation dataset.

Keywords

autoencoder, machine learning, segmentation, u-net, upsampling

Authors

KOLAŘÍK, M.; BURGET, R.; ŘÍHA, K.

Released

28. 10. 2019

Location

Dublin

ISBN

978-1-7281-5764-1

Book

2019 11th International Congress on Ultra Modern Telecommunications and Control Systems and Workshops (ICUMT)

Pages from

1

Pages to

5

Pages count

5

URL

BibTex

@inproceedings{BUT159728,
  author="Martin {Kolařík} and Radim {Burget} and Kamil {Říha}",
  title="Upsampling Algorithms for Autoencoder Segmentation Neural Networks: A Comparison Study",
  booktitle="2019 11th International Congress on Ultra Modern Telecommunications and Control Systems and Workshops (ICUMT)",
  year="2019",
  pages="1--5",
  address="Dublin",
  doi="10.1109/ICUMT48472.2019.8970918",
  isbn="978-1-7281-5764-1",
  url="https://ieeexplore.ieee.org/abstract/document/8970918"
}