首页 / 世界奇闻

AI探索宇宙结构新突破!超精准场级模拟,半秒完成冷暗物质仿真

2024-09-21 12:05世界奇闻

编辑:alan

【新智元导读】AI开启模拟宇宙!近日,来自马克斯·普朗克研究所等机构,利用宇宙学和红移依赖性对宇宙结构形成进行了场级仿真,LeCun也在第一时间转发和推荐。

下面的两组动图展示了计算机对于宇宙形成的模拟:

其中一行是根据物理定律计算的,而另一行是由人工智能学习后生成的。

你能看出来哪一个结果出自AI之手吗?

这就是来自马克斯·普朗克研究所等机构发表的工作:利用宇宙学和红移依赖性对宇宙结构形成进行场级仿真。

AI终于开始模拟宇宙了!LeCun也在第一时间转发和推荐:

作者表示:我们现在可以使用人工智能在多个时间步上对大多数宇宙学特性生成宇宙的冷暗物质模拟。

并且,物理学是这个神经网络设计的核心,它可以看成是PINN(内嵌物理知识神经网络)的一种实现,其损失函数建模了时间相关的粒子坐标和速度之间的特定关系。

在这项工作中,研究人员提出了一个用于大规模结构的场级模拟器,捕获宇宙学依赖性和宇宙结构形成的时间演化。

模拟器将线性位移场映射到特定红移处对应的非线性位移。

模拟器是一个神经网络,包含对Ω和红移z处的线性增长因子D(z)的依赖性进行编码的样式参数。

研究人员在六维N-body相空间上训练模型,将粒子速度预测为模型位移输出的时间导数,显著提高了训练效率和模型准确性。

最终,模拟器在测试数据(训练期间未见过的各种宇宙学和红移)上实现了良好的精度和性能,在z = 0,k ∼ 1 Mpc/h的尺度上达到了百分比级精度,并在较高红移下提高了性能。

通过合并树将预测的结构形成历史与N体模拟进行比较,可以找到一致的合并事件序列和统计特性。

并且,该模拟器速度极快,在单个GPU上半秒内就能够预测128的立方个粒子的非线性位移和速度场。

同时又可以通过多GPU并行处理进行良好的扩展,支持任意大尺寸的实现。

模拟宇宙的AI

随着宇宙学数据分析推向更小的尺度,利用高阶统计数据,并实现场级分析和基于模拟的推理方案,为预测非线性宇宙结构形成提供了更准确的方法。——当然也对算力提出了超高要求。

比如利用N点统计的传统分析方法,需要大量模拟数据集来进行准确的协方差估计。

而基于模拟的推理方法和场级分析,则需要生成许多后期密度场的准确实现,以约束模型参数和初始条件重建。

DESI、Euclid、Vera C. Rubin天文台、SPHEREx和Subaru Prime Focus Spectrograph可以为研究者提供大量最新的星系巡天数据。

为了探明宇宙学参数和初始条件的最佳约束,需要对巡天观测值进行快速、高度准确的预测。

在这项工作中,作者通过添加红移依赖性和对多个红移模拟快照的训练来扩展场级N体模拟器。

由于本模型的时间依赖性和自可微性,研究者可以有效地获得N体粒子速度作为输出粒子位移的时间导数。

可以在训练期间动态评估这些速度,由此定义一个取决于粒子位置和速度的损失函数,在六维N体相空间上进行训练。

强制执行「速度必须等于位移时间导数」的物理约束,可以提高训练效率并提高模型的准确性,特别是对于速度场。

模型结构

作者通过周期模拟框中的坐标x来描述N体粒子。每个粒子都与规则立方晶格上的一个位点q相关联,因此它在红移z处的位置定义为:

这里Ψ是位移场,q是粒子的拉格朗日坐标。在线性Zeldovich近似 (ZA) 中,位移场演变为:

其中 D(z) 是线性增长因子,zi是早期选择的红移,以便线性理论可以很好地描述位移场。

随着引力簇的非线性在后期变得重要,这种线性近似变得不准确,而模拟宇宙结构形成的非微扰方法(如N体模拟)变得必要:

这里f(z)为线性增长率,H(z) 是哈勃率(Hubble rate),使用粒子速度来模拟星系探测中的红移空间扭曲。

