paperbot · PL 论文追踪

全部论文 待读 收藏

P2 GS-Scale: Unlocking Large-Scale 3D Gaussian Splatting Training via Host Offloading

GS-Scale:通过主机内存卸载实现大规模3D高斯泼溅训练
ASPLOS Proceedings of the 31st ACM International Conference on Architectural Support for Programming Languages and Operating Systems, Volume 22026匹配度 2/10 · 该论文属于3D高斯泼溅训练的显存/内存管理优化系统,与用户核心关注的静态分析、Agentic程序、浮点程序分析及贝叶斯程序分析均无直接关系。虽然标签含“机器学习与程序语言”,但研究内容偏向ML系统运行时优化,而非PL→AI或AI→PL的静态分析交叉方向。引用 7 inbox
Donghyun Lee, Dawoon Jeong, Jae W. Lee, Hongil Yoon
crossref: 2026-09-12dblp: 2026-09-12arxiv: 2026-09-12

AI 速览

一句话:GS-Scale通过将高斯参数卸载到主机内存并按需传输,在降低GPU显存的同时保持训练速度。

大规模高质量 3D Gaussian Splatting 训练需存储大量参数、梯度和优化器状态,易超出 GPU 显存。GS-Scale 把所有高斯保存在主机内存,仅按需将子集传到 GPU 做前向和反向传播;为缓解 CPU 端视锥剔除与优化器更新的瓶颈,它采用选择性卸载几何参数、参数转发流水化 CPU 优化器更新、延迟更新零梯度高斯三项优化。评估显示 GPU 显存降低 3.3-5.6 倍,训练速度接近无卸载方案;在 RTX 4070 Mobile 上高斯数可从 400 万扩展到 1800 万,LPIPS 改善 23-35%。

问题
大规模高质量 3D Gaussian Splatting 训练受限于 GPU 显存,参数、梯度和优化器状态占用巨大。
方法
GS-Scale 将全部高斯存储在主机内存,按需把子集传到 GPU,并通过选择性几何参数卸载、参数转发流水化 CPU 优化器更新、延迟优化器更新三项系统优化来减少 CPU 端瓶颈和内存访问。
评估
在大规模数据集上评估,GPU 显存需求降低 3.3-5.6 倍,训练速度与无主机卸载的 GPU 方案相当;在 RTX 4070 Mobile 上支持高斯数量从 400 万扩展到 1800 万,LPIPS 提升 23-35%。
原文摘要(Abstract)

The advent of 3D Gaussian Splatting has revolutionized graphics rendering by delivering high visual quality and fast rendering speeds. However, training large-scale scenes at high quality remains challenging due to the substantial memory demands required to store parameters, gradients, and optimizer states, which can quickly overwhelm GPU memory. To address these limitations, we propose GS-Scale, a fast and memory-efficient training system for 3D Gaussian Splatting. GS-Scale stores all Gaussians in host memory, transferring only a subset to the GPU on demand for each forward and backward pass. While this dramatically reduces GPU memory usage, it requires frustum culling and optimizer updates to be executed on the CPU, introducing slowdowns due to CPU's limited compute and memory bandwidth. To mitigate this, GS-Scale employs three system-level optimizations: (1) selective offloading of geometric parameters for fast frustum culling, (2) parameter forwarding to pipeline CPU optimizer updates with GPU computation, and (3) deferred optimizer update to minimize unnecessary memory accesses for Gaussians with zero gradients. Our extensive evaluations on large-scale datasets demonstrate that GS-Scale significantly lowers GPU memory demands by 3.3-5.6x, while achieving training speeds comparable to GPU without host offloading. This enables large-scale 3D Gaussian Splatting training on consumer-grade GPUs; for instance, GS-Scale can scale the number of Gaussians from 4 million to 18 million on an RTX 4070 Mobile GPU, leading to 23-35% LPIPS (learned perceptual image patch similarity) improvement.

链接与引用

DOI 原文 · arXiv · PDF(开放获取) · DBLP

BibTeX
@inproceedings{abs-2509-15645,
  title = {GS-Scale: Unlocking Large-Scale 3D Gaussian Splatting Training via Host Offloading},
  author = {Donghyun Lee and Dawoon Jeong and Jae W. Lee and Hongil Yoon},
  booktitle = {ASPLOS},
  year = {2026},
  doi = {10.1145/3779212.3790167}
}