English
Related papers

Related papers: Jigsaw: Training Multi-Billion-Parameter AI Weathe…

200 papers

Physics-based atmosphere-land models with prescribed sea surface temperature have notable successes but also biases in their ability to represent atmospheric variability compared to observations. Recently, AI emulators and hybrid models…

Atmospheric and Oceanic Physics · Physics 2026-04-22 Ian Baxter , Hamid Pahlavan , Pedram Hassanzadeh , Katharine Rucker , Tiffany Shaw

Foundation models have impressive performance and generalization capabilities across a wide range of applications. The increasing size of the models introduces great challenges for the training. Tensor parallelism is a critical technique…

Distributed, Parallel, and Cluster Computing · Computer Science 2023-01-23 Shenggan Cheng , Ziming Liu , Jiangsu Du , Yang You

Numerical Weather Prediction (NWP) system is an infrastructure that exerts considerable impacts on modern society.Traditional NWP system, however, resolves it by solving complex partial differential equations with a huge computing cluster,…

Artificial Intelligence · Computer Science 2024-09-26 Junchao Gong , Tao Han , Kang Chen , Lei Bai

Current serverless platforms struggle to optimize resource utilization due to their dynamic and fine-grained nature. Conventional techniques like overcommitment and autoscaling fall short, often sacrificing utilization for practicability or…

Distributed, Parallel, and Cluster Computing · Computer Science 2024-03-04 Qingyuan Liu , Yanning Yang , Dong Du , Yubin Xia , Ping Zhang , Jia Feng , James Larus , Haibo Chen

Aerosol forecasting is essential for air quality warnings, health risk assessment, and climate change mitigation. However, it is more complex than weather forecasting due to the intricate interactions between aerosol physicochemical…

Big Artificial Intelligence (AI) models have emerged as a crucial element in various intelligent applications at the edge, such as voice assistants in smart homes and autonomous robotics in smart factories. Training big AI models, e.g., for…

Machine Learning · Computer Science 2024-04-30 Liekang Zeng , Shengyuan Ye , Xu Chen , Yang Yang

Deep learning models are trained on servers with many GPUs, and training must scale with the number of GPUs. Systems such as TensorFlow and Caffe2 train models with parallel synchronous stochastic gradient descent: they process a batch of…

Distributed, Parallel, and Cluster Computing · Computer Science 2019-01-09 Alexandros Koliousis , Pijika Watcharapichat , Matthias Weidlich , Luo Mai , Paolo Costa , Peter Pietzuch

The computational requirements for training deep neural networks (DNNs) have grown to the point that it is now standard practice to parallelize training. Existing deep learning systems commonly use data or model parallelism, but…

Distributed, Parallel, and Cluster Computing · Computer Science 2018-07-23 Zhihao Jia , Matei Zaharia , Alex Aiken

Currently, the technique of numerical model-based atmospheric environment forecasting has becoming mature, yet traditional numerical prediction methods struggle to balance computational costs and forecast accuracy, facing developmental…

Atmospheric and Oceanic Physics · Physics 2025-04-29 Shengxuan Ji , Yawei Qu , Cheng Yuan , Tijian Wang , Bing Liu , Lili Zhu , Huihui Zheng , Zhenfeng Qiu , Pulong Chen

Modern learning models are characterized by large hyperparameter spaces and long training times. These properties, coupled with the rise of parallel computing and the growing demand to productionize machine learning workloads, motivate the…

Machine Learning · Computer Science 2020-03-17 Liam Li , Kevin Jamieson , Afshin Rostamizadeh , Ekaterina Gonina , Moritz Hardt , Benjamin Recht , Ameet Talwalkar

In this paper, we present Pangu-Weather, a deep learning based system for fast and accurate global weather forecast. For this purpose, we establish a data-driven environment by downloading $43$ years of hourly global weather data from the…

Atmospheric and Oceanic Physics · Physics 2022-11-07 Kaifeng Bi , Lingxi Xie , Hengheng Zhang , Xin Chen , Xiaotao Gu , Qi Tian

Hybrid modeling combining data-driven techniques and numerical methods is an emerging and promising research direction for efficient climate simulation. However, previous works lack practical platforms, making developing hybrid modeling a…

Atmospheric and Oceanic Physics · Physics 2022-09-20 Xin Wang , Wei Xue , Yilun Han , Guangwen Yang

Scaling models has led to significant advancements in deep learning, but training these models in decentralized settings remains challenging due to communication bottlenecks. While existing compression techniques are effective in…

Machine Learning · Computer Science 2025-06-03 Sameera Ramasinghe , Thalaiyasingam Ajanthan , Gil Avraham , Yan Zuo , Alexander Long

This paper presents a novel framework to accelerate route prediction in Drone-as-a-Service operations through weather-aware deep learning models. While classical path-planning algorithms, such as A* and Dijkstra, provide optimal solutions,…

Machine Learning · Computer Science 2026-01-08 Kamal Mohamed , Lillian Wassim , Ali Hamdi , Khaled Shaban

Applications in emerging domains such as XR are being built as compound inference systems, where multiple ML models are composed in the form of a task graph to service each request. Serving these compound systems efficiently raises two…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-03-11 Sriram Devata , Rahul Singh , Sarita Adve

As the artificial intelligence community advances into the era of large models with billions of parameters, distributed training and inference have become essential. While various parallelism strategies-data, model, sequence, and…

Machine Learning · Computer Science 2025-03-13 Ruifeng She , Bowen Pang , Kai Li , Zehua Liu , Tao Zhong

The introduction of large, foundational models to computer vision has led to drastically improved performance on the task of semantic segmentation. However, these existing methods exhibit a large performance drop when testing on images…

Computer Vision and Pattern Recognition · Computer Science 2023-12-18 Blake Gella , Howard Zhang , Rishi Upadhyay , Tiffany Chang , Matthew Waliman , Yunhao Ba , Alex Wong , Achuta Kadambi

As artificial intelligence (AI) continues to rapidly evolve, the realm of Earth and atmospheric sciences is increasingly adopting data-driven models, powered by progressive developments in deep learning (DL). Specifically, DL techniques are…

Machine Learning · Computer Science 2023-12-07 Shengchao Chen , Guodong Long , Jing Jiang , Dikai Liu , Chengqi Zhang

Global climate models parameterize a range of atmospheric-oceanic processes like gravity waves, clouds, moist convection, and turbulence that cannot be sufficiently resolved. These subgrid-scale closures for unresolved processes are a…

Atmospheric and Oceanic Physics · Physics 2025-09-05 Aman Gupta , Aditi Sheshadri , Sujit Roy , Johannes Schmude , Vishal Gaur , Wei Ji Leong , Manil Maskey , Rahul Ramachandran

The rapid adoption of AI in Earth system science promises unprecedented speed and fidelity in the generation of climate information. However, this technological prowess rests on a fragile and unequal foundation: the current trajectory of AI…