Scaling Up a Multispectral Resnet-50 to 128 GPUs

Rocco Sedona, Gabriele Cavallaro, Jenia Jitsev, Alexandre Strube, Morris Riedel, Matthias Book

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

1 Citation (Scopus)

Abstract

Similarly to other scientific domains, Deep Learning (DL) holds great promises to fulfil the challenging needs of Remote Sensing (RS) applications. However, the increase in volume, variety and complexity of acquisitions that are carried out on a daily basis by Earth Observation (EO) missions generates new processing and storage challenges within operational processing pipelines. The aim of this work is to show that High-Performance Computing (HPC) systems can speed up the training time of Convolutional Neural Networks (CNNs). Particular attention is put on the monitoring of the classification accuracy that usually degrades when using large batch sizes. The experimental results of this work show that the training of the model scales up to a batch size of 8,000, obtaining classification performances in terms of accuracy in line with those using smaller batch sizes.

Original languageEnglish
Title of host publication2020 IEEE International Geoscience and Remote Sensing Symposium, IGARSS 2020 - Proceedings
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages1058-1061
Number of pages4
ISBN (Electronic)9781728163741
DOIs
Publication statusPublished - 26 Sept 2020
Event2020 IEEE International Geoscience and Remote Sensing Symposium, IGARSS 2020 - Virtual, Waikoloa, United States
Duration: 26 Sept 20202 Oct 2020

Publication series

NameInternational Geoscience and Remote Sensing Symposium (IGARSS)

Conference

Conference2020 IEEE International Geoscience and Remote Sensing Symposium, IGARSS 2020
Country/TerritoryUnited States
CityVirtual, Waikoloa
Period26/09/202/10/20

Bibliographical note

Publisher Copyright:
© 2020 IEEE.

Other keywords

  • classification
  • convolutional neural network
  • Distributed deep learning
  • high performance computing
  • residual neural network
  • sentinel-2

Fingerprint

Dive into the research topics of 'Scaling Up a Multispectral Resnet-50 to 128 GPUs'. Together they form a unique fingerprint.

Cite this