史上最大重建25km²!NeRF-XL:真正有效利用多卡联合训练!


原标题:nerf-xl: scaling nerfs with multiple gpus

论文链接:https://research.nvidia.com/labs/toronto-ai/nerfxl/assets/nerfxl.pdf

项目链接:https://research.nvidia.com/labs/toronto-ai/nerfxl/

作者单位:NVIDIA 加州大学伯克利分校

☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 免费无限量使用 DeepSeek R1 模型☜☜☜

史上最大重建25km²!NeRF-XL:真正有效利用多卡联合训练!

论文思路:

本文提出了NeRF-XL,这是一种原理性的方法,用于在多个图形处理器(GPUs)之间分配神经射线场(NeRFs),从而使得具有任意大容量的NeRF的训练和渲染成为可能。本文首先回顾了现有的几个GPU方法,这些方法将大型场景分解成多个独立训练的NeRFs [9, 15, 17],并确定了这些方法的几个基本问题,这些问题在使用额外的计算资源(GPUs)进行训练时阻碍了重建质量的提高。NeRF-XL解决了这些问题,并允许通过简单地使用更多的硬件来训练和渲染具有任意数量参数的NeRFs。本文方法的核心是一个新颖的分布式训练和渲染公式,这在数学上等同于经典的单GPU案例,并最小化了GPU之间的通信。通过解锁具有任意大参数数量的NeRFs,本文的方法是第一个揭示NeRFs GPU扩展规律(scaling laws)的方法,显示出随着参数数量的增加而提高的重建质量,以及随着更多的GPU的增加而提高的速度。本文在多种数据集上展示了NeRF-XL的有效性,包括包含约258K张图像、覆盖了25平方公里的城市区域的MatrixCity [5]。

论文设计:

近期在新视角合成的进步极大地提高了我们捕获神经辐射场(NeRFs)的能力,使得这一过程变得更加易于接近。这些进步使得我们能够重建更大的场景和场景内更精细的细节。无论是通过增加空间规模(例如,捕获数公里长的城市景观)还是提高细节水平(例如,扫描田野中的草叶),扩大捕获场景的范围都涉及到将更多的信息量纳入NeRF中,以实现准确的重建。因此,对于信息含量高的场景,重建所需的可训练参数数量可能会超过单个GPU的内存容量。

本文提出了NeRF-XL,这是一个原理性的算法,用于在多个GPU之间高效分配神经渐射场景(NeRFs)。本文的方法通过简单增加硬件资源,使得捕获高信息含量的场景(包括大规模和高细节特征的场景)成为可能。NeRF-XL的核心是在一组不相交的空间区域之间分配NeRF参数,并跨GPU联合训练它们。不同于传统的分布式训练流程在后向传播中同步梯度,本文的方法只需要在前向传播中同步信息。此外,通过仔细渲染方程程和分布式设置中相关的损失项,本文大幅减少了GPU之间所需的数据传输。这种新颖的重写提高了训练和渲染的效率。本文方法的灵活性和可扩展性使得本文能够够够多个GPU高效地优化和使用多个GPU高效性能优化。

本文的工作与最近采用了GPU算法来建模大规模场景的方法形成了对比,这些方法通过训练一组独立立体的NeRFs来实现[9, 15, 17]。虽然这些方法不需要GPU之间的通信,但每个NeRF都需要建模整个空间,包括背景区域。这导致随着GPU数量的增加,模型容量中的冗余度增加。此外,这些方法在渲染时需要混合NeRFs,这会降低视觉质量并在重叠区域引入伪影。因此,与NeRF-XL不同的是,这些方法在训练中使用更多的模型参数(相当于更多的GPU时),未能实现视觉质量的提升。

本文通过一系列多样化的捕获案例来展示本文方法的有效性,包括街道扫描、无人机飞越和以物体为中心的视频。这些案例的范围从小场景(10平方米)到整个城市(25平方公里)。本文的实验表明,随着本文将更多的计算资源分配给优化过程,NeRF-XL开始能够实现改善的视觉质量(通过PSNR测量)和渲染速度。因此,NeRF-XL使得在任何空间规模和细节的场景上训练工具有任意容量的NeRF成为可能。

史上最大重建25km²!NeRF-XL:真正有效利用多卡联合训练!

图 1:本文基于原理的多GPU分布式训练算法能够将NeRFs扩展到任意大的规模。

史上最大重建25km²!NeRF-XL:真正有效利用多卡联合训练!

图 2:独立训练与多GPU联合训练。独立地训练多个NeRFs[9,15,18]要求每个NeRF既要建模焦点区域也要建模其周围环境,这导致了模型容量的冗余。相比之下,本文的联合训练方法使用不重叠的NeRFs,因此没有任何冗余。

史上最大重建25km²!NeRF-XL:真正有效利用多卡联合训练!

图 3:独立训练需要在新视角合成时进行混合。无论是在2D[9, 15]还是3D[18]中进行混合,都会在渲染中引入模糊。

