Переводчик на важных переговорах не просто быстро переводит слова, но следит, чтобы смысл не потерялся. Так и CAMulator: он берёт сложную климатическую модель CAM6 и переводит её на язык нейросети, которая работает в сотни раз быстрее, при этом не теряя физической сути. Раньше такие «переводчики» либо упрощали картинку, либо забывали про законы сохранения, и их прогнозы рассыпались при долгом счёте. CAMulator стал первым, кто держит и полное разрешение, и строгие правила физики.
Секрет в архитектуре WXFormer: она сначала училась предсказывать погоду на шаг вперёд, а затем её научили исправлять ошибки так, чтобы масса воздуха, влага и энергия не уплывали. Это как переводчик, который после каждой фразы проверяет, не потерял ли он важную деталь. Обучение заняло всего три дня на четырёх видеокартах, но теперь модель стабильно работает бесконечно долго.
Конечно, без изъянов не обошлось. Когда CAMulator заставляют смотреть в будущее, где океан теплее на два или четыре градуса, он в полярных областях рисует слишком холодную зиму. Причина проста: в его обучении не было интерактивного морского льда, и он не знает, как лёд отвечает на потепление. Это как переводчик, который отлично знает бытовую речь, но теряется в технических терминах.
Теперь, когда переводчик готов, открываются новые возможности. Можно быстро создавать сотни вариантов будущего климата, оценивать неопределённости и проверять гипотезы, на которые раньше не хватало компьютерного времени. А если соединить CAMulator с моделями океана и льда, он перестанет ошибаться в полярных широтах и станет ещё точнее. Так быстрый переводчик помогает понять, что ждёт планету через десятилетия.
CAMulator обучался всего три дня, но способен симулировать 480 лет климата за один день на одном графическом процессоре. Для сравнения, CAM6 на суперкомпьютере Derecho выдаёт 14 лет в день на 10 процессорных узлах.