[论文解读] High-level, high-resolution ocean modeling at all scales with Oceananigans

Gregory L. Wagner, Simone Silvestri+13 more|ArXiv.org|Feb 19, 2025

Methane Hydrates and Related Phenomena被引用 3

一句话总结

Oceananigans 提供面向 GPU 的、基于 Julia 的可编程接口，用于从毫米级湍流到全球环流的海洋运动模拟，实现跨尺度的高保真、可扩展建模。

ABSTRACT

We describe the user interface, governing equations, and numerical methods underpinning the community ocean modeling software called "Oceananigans". Oceananigans development has been lead by the Climate Modeling Alliance to build a trainable climate model with quantifiable uncertainty. Oceananigans is written in the Julia programming language, which, like similar recent efforts based on modern programming languages, distinguishes it from usual software based on Fortran. Oceananigans can efficiently simulate all scales of ocean motion, ranging from millimeter-scale turbulence in a small box to planetary-scale ocean circulation. Oceananigans design combines (i) a basic structured finite volume algorithm (ii) optimized for high-resolution simulations on GPUs which is (iii) exposed behind a high-level, programmable user interface. This design negotiates a dual mandate for highest-possible performance (to support state-of-the-art applications) and enhanced accessibility (to facilitate adoption and development). The dual mandate aims ultimately to accelerate the progress of Earth system science. Achieving this aim, however, requires a substantial and sustained increase in the collective effort of Oceananigans development.

研究动机与目标

• 描述 Oceananigans 用户界面、运动方程和数值方法在全尺度海洋建模中的应用。
• 演示“高级接口加上 GPU 优化数值”如何实现跨尺度的高分辨率模拟。
• 展示在一个共同框架内耦合静水与非静水动力学的能力。
• 论证现代编程语言如何加速地球系统建模中的模型开发和不确定性量化。

提出的方法

• 将 Oceananigans 作为一个基于 Julia 的库介绍，采用四段式脚本工作流（网格、模型、仿真、后处理）。
• 采用面向 GPU 的基本结构化有限体积方法，以实现高分辨率模拟。
• 提供非静水和静水模型，具有灵活的网格（直角坐标、曲线坐标、立方球坐标）和多种垂直坐标。
• 提供可编程接口，允许在仿真中嵌入用户定义的强迫、边界条件、示踪量和封闭关系。
• 结合先进的数值选项（WENO 对流、各种扩散/闭合方案、压力求解器）并通过 ClimaOcean 支持耦合组件。
• 通过示例（2D 湍流、被动示踪量、圆柱周流）和惰性表达树用于诊断，展示可扩展性。

实验结果

研究问题

• RQ1 现代的脚本优先接口如何实现跨尺度海洋模型的快速原型设计与执行？
• RQ2 需要哪些数值策略（如 WENO 对流、压力修正、闭合式）才能在 GPU 加速硬件上准确模拟非静水和静水动力学？
• RQ3 在保持易用性的前提下，基于 Julia 的框架能在多大程度上提供与传统 Fortran 代码相当或更优的性能？
• RQ4 将 Oceananigans 与海冰和大气组件耦合用于端到端地球系统仿真的潜力有多大？
• RQ5 有哪些基准测试和性能指标能够证明 Oceananigans 有能力加速与气候相关的高分辨率仿真？

主要发现

• Oceananigans 提供可编程的库风格接口，支持在 GPU 上进行复杂流动的一键式仿真（如 2D 湍流）。
• 该系统在直角坐标和曲线坐标网格上支持非静水和静水建模，且具有多种垂直坐标选项和闭合关系。
• 面向 GPU 的结构化有限体积离散化实现高性能，相较传统 CPU 模型有显著加速，甚至实现近全球分辨率的模拟。
• 该框架在灵活、脚本驱动的工作流中整合强迫、示踪量和诊断，支持实时分析和可配置边界。
• 初步性能基准显示显著的效率提升，使高分辨率模拟成为可能，具备以中尺度分辨辨全局气候规模研究的潜力。
• Oceananigans 旨在通过可训练参数化与可微分工作流来实现集合标定和不确定性量化。