Canva AI Canva AI

Canva平台AI图片生成工具

Canva AI 1374 查看详情 Canva AI

史上最大重建25km²!NeRF-XL:真正有效利用多卡联合训练!

图 4:独立训练导致不同的相机优化。在NeRF中,相机优化可以通过变换不准确的相机本身或所有其他相机以及底层3D场景来实现。因此,伴随相机优化独立训练多个NeRF可能导致相机校正和场景几何的不一致性,这给混合渲染带来了更多困难。

史上最大重建25km²!NeRF-XL:真正有效利用多卡联合训练!

图 5:3D混合可能造成的视觉伪影。左图展示了使用2个GPU训练的MegaNeRF结果。在0%重叠时,由于独立训练,边界出现了伪影;而在15%重叠时,由于3D混合,出现了严重的伪影。右图阐释了这种伪影的成因:虽然每个独立训练的NeRF渲染出正确的颜色,但混合后的NeRF并不保证正确的颜色渲染。

史上最大重建25km²!NeRF-XL:真正有效利用多卡联合训练!

图 6:本文的训练流程。本文的方法联合训练所有GPU上的多个NeRFs,每个NeRF覆盖一个不相交的空间区域。GPU之间的通信仅发生在前向传播中,而不发生在后向传播中(如灰色箭头所示)。(a) 本文可以通过评估每个NeRF以获得样本颜色和密度,然后将这些值广播到所有其他GPU以进行全局体渲染(见第4.2节)。(b) 通过重写体渲染方程,本文可以将数据传输量大幅减少到每条光线一个值,从而提高效率(见第4.3节)。

实验结果:

史上最大重建25km²!NeRF-XL:真正有效利用多卡联合训练!

图 7:定性比较。与先前的工作相比,本文的方法有效地利用多GPU配置,在所有类型的数据上提高了性能。

史上最大重建25km²!NeRF-XL:真正有效利用多卡联合训练!

图 8:定量比较。基于独立训练的先前工作未能随着额外GPU的增加而实现性能提升,而本文的方法随着训练资源的增加,享受到了渲染质量和速度的提升。

史上最大重建25km²!NeRF-XL:真正有效利用多卡联合训练!

图 9:本文方法的可扩展性。更多的GPU允许有更多的可学习参数,这导致了更大的模型容量和更好的质量。

史上最大重建25km²!NeRF-XL:真正有效利用多卡联合训练!

图 10:大规模捕获上的更多渲染结果。本文在更大的捕获数据集上使用更多的GPU测试了本文方法的鲁棒性。请参阅本文的网页,以获取这些数据的视频导览。

史上最大重建25km²!NeRF-XL:真正有效利用多卡联合训练!

图 11:在University4数据集上与PyTorch DDP的比较。PyTorch 分布式数据并行(Distributed Data Parallel,DDP)旨在通过跨GPU分布光线来加快渲染速度。相比之下,本文的方法是跨GPU分布参数,突破了集群中单个GPU的内存限制,并且能够扩大模型容量以获得更好的质量。

史上最大重建25km²!NeRF-XL:真正有效利用多卡联合训练!

图 12:University4上的同步成本。本文基于分区的体渲染(见第4.3节)允许 tile-based 通信,这比原始的基于样本的通信(见第4.2节)成本要低得多,因此能够实现更快的渲染。

总结:

总结来说,本文重新审视了将大规模场景分解为独立训练的NeRFs(神经辐射场)的现有方法,并发现了阻碍额外计算资源(GPUs)有效利用的重大问题,这与利用多GPU设置来提升大规模NeRF性能的核心目标相矛盾。因此,本文引入了NeRF-XL,这是一种原理性的算法,能够有效地利用多GPU设置,并通过联合训练多个非重叠的NeRFs来在任何规模上增强NeRF性能。重要的是,本文的方法不依赖于任何启发式规则,并且在多GPU设置中遵循NeRF的扩展规律(scaling laws),适用于各种类型的数据。

引用:

@misc{li2025nerfxl,title={NeRF-XL: Scaling NeRFs with Multiple GPUs}, author={Ruilong Li and Sanja Fidler and Angjoo Kanazawa and Francis Williams},year={2025},eprint={2404.16221},archivePrefix={arXiv},primaryClass={cs.CV}}

以上就是史上最大重建25km²!NeRF-XL:真正有效利用多卡联合训练!的详细内容,更多请关注其它相关文章!


# 提高了  # 济南个人网站建设定位  # fb怎样营销推广  # 南京seo营销技巧  # seo面包店优化  # 黄山外贸seo公司  # 网店的推广和营销的作用  # 南通seo网络推广咨询价格  # SEO入门单反推荐学生  # 网站的线上推广策略  # 万达网站建设  # 模型  # 可以通过  # 所需  # 提出了  # 是在  # 几个  # 的是  # 更大  # 史上最大  # 多个  # 训练 


