原标题：在TensorFlow里模拟宇宙！提升模拟计算速度，成本降低1000倍
大数据文摘出品
来源：tensorflowblog
编译：睡不着的iris、刘俊寰
去年6月，俄罗斯天文学家兼码农小哥开发了一款名叫Space Engine的宇宙模拟游戏，目前，该游戏已经收到2639份测评，绝大部分都是好评。
不过，Space Engine还只是在游戏层面对宇宙的一种呈现，试想以下，如果用数值模拟的基础工具，从观测的海量宇宙数据进行分析……
停！看到这儿也大概能猜到，这样的模拟计算肯定代价不菲，而且通常需要在大规模的超级算计算机上离线运行。
但是，若是提升模拟计算速度，将其与机器学习组件整合到统一框架内，会有什么不同呢？
这便是新的N体宇宙模拟代码FlowPM在做的。
没错，文摘菌今天给大家展示的就是，如何使用TensorFlow模拟自己的小宇宙，并说明为什么它对宇宙学有着激动人心的前景展望，搓搓小手跟进了~
图1：（蓝色）通过2dFGRS测量的宇宙结构，（红色）使用N体模拟生成的千禧年宇宙结构。
N体宇宙模拟
简单来说，N体模拟的方法是为了解释宇宙演化中的非线性演化问题，科学家们提出的一套研究宇宙学的数值模拟方法。
在宇宙学领域，N体模拟还原了宇宙是如何从诞生之初的大爆炸演变至今的。
如果我们按照物质初始分布，将大量的粒子放置在一个盒子中，由于引力的牵引，粒子会不断移动。而最终的粒子状态，就是对浩瀚宇宙结构的再现。例如，从这个分布中，你可以看到星系和星系群是如何分布的。
这些结构的形成和演化取决于一些与宇宙相关的基础问题，诸如宇宙中有多少物质？宇宙膨胀有多快？导致宇宙膨胀的黑物质和暗能量有哪些特性？
现代的宇宙学研究用强大的望远镜覆盖了辽阔的观测区域，绘制出了这些结构，时间甚至可以追溯到几十亿年前。通过将N体模拟的预测与观测数据进行比较，可以帮助我们解答前述基础问题，从而提高对宇宙的诞生和演化的理解。
FlowPM：一个TensorFlow的宇宙N体求解程序
随着几十年研究的更迭，N体模拟已是宇宙学界的主要研究工具。
得益于机器学习和统计推理领域的最新进展，一个完全基于TensorFlow实现的N体宇宙模拟程序FlowPM应运而生了。
在这里，我们使用Google Colab notebook对其进行实验，同时，以下主要提供TF1的编写代码。
首先需要设置好边长、网格大小和执行代码片段等模拟参数，随后的执行模拟代码片段就非常简单了，完整的代码如下所示，如此就能生成下图所示的大型结构。
# Generate Gaussian initial conditions for the matter distribution
initial_conditions = flowpm.linear_field(N, L, ipklin, batch_size=batch)
# Sample particles, i.e. generate the initial displacement and velocity
state = flowpm.lpt_init(initial_conditions, a0=a0)
# Evolve particles from initial state down to present time with N-Body simulations
final_state = flowpm.nbody(state, stages, N)
# Visualize final density field i.e interpolate the particles to a grid
Final_field = flowpm.cic_paint(tf.zeros_like(initial_conditions), final_state[0])
#Execute the graph!
with tf.Session() as sess:
ic, istate, fstate, sim = sess.run([initial_conditions, state, final_state, final_field])
图2 ：（左）N体模拟启动时，宇宙中物质的初始分布；（右）在模拟的最终快照，物质的最终分布。可以清楚地看到大型结构，具有晕圈，长丝和空洞塌陷。
或许不少人会疑惑，TensorFlow真有这么香？文摘菌把话撂这，还这真有这么香，TensorFlow的优势主要体现在以下两个方面。
首先是分析和推理，TensorFlow模拟为宇宙学家提供了前所未有的独特功能，也就是我们所说的可微性（differentiability）。这个功能为科学家提供一种新的分析工具，例如开发基于模拟的有效推理技术。
不仅如此，这个功能还支持量化各类输入参数的调整对最终观察的影响。在伯克利宇宙中心，研究者们对时光倒流和重构初始状态十分感兴趣，这也有助于更好地了解现在的宇宙结构，但这是有前提的，必须先解决一个百万维度的高度非线性优化问题，即模拟过程得可微，而这，FlowPM就能做到，重构后的图如下所示。
从最终的黑物质的大型宇宙结构（右边）如何重建宇宙初始状态（左边）
另一个就是混合物理/深度学习模拟，在N体模拟运用深度学习组件时，你会需要开发一个混合前向模型。但是，利用当前的计算资源，不可能高度准确地同时模拟所有的组件，只能依据科研目标，在不同元素之间进行取舍，例如宇宙可观察的深度、星系可见的范围和形成星系的多维物理过程等。
然而现在，我们可以用深度学习替代模型，将这些元素自然地整合进N体模拟中。在BCCP，科学家们开发了一个这样的混合模拟，如下图所示。假如我们要模拟宇宙中的气体动量密度，目前，这类观察采用混合模拟实现代价非常大，但使用一个端到端的FlowPM模型，成本至少可以降低1000倍。
一个混合模拟的示例，用两层的非线性变换（网络）补充PM模拟的暗物质输出（左），以模拟宇宙（中心）的气体动量密度。将使用昂贵1000倍的混合模拟所模拟与真相进行比较（右）
采用Mesh TensorFlow分布式框架仿真
虽然原理看上去很简单，但实际的挑战还是很大的，毕竟宇宙真的很庞大！
为了准确匹配当前和未来的观测，我们要同时模拟数以亿计的粒子的变化。
要实现这个，N体宇宙模拟主要有以下两个难点：
Mesh TensorFlow框架可以轻松地使用分布式张量来表示模拟，并持续追踪分布式梯度和设备之间的存储通信，通过Mesh TensorFlow编写N体求解程序，可以将巨大的仿真量跨多个超级计算机进行分布。
在这种模拟中，每一步操作和网格组件在一个时间步长可以演化出不同的空间区域。在云端TPU部署相同的模拟代码，可以模拟128个128x128x128网格的独立宇宙，或者部署在国家计算机（NERSC）的64个GPU上模拟一个1024x1024x1024的宇宙。
除了支持大规模仿真，模型并行框架允许通过将计算任务分配给多个处理器来加速中间尺寸的模拟，如下图所示，FlowPM的平均模拟速度比当前的可微python模拟器FastPM快了40倍不止。
使用FastPM（在Cori Haswell内核上运行基于CPU的python代码）和FlowPM（在Cori GPU上运行基于GPU的Mesh TensorFlow代码）模拟2563网格PM，比较1个步长内时间缩放与处理器数量
展望
过去的三十多年，数值模拟是大型宇宙的研究基石。现在，有了FlowPM，研究即将迈出新的一步，将模拟器集成在包含深度学习组件的统一框架内，同时保持对潜在现象的准确物理理解。
在宇宙学中，这种结合不仅促生了新颖的分析工具，在研究体系中加入深度模型攻克棘手的历史难题。这些活跃的研究领域，加速催生下一代宇宙研究，在未来的十年之际可以观察宇宙中成千上万的个体。
多亏了Mesh TensorFlow模型并行框架，使得物理建模和机器学习的融合得以实现，希望成分分析和FlowPM计算工具的发展，让宇宙学以外的领域也得以受益。
相关报道：
https://blog.tensorflow.org/2020/03/simulating-universe-in-tensorflow.html游戏网
| 相关下载 |
《英雄联盟》宇宙官方微博今日凌晨再次公布了新的动画短片，这次故事的主人公是来自暗影岛的魂锁典狱长—锤石。《英雄联盟》宇宙官方微博今日凌晨再次公布了新的动画短片，这次详情>>
炉石传说手游中很多小伙伴不知道标准宇宙贼卡组搭配，也不知道宇宙贼卡组攻略，今天小编就带着大家了解一下炉石传说标准宇宙贼卡组攻略;炉石传说标准宇宙贼；AAECAYO6Ah60AZoCsgL详情>>
炉石传说手游中很多小伙伴不知道标准宇宙贼卡组搭配，也不知道宇宙贼卡组攻略，今天小编就带着大家了解一下炉石传说标准宇宙贼卡组攻略; 炉石传说标准宇宙贼； AAECAYO6Ah60AZo详情>>
炉石传说手游中很多小伙伴不知道宇宙猎卡组搭配及玩法，也不知道宇宙猎怎么玩，今天小编就带着大家了解一下炉石传说新版本宇宙猎详解;炉石传说宇宙猎阵容： 恶魔猎手### 宇宙猎# 详情>>
炉石传说手游中很多小伙伴不知道宇宙猎卡组搭配及玩法，也不知道宇宙猎怎么玩，今天小编就带着大家了解一下炉石传说新版本宇宙猎详解; 炉石传说宇宙猎阵容： 恶魔猎手### 宇宙详情>>
炉石传说中恶魔猎手组成的宇宙猎卡组非常强，那么宇宙猎卡组怎么玩呢？下面小编就给大家带来了炉石传说宇宙猎卡组搭配及玩法攻略，一起来看看吧。宇宙猎卡组搭配职业：恶魔猎手# 模详情>>
原标题：什么是平行宇宙？ 作者：知否KnowHow，题图来自：《星际迷航》剧照 由于最近爱情烧脑剧《想见你》的热播，引起了人们对时空的讨论。有人说黄雨萱与详情>>
原标题：漫威漫画：新《刀锋战士》或定档2022年10月，新成员加入漫威宇宙 漫威宇宙中，存在着各式各样的漫画角色，其中有一位性格独树一帜的超级英雄，一直被详情>>
原标题：《龙珠英雄》创生篇情报，悟空、比鲁斯宇宙空间大战，宇宙树苗将长成星云巨树 《龙珠英雄》动画和漫画都是根据《超级龙珠英雄》游戏改编的作品详情>>
4月8号，炉石传说全新年份“凤凰年”开始，新版本“外域的灰烬”即将到来。不过和往常不一样的是，今年增加了回调buff环节，这也给了那些即将退环境的“详情>>
原标题：晚安 | 痴迷是年轻人的游戏 晚安｜ 第一千五百零一夜 来自《致命魔术》 音 乐 如果宇宙有声音，那会是怎么样的旋律呢？ 「痴迷是年轻人的游详情>>
关于这款游戏 《宇宙沙盘》是一款基于物理法则的宇宙模拟器，可以让你在庞大的宇宙尺度上进行创造或者毁灭。它能够实时模拟重力、星球气候、天体详情>>
原标题：宇宙模拟器！《宇宙沙盘》Steam促销 仅售60元 关于这款游戏 《宇宙沙盘》是一款基于物理法则的宇宙模拟器，可以让你在庞大的宇宙尺度上进行创详情>>
原标题：78民间测评：HCM-PRO 铁人宇宙型/宇宙指挥官型 MSJ-06Ⅱ-ET 作者：阿爸哇哭的巴尼 文章转自78DM 作为HCM-PRO这个系列后期真正集大成的产品之一详情>>
导 读 炉石传说手游宇宙魔块术卡组怎么搭配，相信不少玩家都有这样的疑问，今天小编就为大家带来炉石传说手游宇宙魔块术卡组搭配攻略，希望可以帮到玩家。 炉石传说手游宇宙魔块详情>>
大家好，欢迎阅读234游戏的小编带来的游戏资讯，这一篇小编将跟大家分享一下《无人深空——打造宇宙最帅的舰队需要多少钱？》的相关视频，快跟着234游戏的小编一起来看看详情>>
随着巨龙降临版本走向末端，巨龙年的帷幕也将落下，一年一度的荣誉室也即将加入新的卡牌，让我们重点分析一下，有哪些卡会...等等！跑错片场了！咳咳，从头说详情>>
炉石传说这两天也是把新冒险迦拉克隆的觉醒中的第4章也就是最后一章给开放了，相信很多小伙伴都已经通关了吧，那么这个模式的新卡也是全部都能拿到了，详情>>
导 读 炉石传说宇宙龙猎卡组推荐。宇宙龙猎是巨龙降临版本新晋崛起的一套狂野卡组，新卡腐巢幼龙的出现让这套体系强度有了进一步的提升。接下来就让我们一起来了解一下宇宙详情>>
本栏目定位是外/国服直播卡组搬运 最后一只脏牧传说前百宇宙牧，并不是强度卡组缺卡勿合，打到前百是出于自己的职业理解 也可以换一些JK组件进来详情>>
创新思路之宇宙心火牧 宇宙心火牧 机械骑 宇宙JK萨 任务术 ◆宇宙心火牧 AAECAa0GHvgC5QT2B9UIpQnRCtIK8gz3DPsM5fcC0P4CoIAD3oIDp4cD2JgD详情>>
比利君的JK贼，直播时最高打到国服11，带了新卡213强无敌。弥补了前期盗贼的亏空的曲线。 AAECAaIHCLICrwSggAOSlwPBrgOqrwPjtAPLwAMLtAHtAs0DiAe详情>>
据外媒报道，乐高发布了官方的国际空间站套装，其中包括一个轨道平台的比例模型、一个可停靠的小型航天飞机、一个可部署的卫星和两个宇航员的迷你人偶。整个套装详情>>
炫酷的肉搏对决、惊险的高空索降、以及“海、陆、空”三线全景展现的视效奇观，是电影《特警队》在刚刚过去的12月带给粉丝朋友详情>>
本栏目定位是外/国服直播卡组搬运 童话镇登顶JK贼，来自国服第一的卡组，这几天JK贼的强度越来越高，不管是国服还是外服，登顶的几乎一水的JK贼 不过详情>>
原标题：我脏牧又回来啦！标准宇宙脏牧解析 谁说脏牧就没有胜率？ ●前言 快乐炉石从我做起，大家好这里是雪叔。 随着版本的更新，萨满被削弱，天梯环境进一详情>>
谁说脏牧就没有胜率？ ●前言 快乐炉石从我做起，大家好这里是雪叔。 随着版本的更新，萨满被削弱，天梯环境进一步更新，996马云贼被捧上神坛，但是从整体详情>>
原标题：圆谷的奥特曼宇宙和漫威的超级英雄宇宙，为什么漫威的这么受欢迎？ 相信很多小伙伴都是知道圆谷和漫威这两家公司的，这两家公司一个是拍摄特摄剧详情>>
原标题：外服卡组推荐：宇宙龙德/迦拉克隆园/克战/控制萨 德鲁伊还有希望崛起吗？ 大家好！我是：曼曼曼曼曼。这里是外服直播卡组搬运，每天都会给大家精选外详情>>
德鲁伊还有希望崛起吗？ 大家好！我是：曼曼曼曼曼。这里是外服直播卡组搬运，每天都会给大家精选外服的四套强力有新意的卡组，请大家多多支持，不认识的主播详情>>
作为经典复刻石器时代2.5版的手游《宇宙起源》，不仅做到了对石器时代2.5版本极高的还原度，更是在此基础上做出人性化的改动。熟悉的“石器味儿”里有更多不一样的奥秘，值得我们呼朋唤友一起去探索。今天要说的就是万众期待的官方角色交易系统。为更好的为玩家规避交易风险，让新老玩家氪金更安心，《宇宙起源》即将开启的官方角色交易平台，为玩家交易保驾护航！详情>>
近期炉石传说中宇宙龙法备受关注，宇宙龙法怎么玩呢？宇宙龙法卡组如何搭配呢？今天小编就带来一篇炉石传说宇宙龙法攻略大全 宇宙龙法卡组搭配一览，一起来看看吧：宇宙龙法 ### 宇宙详情>>
原标题：“原力”回归，星战衍生宇宙才能觉醒 第一部星战电影《星球大战》。 《星球大战》是乔治·卢卡斯创作的经典科幻系列电影，如今版权则由迪士详情>>
《一路成年》综艺的热播，让小绵羊徐锦江在网络平台圈粉无数。如今一路成年的旅程已经结束了，但是爸爸和孩子走过的路，一起经历的美好回忆会详情>>
俗话说，遇事不决，量子力学。最近俄罗斯研究人员提出了关于宇宙的惊人理论：整个宇宙都是量子的，而且存在多个其他的宇宙，与我们宇宙进行着交互，所以从宏观上看，我们都生活在一台巨详情>>
由梁家辉、徐锦江、吴刚等一众影视大腕加盟的“亲子真人秀”《一路成年》，不知不觉已经到了第十二期，老父亲们和孩子的美好旅行详情>>
劫，一度是均衡教派的侠士，如今则带领了一群刺客，为了保护家园艾欧尼亚，甚至不惜杀害无辜!而当那个精神变态的魔法奇才卡达·烬再次现身作案，劫必须面对的旧日详情>>
星迷宇宙以IP赋能为核心，IP衍生品为基础，通过线下合作店嫁接模式，将优质 IP与线下零售相结合，已在全国多地开展多家星迷宇宙线下合作店。近详情>>
原标题：外服卡组推荐：宇宙贼/宇宙战/中速佛祖骑/任务贼 文：曼曼曼曼曼QAQ 编辑：枫尘 前言 大家好！我是：曼曼曼曼曼。这里是外服直播卡组搬运，每天都会给详情>>
炉石传说宇宙奇迹法用什么卡好？今天为大家带来的是一套成功上了传说的宇宙奇迹法卡组，下面一起来看看炉石传说宇宙奇迹法卡组推荐吧。 宇宙奇迹法卡组推荐 # 职业：法详情>>
