训练NeRF只需5秒?!英伟达这项新技术给谷歌研究员整不会了 | 开源

新闻 人工智能
这是来自于英伟达的最新技术——基于多分辨率哈希编码的即时神经图形基元。

本文经AI新媒体量子位(公众号ID:QbitAI)授权转载,转载请联系出处。

 “这太疯狂了!”

“18个月前,训练NeRF需要5小时…”

“2个月前,训练NeRF需要是5分钟!”

“现在,训练NeRF需要5秒!!??”

究竟发生了什么?竟让谷歌研究科学家惊叹至此。

训练NeRF只需5秒?!英伟达这项新技术给谷歌研究员整不会了 | 开源

(插播一下此前风靡的NeRF模型:伯克利&谷歌联合打造,少数静态图生成3D逼真效果)

而且类似的评价还有很多~

训练NeRF只需5秒?!英伟达这项新技术给谷歌研究员整不会了 | 开源
训练NeRF只需5秒?!英伟达这项新技术给谷歌研究员整不会了 | 开源

介绍这项技术的博文也热度非常,不到一天时间就收获了上千赞。

原来,这是来自于英伟达的最新技术——基于多分辨率哈希编码的即时神经图形基元。

一只狐狸NeRF模型的只需要训练5秒钟!

训练NeRF只需5秒?!英伟达这项新技术给谷歌研究员整不会了 | 开源

而且不光训练NeRF,该技术还在其他任务上也达到了惊人的训练速度。

实现了几个数量级的组合加速。

能够在几秒钟内训练高质量的神经图形基元,并在几十毫秒内渲染分辨率1920x1080的图形。

训练NeRF只需5秒?!英伟达这项新技术给谷歌研究员整不会了 | 开源

单个GPU上实现多任务即时训练

先来看效果。

有没有种在实验室的眩晕感,这是来自iPhone视频训练5分钟的360度实时渲染效果~

还有想以34张真实照片重构的3D图像~

除了NeRF之外吗,还有三个神经图形基元的实现。

Gigapixel Image

作者还友情提示,网络是从头开始训练的,如果眨眼的话,可能会错过它。

每种情况都是用tiny-cuda-nn框架训练和渲染的,本次实验是在RTX 3090测试的。

以全连接网络为参数的神经图形基元,训练和评估成本较高。

本文使用了一种多功能的新输入编码来降低成本。

这种编码允许在不牺牲质量的情况下使用更小的网络,从而大大减少浮点数和内存访问数量。

训练NeRF只需5秒?!英伟达这项新技术给谷歌研究员整不会了 | 开源

具体而言,一个小型神经网络通过可训练特征向量的多分辨率哈希表进行扩充。这些特征向量的值通过随机梯度下降优化。

换句话说,这种编码与任务无关。

研究团队表示,他们在所有任务中都使用相同的案例和超参数,只改变了哈希表的大小。

目前该项目已开源,感兴趣的旁友可戳下方链接~

GitHub链接:
https://github.com/NVlabs/instant-ngp
论文链接:
https://nvlabs.github.io/instant-ngp/assets/mueller2022instant.pdf
项目链接:
https://nvlabs.github.io/instant-ngp/

 

 

 

责任编辑:张燕妮 来源: 量子位
相关推荐

2010-09-09 08:41:34

2011-07-30 13:22:49

2024-12-20 15:20:00

模型训练数据

2022-06-15 18:57:43

人工智能

2020-12-23 17:50:46

AI语言模型AI伦理

2020-07-30 14:05:53

谷歌华为芯片

2022-06-06 10:58:52

训练DeepMind研究

2023-04-05 14:25:58

LLM谷歌OpenAI

2021-06-29 15:33:28

谷歌Transformer模型

2022-06-06 10:54:18

模型训练开源

2020-03-16 10:16:19

代码开发工具

2020-08-13 09:30:30

谷歌Android开发者

2019-08-01 08:15:06

机器学习谷歌算法

2021-02-21 00:18:47

恶意软件研究职业技术

2009-05-05 14:28:56

mcafee虚拟化虚拟机

2022-02-07 15:05:07

模型AI训练

2020-09-21 14:25:26

Google 开源技术

2023-02-24 15:56:24

AI象棋

2020-08-24 08:15:29

软件互联网分布式

2016-11-13 23:11:15

点赞
收藏

51CTO技术栈公众号