相关栏目: 【 Google疑问12 】 【 Facebook疑问10 】 【 优化推广96088 】 【 技术知识133117 】 【 IDC资讯59369 】 【 网络运营7196 】 【 IT资讯61894 】


相关推荐: 13条咒语挖掘GPT-4最大潜力,Github万星AI导师火了,网友:隔行再也不隔山了  Unity 推出面向开发者的 AI 软件市场 AI Hub,股价飙涨 15%  Gartner发布中国企业人工智能趋势浪潮3.0  掌阅科技对话式AI应用“阅爱聊”开启内测  一次购买全年省心,入手科沃斯这几台机器人,省下时间就是金钱  AI大模型火了!科技巨头纷纷加入,多地政策加码加速落地  尼康尼克尔Z 180-600mm f/5.6-6.3 VR镜头发布:12499元 拍鸟神器  可按用户语气自动回复消息,Zoom 推出基于生成式 AI 的新功能  北京公司实施AI技术,推行4.5天工作制,抵制996文化,提升员工工作幸福感  美图秀秀发布七款 AI 工具:修图一样修视频、打造电影级上镜脸  看似低调,实则稳健:字节在AI路上会遇到什么?  网易云音乐内测上线“私人DJ” 打造AI推荐音乐助手  美图秀秀发布7款AI产品:支持用户创作、商业创作  LinkedIn 推出生成式 AI 辅助撰写帖文功能,将向所有用户开放  看懂AI,找到增长新势能 | 笔记侠AI峰会等你来  首家承认ChatGPT影响其收入的公司Chegg选择拥抱AI ,裁减4%员工  央视报道!星纪魅族集团车载人机交互技术成世界移动通信大会焦点  华为大模型登Nature正刊!审稿人:让人们重新审视预报模型的未来  华为联合合作伙伴 共同发布昇腾AI大模型训推一体化解决方案  30+大模型齐聚,大模型成世界人工智能大会“顶流”  Intel酷睿Ultra发布会官宣!迈向全新的AI时代  ChatGPT设计出的第一个机器人来了!【附人工智能行业预测】  马斯克发推讽刺人工智能:机器学习的本质就是统计  云鲸发布全新的扫拖机器人J4系列  如何获得元宇宙的第一个属于自己的空间  【原创】奥比中光:与英伟达合作开发的3D开发套件正式发布 连接英伟达AI应用生态  AI智能室内效果图设计软件效果,确实惊到我了!  ChatGPT只讲这25个笑话!实验上千次有90%重复,网友:幽默是人类最后的尊严  两型无人机完成交付!国家级机动观测业务正式启动  英伟达CEO宣称生成式AI已迎来“划时代时刻”  了解 AGI:智能的未来?  第四范式“式说”大模型入选《2025年通用人工智能创新应用案例集》  WHEE使用教程  AI绘画,还需要懂数学?  Prompt解锁语音语言模型生成能力,SpeechGen实现语音翻译、修补多项任务  一文看懂基础模型的定义和工作原理  AI浪潮席卷,时空壶为何能成为AI翻译时代的破局者  数据显示:人工智能相关专业热度上升最快 考古、美术、生物医学工程等小众专业火了  智能技术提高现代商业运营的7七种方式  盘古大模型3.0正式发布 AI开发正走向新“工业化开发模式”  OpenAI 向所有付费 API 用户开放 GPT-4  鸿蒙4即将支持大规模AI模型  游族AI创新院揭牌成立 推进AI赋能游戏业务  Valve Index VR 头显销量下滑,上市四年的长青树渐失光彩  PHP和OpenCV库:如何实现人脸识别  “聚智启新,‘蓉’力同行” 成都市人工智能产业融通对接会成功举办  通用医疗人工智能如何革新医疗行业?  AI技术加速迭代:周鸿祎视角下的大模型战略  视觉中国推出AI灵感绘图功能  WHEE网页地址入口 

 2024-04-30

了解您产品搜索量及市场趋势,制定营销计划

同行竞争及网站分析保障您的广告效果

点击免费数据支持

提交您的需求,1小时内享受我们的专业解答。

运城市盐湖区信雨科技有限公司


运城市盐湖区信雨科技有限公司

运城市盐湖区信雨科技有限公司是一家深耕海外推广领域十年的专业服务商,作为谷歌推广与Facebook广告全球合作伙伴,聚焦外贸企业出海痛点,以数字化营销为核心,提供一站式海外营销解决方案。公司凭借十年行业沉淀与平台官方资源加持,打破传统外贸获客壁垒,助力企业高效开拓全球市场,成为中小企业出海的可靠合作伙伴。

 8156699

 13765294890

 8156699@qq.com

Notice

We and selected third parties use cookies or similar technologies for technical purposes and, with your consent, for other purposes as specified in the cookie policy.
You can consent to the use of such technologies by closing this notice, by interacting with any link or button outside of this notice or by continuing to browse otherwise.