研究人员设计了场级模拟器,根据目标红移处的ZA位移场来预测z = 3-0范围内任何红移处的非线性粒子位移和速度。

场级模拟器采用U-Net/V-Net设计,使用PyTorch的map2map库实现和训练模型。

如上图所示,模型的输入具有三个通道,对应于所需红移处ZA位移的笛卡尔分量,排列在3D网格中。

输入经过四个ResNet 3×3×3卷积,第一个卷积运算将3个输入通道转换为64个内部通道。

在四次卷积操作之后,结果的副本被存储以供网络的上采样端使用,然后使用2×2×2卷积核对结果进行下采样。

该架构的感受野对应于给定焦点单元两侧的48个网格点。预测单个粒子的位移时,以焦点粒子为中心的大小为97的区域需要通过网络,对应于拉格朗日体积189.45 Mpc/h。

不过,由于网络缺少填充区域中所有单元的信息,因此这些区域的粒子位移并不准确,需要从输出中移除。

网络有限的感受野也有一个优点:它在线性理论准确的大尺度上保留了ZA场。

以上的操作可以针对固定宇宙学的单个红移来训练来自模拟快照的数据。

为了扩展网络功能,允许网络学习N体映射作为Ω和红移的函数,作者对其进行了增强以包含样式参数 。

在执行任何卷积(包括下采样/上采样操作)之前,快照的Ω和D(z)值将传递并映射到与卷积核尺寸匹配的内部数组,然后使用这些参数调节网络权重。

模型训练

训练数据

研究人员从一组具有不同宇宙学参数和一组固定快照红移的模拟中随机采样快照,同时训练样式参数和网络参数,使用Quijote Latin超立方体模拟,在边长1 Gpc/h的空间中使用512个粒子运行。

所有这些模拟的拉格朗日空间分辨率均为1.95 Mpc/h,整个数据集包含2000个模拟,每个模拟都有一组独特的五个ΛCDM宇宙学参数Ω。

研究人员将2000个模拟分为三组:1874个用于训练,122个用于验证,4个用于测试。为了鼓励各向同性,这里使用数据增强,通过立方体的对称性随机变换输入和目标数据。

损失函数

模型训练使用的损失函数包含四个项。第一个是粒子位移的平均平方误差(MSE),比较粒子的模拟器位移预测和真实的N体位移:

第二项是欧拉密度的MSE:

第三项是采用粒子速度的MSE:

最后一项对应于欧拉动量场的MSE损失。这里将粒子速度分布到与初始拉格朗日网格具有相同分辨率的欧拉矢量场网格并计算,p是网格单元中每粒子质量的欧拉动量。

于是,红移z处快照的最终损失函数为:

实验结果

研究人员通过构建模拟器输出和N体模拟真值的欧拉密度、和动量自功率谱和互功率谱来评估模拟器的准确性。

对于密度场,使用CIC插值方案将粒子分布到512网格并估计功率谱,下表列出了用于测试本文模拟器的五个模拟宇宙学参数:

模拟器在训练过程中从未遇到过以上5种测试模拟中的任何一种。

下面测试在训练数据中的五个固定红移之间进行插值时的模型性能:

上图显示了SNN模拟器的功率谱误差(比例函数),每条曲线的颜色表示红移。

欧拉密度误差仅源于粒子位移的误差,当红移z = 0时,新的瞬态模型的随机性与原始模型的随机性相当,并且传递函数误差通常比原始模型有所改善。

在上图的最右列中,可以看到由于模拟器无法完美预测BAO幅度而导致的振荡误差。不过模拟器的这些错误特征低于1%,并且可能会随着更多的训练数据而得到改善。

红移相关模型的性能与z = 0时的真实空间密度统计数据相当,并且在较高红移时对于红移空间和真实空间统计数据的性能明显更好。

随着红移的减小,误差平滑且单调地增加。这表明模拟器可以在其训练数据中的少量固定红移快照之间有效地进行插值,而不会过度拟合,否则我们会在中间看到错误的振荡特征。

参考资料:

https://x.com/cosmo_shirley/status/1825749316134158627

