Detail publikace

Upsampling Algorithms for Autoencoder Segmentation Neural Networks: A Comparison Study

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

Originální název

Upsampling Algorithms for Autoencoder Segmentation Neural Networks: A Comparison Study

Typ

článek ve sborníku ve WoS nebo Scopus

Jazyk

angličtina

Originální abstrakt

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.

Klíčová slova

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

Autoři

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

Vydáno

28. 10. 2019

Místo

Dublin

ISBN

978-1-7281-5764-1

Kniha

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

Strany od

1

Strany do

5

Strany počet

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"
}