HackerTrans
TopNewTrendsCommentsPastAskShowJobs

MrNeRF

no profile record

Submissions

$1700 bounty delivers 2.4× speedup for CUDA Gaussian Splatting training

github.com
3 points·by MrNeRF·11 maanden geleden·1 comments

Optimzed Fused-SSIM

github.com
2 points·by MrNeRF·vorig jaar·1 comments

comments

MrNeRF
·11 maanden geleden·discuss
This started as a $1700 bounty to optimize the CUDA Gaussian Splatting renderer. The PR delivers impressive performance improvements with a 2.4× speedup. Training time for 7 scenes from the mipnerf360 dataset dropped from 50 minutes to ~20 minutes without impacting final splat quality.

Original bounty: https://github.com/MrNeRF/gaussian-splatting-cuda/issues/135 PR: https://github.com/MrNeRF/gaussian-splatting-cuda/pull/245
MrNeRF
·vorig jaar·discuss
I significantly improved the forward and backward pass, as well as the inference of SSIM.

Optimized Fused-SSIM (Forward) 2.49ms (Backward) 2.68ms (Inference) 1.43ms Original Fused-SSIM (Forward) 3.66ms (Backward) 3.52ms (Inference) 2.59ms

This is the new sota and probably the fastest implementation out there. I would love to see if someone could improve here and get another factor. More details are found in the repo.
MrNeRF
·3 jaar geleden·discuss
Open source software deserves celebration. Its impact on the world is profound and transformative. Recently, I began porting the 3D Gaussian splatting paper to C++ and CUDA with an aim to optimize the training speed of such models. This endeavor was only possible because the original authors open-sourced their work. The feedback has been encouraging, and I feel that I'm making a difference—perhaps even more than in my regular job. That said, it's essential to remember that contributing to open source, while rewarding, can be demanding if it's not your full-time occupation. Yet, through such efforts, we ensure these tools remain freely accessible and aren't monopolized by corporations. I provide you the link to this GitHub project in case you are interested. https://github.com/MrNeRF/gaussian-splatting-cuda