猜你喜欢

  • 世界奇闻

    詹姆斯韦伯望远镜发现宇宙最早星系:巨大得令人惊叹

    人类对宇宙的探索之旅历经千难万险,而在这无尽的追寻中,詹姆斯韦伯太空望远镜堪称一座里程碑式的成就。作为有史以来最强大、最复杂的空间观测站之一,韦伯望远镜凭借其卓越的红外观测能力,正在逐步揭开宇宙深层的秘密。最近,这座科学巨擘再次引起了世人的关注,因为它揭示了两个迄今为止发现的最早、最遥远的星系,它们..

    2025-07-08
  • 世界记录

    钻石联赛尤金站 诞生两项世界纪录 女子5000米突破14分大关

    北京时间7月6日,在美国进行的2025世界田联钻石联赛尤金站比赛诞生两项新的世界纪录。肯尼亚选手切贝特以13分58秒06的成绩打破女子5000米世界纪录,她的同胞基普页真在女子1500米比赛中以3分48秒68的成绩刷新自己保持的世界纪录。切贝特 世界田联官网供图巴黎奥运会女子10000米、5000米双料冠军切贝特去年就是在尤金海沃德..

    2025-07-08
  • 世界记录

    连续打破世界纪录,她们突破女性运动员新极限

    在尤金举行的钻石联赛(普雷方丹精英赛)中,两位来自肯尼亚的女子中长跑选手比阿特丽斯・切贝特和费思・基普耶根,通过在女子5000米以及女子1500米这两项比赛中的惊艳表现,不仅将这两个项目的女子世界纪录带进全新纪元,更携手让全世界看到了女子运动员的能量和独属于她们的魅力。切贝特女子5000米首破14分大关在北京时间..

    2025-07-08
  • 国产机器狗速度突破每秒10米 刷新世界纪录

    【国产机器狗速度突破每秒10米 刷新世界纪录】财联社7月7日电,在“智绘光谷”CMG世界机器人技能大赛的首赛上,“新一代黑豹2.0”凭借卓越性能与前沿技术,以10.3米/秒的速度打破由波士顿动力WildCat创造的8.89米/秒的机器狗纪录,成功改写了沉寂十年之久的世界纪录。..

    2025-07-08
  • 探索百科

    平行宇宙真的存在吗 至今是无法证明(没有证据)

    关于平行宇宙的问题至今是无法证明的,没有任何证据。现如今人们会利用各种科学技术去了解宇宙,然而就发现宇宙是由一个奇点大爆炸而引发现如今宇宙依然在不断的膨胀,因此很多科学家就觉得在宇宙当中有可能还会存在平行宇宙。不过这一直以来都是未解之谜也困扰着很多科学家。平行宇宙真的存在吗关于平行宇宙真的存在吗,这..

    2025-07-08
  • 世界奇闻

    球幕科普电影《何以知天》在京首发 邀公众探索宇宙

    中新网北京7月5日电 我国首部以天文学家探索宇宙的方法论为主题的球幕科普电影《何以知天》首发活动4日在北京天文馆宇宙剧场举办。该片历时一年打磨,将在暑期登陆北京天文馆,为公众揭开“人类如何认知宇宙”的神秘面纱。图为球幕科普电影《何以知天》海报。北京天文馆供图宇宙浩瀚无垠,人类为何能跨越时空限制,描绘出百..

    2025-07-07
  • 排行榜

    电饭煲品牌排行榜前十名:精准测评教你怎么选“真香”机型

    在当下快节奏的生活中,电饭煲早已成为厨房不可或缺的核心家电,而消费者对其的需求也从“能煮饭”升级为“煮好饭”与“多功能体验”的双重追求。而优质电饭煲需具备四大特征:一、焖煮性能:可依米种、数量自动调参数,保障口感营养。二、加热均匀:立体加热技术,避免煳底、夹生问题。三、内胆:形状符合米饭膨胀规律,涂..

    2025-07-06
  • 排行榜

    电风扇十大排名分享:2024王牌机型精准种草

    正值夏天季节,一款优质的电风扇当然必不可缺。但如今电风扇市场各种产品令人眼花缭乱,其中有些劣质的产品容易吹起细菌、病毒、宠物毛发、颗粒物,从而污染了空气,而且风力刺激性大,容易刺激头皮吹得头晕头痛。那如何从中选购到高品质的电风扇呢?作为一名家电测评师,今天我就给大家分享我实测的电风扇十大排名,希望其..

    2025-07-05

微信分享

微信分享二维码

扫描二维码分享到微信或朋友圈

链接已复制