Exascale Deep Learning for Climate Analytics
Not on the map. Why?
Abstract (arXiv)
We extract pixel-level masks of extreme weather patterns using variants of Tiramisu and DeepLabv3+ neural networks. We describe improvements to the software frameworks, input pipeline, and the network training algorithms necessary to efficiently scale deep learning on the Piz Daint and Summit systems. The Tiramisu network scales to 5300 P100 GPUs with a sustained throughput of 21.0 PF/s and parallel efficiency of 79.0%. DeepLabv3+ scales up to 27360 V100 GPUs with a sustained throughput of 325.8 PF/s and a parallel efficiency of 90.7% in single precision. By taking advantage of the FP16 Tensor Cores, a half-precision version of the DeepLabv3+ network achieves a peak and sustained throughput of 1.13 EF/s and 999.0 PF/s respectively.
Key facts. Cited by 28 papers in the map (data through 7 Oct 2026). First posted on arXiv 3 Oct 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 11; 2022: 4; 2023: 7; 2024: 4; 2025: 2.
28 papers.
Sort:
- 2021Graph Neural Controlled Differential Equations for Traffic Forecastingcited by 133
- 2021Will Artificial Intelligence supersede Earth System and Climate Models?cited by 37
- 2021An Efficient Statistical-based Gradient Compression Technique for Distributed Training Systemscited by 32
- 2021Climate Modeling with Neural Diffusion Equationscited by 20
- 2021Bridging observation, theory and numerical simulation of the ocean using Machine Learningcited by 15
- 2021MLPerf HPC: A Holistic Benchmark Suite for Scientific Machine Learning on HPC Systemscited by 15
- 2021Clairvoyant Prefetching for Distributed Machine Learning I/Ocited by 14
- 2023Explainable GeoAI: Can saliency maps help interpret artificial intelligence's learning process? An empirical study on natural feature detectioncited by 14
- 2023Graph Neural Rough Differential Equations for Traffic Forecastingcited by 14
- 2023Learning from Predictions: Fusing Training and Autoregressive Inference for Long-Term Spatiotemporal Forecastscited by 11
- 2021JUWELS Booster -- A Supercomputer for Large-Scale AI Researchcited by 10
- 2021PCE-PINNs: Physics-Informed Neural Networks for Uncertainty Propagation in Ocean Modelingcited by 8
16 more
- 2021COMET: A Novel Memory-Efficient Deep Learning Training Framework by Using Error-Bounded Lossy Compressioncited by 8
- 2023Adaptive learning of effective dynamics: Adaptive real-time, online modeling for complex systemscited by 4
- 2024EvoMesh: Adaptive Physical Simulation with Hierarchical Graph Evolutionscited by 4
- 2022Analysis, Characterization, Prediction and Attribution of Extreme Atmospheric Events with Machine Learning: a Reviewcited by 3
- 2022Deep NURBS -- Admissible Physics-informed Neural Networkscited by 3
- 2025AI Benchmark Democratization and Carpentrycited by 3
- 2021An Oracle for Guiding Large-Scale Model/Hybrid Parallel Training of Convolutional Neural Networkscited by 2
- 2024Scalable Artificial Intelligence for Science: Perspectives, Methods and Exemplarscited by 2
- 2023KAKURENBO: Adaptively Hiding Samples in Deep Neural Network Trainingcited by 1
- 2024Streamlining Ocean Dynamics Modeling with Fourier Neural Operators: A Multiobjective Hyperparameter and Architecture Optimization Approachcited by 1
- 2022SOLAR: A Highly Optimized Data Loading Framework for Distributed Training of CNN-based Scientific Surrogatescited by 0
- 2022SAIH: A Scalable Evaluation Methodology for Understanding AI Performance Trend on HPC Systemscited by 0
- 2023Unsupervised Discovery of Extreme Weather Events Using Universal Representations of Emergent Organizationcited by 0
- 2023Evaluating Loss Functions and Learning Data Pre-Processing for Climate Downscaling Deep Learning Modelscited by 0
- 2024Analyzing the Impact of Climate Change With Major Emphasis on Pollution: A Comparative Study of ML and Statistical Models in Time Series Datacited by 0
- 2025GAIA: A Foundation Model for Operational Atmospheric Dynamicscited by 0
- 2025AI Benchmark Democratization and Carpentry12 Dec 2025
- 2025GAIA: A Foundation Model for Operational Atmospheric Dynamics15 May 2025
- 2024EvoMesh: Adaptive Physical Simulation with Hierarchical Graph Evolutions3 Oct 2024
- 2024Scalable Artificial Intelligence for Science: Perspectives, Methods and Exemplars24 Jun 2024
- 2024Analyzing the Impact of Climate Change With Major Emphasis on Pollution: A Comparative Study of ML and Statistical Models in Time Series Data24 May 2024
- 2024Streamlining Ocean Dynamics Modeling with Fourier Neural Operators: A Multiobjective Hyperparameter and Architecture Optimization Approach7 Apr 2024
- 2023KAKURENBO: Adaptively Hiding Samples in Deep Neural Network Training16 Oct 2023
- 2023Evaluating Loss Functions and Learning Data Pre-Processing for Climate Downscaling Deep Learning Models19 Jun 2023
- 2023Unsupervised Discovery of Extreme Weather Events Using Universal Representations of Emergent Organization25 Apr 2023
- 2023Adaptive learning of effective dynamics: Adaptive real-time, online modeling for complex systems4 Apr 2023
- 2023Graph Neural Rough Differential Equations for Traffic Forecasting20 Mar 2023
- 2023Explainable GeoAI: Can saliency maps help interpret artificial intelligence's learning process? An empirical study on natural feature detection16 Mar 2023
16 more
- 2023Learning from Predictions: Fusing Training and Autoregressive Inference for Long-Term Spatiotemporal Forecasts22 Feb 2023
- 2022SAIH: A Scalable Evaluation Methodology for Understanding AI Performance Trend on HPC Systems7 Dec 2022
- 2022SOLAR: A Highly Optimized Data Loading Framework for Distributed Training of CNN-based Scientific Surrogates1 Nov 2022
- 2022Deep NURBS -- Admissible Physics-informed Neural Networks25 Oct 2022
- 2022Analysis, Characterization, Prediction and Attribution of Extreme Atmospheric Events with Machine Learning: a Review3 Jun 2022
- 2021Graph Neural Controlled Differential Equations for Traffic Forecasting7 Dec 2021
- 2021COMET: A Novel Memory-Efficient Deep Learning Training Framework by Using Error-Bounded Lossy Compression18 Nov 2021
- 2021Climate Modeling with Neural Diffusion Equations11 Nov 2021
- 2021MLPerf HPC: A Holistic Benchmark Suite for Scientific Machine Learning on HPC Systems21 Oct 2021
- 2021JUWELS Booster -- A Supercomputer for Large-Scale AI Research30 Jun 2021
- 2021PCE-PINNs: Physics-Informed Neural Networks for Uncertainty Propagation in Ocean Modeling5 May 2021
- 2021Bridging observation, theory and numerical simulation of the ocean using Machine Learning26 Apr 2021
- 2021An Oracle for Guiding Large-Scale Model/Hybrid Parallel Training of Convolutional Neural Networks19 Apr 2021
- 2021An Efficient Statistical-based Gradient Compression Technique for Distributed Training Systems26 Jan 2021
- 2021Will Artificial Intelligence supersede Earth System and Climate Models?22 Jan 2021
- 2021Clairvoyant Prefetching for Distributed Machine Learning I/O21 Jan 2021
- 2021Graph Neural Controlled Differential Equations for Traffic Forecastingcited by 133
- 2021Will Artificial Intelligence supersede Earth System and Climate Models?cited by 37
- 2021An Efficient Statistical-based Gradient Compression Technique for Distributed Training Systemscited by 32
- 2021Climate Modeling with Neural Diffusion Equationscited by 20
- 2021Bridging observation, theory and numerical simulation of the ocean using Machine Learningcited by 15
- 2021MLPerf HPC: A Holistic Benchmark Suite for Scientific Machine Learning on HPC Systemscited by 15
- 2021Clairvoyant Prefetching for Distributed Machine Learning I/Ocited by 14
- 2023Explainable GeoAI: Can saliency maps help interpret artificial intelligence's learning process? An empirical study on natural feature detectioncited by 14
- 2023Graph Neural Rough Differential Equations for Traffic Forecastingcited by 14
- 2023Learning from Predictions: Fusing Training and Autoregressive Inference for Long-Term Spatiotemporal Forecastscited by 11
- 2021JUWELS Booster -- A Supercomputer for Large-Scale AI Researchcited by 10
- 2021PCE-PINNs: Physics-Informed Neural Networks for Uncertainty Propagation in Ocean Modelingcited by 8
12 most cited of 28 All 28 (JSON)
- 2025AI Benchmark Democratization and Carpentry12 Dec 2025
- 2025GAIA: A Foundation Model for Operational Atmospheric Dynamics15 May 2025
- 2024EvoMesh: Adaptive Physical Simulation with Hierarchical Graph Evolutions3 Oct 2024
- 2024Scalable Artificial Intelligence for Science: Perspectives, Methods and Exemplars24 Jun 2024
- 2024Analyzing the Impact of Climate Change With Major Emphasis on Pollution: A Comparative Study of ML and Statistical Models in Time Series Data24 May 2024
- 2024Streamlining Ocean Dynamics Modeling with Fourier Neural Operators: A Multiobjective Hyperparameter and Architecture Optimization Approach7 Apr 2024
- 2023KAKURENBO: Adaptively Hiding Samples in Deep Neural Network Training16 Oct 2023
- 2023Evaluating Loss Functions and Learning Data Pre-Processing for Climate Downscaling Deep Learning Models19 Jun 2023
- 2023Unsupervised Discovery of Extreme Weather Events Using Universal Representations of Emergent Organization25 Apr 2023
- 2023Adaptive learning of effective dynamics: Adaptive real-time, online modeling for complex systems4 Apr 2023
- 2023Graph Neural Rough Differential Equations for Traffic Forecasting20 Mar 2023
- 2023Explainable GeoAI: Can saliency maps help interpret artificial intelligence's learning process? An empirical study on natural feature detection16 Mar 2023
12 newest of 28 All 28 (JSON)