REFINE: Super-efficient Pruning for 3D Gaussian Splatting via Rendering-Free Primitive Importance¶
会议: ECCV 2026
论文: ECCV 原文
代码: https://zhangchen2022.github.io/REFINE.github.io/
领域: 3D视觉
关键词: 3D高斯泼溅, 模型剪枝, 基元重要性, 免渲染, 高效压缩
一句话总结¶
针对 3D 高斯泼溅(3DGS)后处理剪枝在“参数空间启发式失效”与“渲染敏感度开销过大”之间的两难困境,REFINE 提出了一种免渲染的基元重要性度量,通过解析解耦视线能见度与几何投影能量并结合内容自适应校准,将剪枝计算量降低超过 3000 倍且维持 SOTA 级画质。
研究背景与动机¶
3D 高斯泼溅(3DGS)凭借显式高斯基元建模与基于分块的高效光栅化管道,在保证高质量实时新视角合成方面取得了巨大突破。然而,3DGS 的自适应密度控制机制常常在场景中引入数百万个冗余高斯,单场景参数量庞大、存储占用轻松突破 GB 级别,这严重阻碍了其在移动端、VR/AR 头显等计算与内存受限边缘设备上的部署与流式传输。
针对冗余高斯的后处理剪枝技术应运而生,但现有方法陷入了两难抉择:一类是参数空间启发式方法(如 LightGaussian),通过不透明度或体积等几何属性直接剪枝,计算极快,却因完全忽视了光栅化管线的非线性物理映射而极易误删高频细节基元,导致渲染严重模糊;另一类是基于渲染感知的方法(如 PUP 3D-GS、MesonGS),通过光栅化前向渲染(甚至反向传播)累积像素敏感度或 Fisher 信息,保留了感知质量,却困于极其耗时的多视角遍历与光栅化循环,带来高达上万 GFLOPs 的计算负担。
这种分歧的核心矛盾在于:如何在不执行任何前向光栅化遍历的前提下,精确衡量三维参数扰动映射到二维图像空间的感知误差。本文的核心 idea 是:将基于 Fisher 信息的渲染 Hessian 场解析解耦为视角相关的“不透明度能见度”与“局部仿射几何投影能量”两项闭式积,并配合场景内容自适应的属性敏感度校准,在参数流形上直接闭式求解每个高斯基元的视觉重要性。
方法详解¶
整体框架¶
REFINE 是一种完全无需前向渲染、即插即用的 3DGS 后处理剪枝算法。对于包含 \(N\) 个预训练高斯基元的点云场景,方法不需要遍历相机视角渲染图像,而是直接依据相机位姿与几何先验,闭式估计高斯扰动对应的二阶 Hessian 重要性场,评估每个基元的几何、颜色与不透明度贡献。
整体处理流程主要包含三步:首先引入基元独立性与属性正交性结构假设,将高维密集的 Hessian 矩阵简化为主对角块;随后将渲染雅可比矩阵解析拆解为保守不透明度能见度和深度反平方投影衰减;最后结合场景全局统计特征自适应计算各属性的权重分配因子,得到每个基元的全局重要性评分并执行一次性排序截断。
%%{init: {'flowchart': {'rankSpacing': 24, 'nodeSpacing': 28, 'padding': 6, 'wrappingWidth': 400}}}%%
flowchart TD
A["输入:预训练 3DGS 基元集合与相机位姿"] --> B["解析解耦 Hessian 场<br/>视角不透明度能见度 × 几何投影能量"]
B --> C["内容自适应超参数校准<br/>计算纹理方差、透明度与尺度拉伸统计量"]
C --> D["重要性评分计算与全局排序<br/>无需渲染、O(N) 闭式代数打分"]
D --> E["按目标比例剪除低重要性基元并输出紧凑 3DGS"]
关键设计¶
1. 解析解耦 Hessian 场:以闭式能见度与几何投影能量替代前向光栅化 评估基元移除所引起的图像退化,本质上由渲染函数 \(R(\cdot)\) 对基元参数的雅可比矩阵 \(J_R\) 构成的 Gauss-Newton Hessian 矩阵 \(H = J_R^\top J_R\)(即缩放 Fisher 信息矩阵)决定。针对百万基元导致密集 Hessian 无法计算的问题,REFINE 基于基元间独立性假设和属性间正交性假设,将二阶矩阵近似为对角矩阵 \(W = \text{diag}(H)\)。进而,REFINE 将关键导数 \(\frac{\partial R^v}{\partial G_i^k}\) 近似分解为视点能见度 \(V_i^v\) 与几何投影影响 \(P_i^v\) 的乘积: $\(P_i^v \propto \frac{1}{(z_i^v)^2 + \eta}, \quad V_i^v \approx \alpha_i \cdot T_i(p) \approx \alpha_i\)$ 其中能见度 \(V_i^v\) 受到累积透射率 \(T_i\) 与基元不透明度 \(\alpha_i\) 约束,为免去昂贵的深度排序光栅化,REFINE 采用最坏情况下的保守零遮挡假设(\(T_i \approx 1.0\));几何投影 \(P_i^v\) 则借助 EWA 局部仿射变换雅可比矩阵的 Frobenius 范数能量,在局部简化下推导出其严格正比于相机深度坐标 \(z_i^v\) 的反平方(引入数值稳定常数 \(\eta = 0.05\))。这使得几何衰减得以在视角下以代数闭式快速评估。
2. 内容自适应超参数校准:动态平衡几何、颜色与不透明度的贡献灵敏度 不同场景的物理特征对属性的敏感度存在显著差异,例如具有大面积植被的场景对不透明度更敏感,而刚性结构场景则依赖几何尺度。为避免固定的静态加权系数破坏剪枝后的结构,REFINE 提出了内容自适应权重校准机制。首先在整个场景统计提取三个紧凑特征: $\(F_{col} = \frac{1}{N} \sum_{i=1}^N (Y_i - \bar{Y})^2, \quad F_{opa} = \frac{1}{N} \sum_{i=1}^N \alpha_i (1 - \alpha_i), \quad F_{gem} = \frac{1}{N} \sum_{i=1}^N \frac{\max(s_i)}{\min(s_i) + \eta}\)$ 其中 \(F_{col}\) 度量感知亮度 \(Y_i\) 的方差以捕捉高频纹理丰富度;凹函数 \(F_{opa}\) 评估半透明模糊区域比例;\(F_{gem}\) 统计基元尺度的平均结构拉伸比。随后通过在全场景期望值下对数平滑与归一化,自适应导出属性敏感度超参数 \(\lambda^{gem}, \lambda^{col}, \lambda^{opa}\)。
3. 重要性评分计算与超高效后处理剪枝:零渲染闭式代数打分 将自适应超参数与视点集 \(\mathcal{V}\) 上的期望解析权重融合,得到第 \(i\) 个基元关于属性子集 \(k\) 的 Hessian 权重: $\(w_i^k = \lambda^k \cdot \frac{1}{|\mathcal{V}|} \sum_{v \in \mathcal{V}} \frac{\alpha_i}{(z_i^v)^2 + \eta}\)$ 结合各属性的参数模长,基元总重要性分数直接定义为各属性加权和 \(D(G_i) = \sum_{k \in \{gem, col, opa\}} w_i^k \cdot \tilde{G}_i^k\)。在指定剪枝比例 \(\beta \in (0, 1)\) 下,算法仅需执行一次全局标量排序,剔除得分最低的后 \(\beta \cdot N\) 个基元。这一过程全流程由纯粹的 \(O(N)\) 向量代数运算完成,彻底摆脱了光栅化着色器、内存暂存与反向传播。
实验关键数据¶
主实验¶
在 Mip-NeRF 360、Tanks & Temples 和 Deep Blending 真实数据集上,针对原始 3DGS 执行 Zero-Shot 剪枝(剪除基元后完全不进行额外微调渲染优化),对比代表性参数方法与渲染敏感度方法:
| 数据集 | 方法 | 剪枝率 10% PSNR↑ / SSIM↑ | 剪枝率 30% PSNR↑ / SSIM↑ | 剪枝率 50% PSNR↑ / SSIM↑ | 剪枝率 70% PSNR↑ / SSIM↑ |
|---|---|---|---|---|---|
| Mip-NeRF 360 | 原始 3D GS | 27.35 / 0.814 | 27.35 / 0.814 | 27.35 / 0.814 | 27.35 / 0.814 |
| GHAP | 20.65 / 0.607 | 20.62 / 0.594 | 20.44 / 0.571 | 19.93 / 0.530 | |
| LightGaussian | 25.48 / 0.793 | 23.61 / 0.775 | 22.21 / 0.753 | 19.54 / 0.674 | |
| MesonGS | 25.87 / 0.800 | 25.86 / 0.800 | 25.68 / 0.795 | 24.35 / 0.760 | |
| PUP 3D-GS | 27.34 / 0.814 | 27.24 / 0.812 | 26.07 / 0.790 | 25.12 / 0.786 | |
| REFINE (本文) | 27.34 / 0.814 | 27.29 / 0.813 | 26.61 / 0.800 | 24.43 / 0.745 | |
| Tanks & Temples | 原始 3D GS | 23.39 / 0.842 | 23.39 / 0.842 | 23.39 / 0.842 | 23.39 / 0.842 |
| GHAP | 16.84 / 0.592 | 17.00 / 0.587 | 16.97 / 0.576 | 16.82 / 0.553 | |
| LightGaussian | 22.78 / 0.838 | 22.62 / 0.836 | 21.37 / 0.816 | 18.09 / 0.729 | |
| MesonGS | 22.08 / 0.817 | 22.05 / 0.816 | 21.98 / 0.812 | 21.41 / 0.786 | |
| PUP 3D-GS | 23.39 / 0.842 | 23.38 / 0.841 | 23.18 / 0.832 | 21.39 / 0.787 | |
| REFINE (本文) | 23.38 / 0.841 | 23.33 / 0.839 | 22.97 / 0.828 | 20.98 / 0.776 | |
| Deep Blending | 原始 3D GS | 29.52 / 0.903 | 29.52 / 0.903 | 29.52 / 0.903 | 29.52 / 0.903 |
| LightGaussian | 28.91 / 0.899 | 28.79 / 0.896 | 27.14 / 0.874 | 21.61 / 0.778 | |
| PUP 3D-GS | 29.52 / 0.903 | 29.51 / 0.903 | 29.28 / 0.899 | 29.14 / 0.894 | |
| REFINE (本文) | 29.52 / 0.903 | 29.49 / 0.903 | 29.28 / 0.899 | 27.93 / 0.871 |
此外,在计算复杂度与执行时间方面(原论文 Table 2),REFINE 展现出了极其惊人的优势:在 Mip-NeRF 360 上,渲染方法 PUP 和 MesonGS 分别需要高达 9582.81 GFLOPs 和 9539.19 GFLOPs,单场景平均剪枝时间 37~44 秒;而 REFINE 仅需 3.14 GFLOPs,计算量削减超过 3000 倍,耗时仅需 3.85 秒(50% 剪枝率下,提速近 10~20 倍实际设备延迟)。
消融实验¶
在剪枝率 50% 下对组件有效性与权重设计的消融分析(原论文 Table 3):
| 配置 | PSNR ↑ | SSIM ↑ | LPIPS ↓ | 说明 |
|---|---|---|---|---|
| Full Model (Ours) | 28.35 | 0.862 | 0.198 | 完整模型(联合能见度 + 几何投影 + 自适应权重) |
| w/o \(V_i^v\)(去掉视点能见度) | 27.74 | 0.849 | 0.212 | 仅依赖几何投影,忽略不透明度遮蔽关系,PSNR 下降 0.61 dB |
| w/o \(P_i^v\)(去掉几何投影) | 27.24 | 0.842 | 0.214 | 仅依赖不透明度,忽略空间深度与相机距离映射,PSNR 下降 1.11 dB |
| Equal \(\lambda^k\)(静态均等属性加权) | 27.82 | 0.852 | 0.209 | 使用固定统一超参数,无法适应纹理/半透明结构差异,PSNR 下降 0.53 dB |
关于 Hessian 近似合理性的定量验证(原论文 Table 4 与 Sec 4.3):通过对角能量比(Diagonal Energy Ratio, DER)测试发现,高斯基元间的相互依赖在采样区域中对角块能量平均高达 86%(即基元间二阶交叉梯度仅占 14%);属性正交性平均 DER 达到 0.79(在 bicycle 场景高达 0.91),强力支撑了基元独立性与属性正交性的理论合理性。
关键发现¶
- 几何投影能量 \(P_i^v\)(深度反平方衰减)是影响最大的单项机制:去除后 PSNR 下降超过 1.1 dB,证明离相机越近的基元在扰动时产生的屏幕弥散与误差越大,具有不可替代的几何感知约束。
- 启发式方法(如 LightGaussian)在剪枝比例超过 50% 时迅速崩溃(PSNR 陡降至 22.21),而 REFINE 在 50% 剪枝率下维持 26.61 dB,逼近甚至在部分指标上超越了基于多视角完整渲染的 PUP 3D-GS。
- 在包含极重度遮挡和稠密堆叠的室内场景(如 Deep Blending 与 kitchen 场景,DER 仅为 0.67),当剪枝率达到 70% 的极端情况时,基元间的跨基元耦合效应显现,零遮挡假设与独立性近似会导致一定程度的性能滑坡。
亮点与洞察¶
- 将复杂的全局渲染敏感度解析为局部闭式几何量:巧妙通过 EWA 泼溅的仿射雅可比能量推导得出深度的反平方衰减关系,用纯代数形式绕过了费时光栅化渲染循环。
- 基于统计特征的内容自适应属性校准:提取场景全局的感知亮度方差、透明度歧义度和尺度各向异性,动态自平衡不同属性的重要性权重,无需反向传播调参。
- 极其实用的工业级落地价值:将后处理剪枝计算量压缩至 1~3 GFLOPs 数量级,在无须微调的前提下几秒内完成百万级基元的极致精简,极大拓宽了 3DGS 在边缘算力受限设备上的落地前景。
局限与展望¶
- 极端遮挡环境下的独立性假设失效:作者指出当场景内部存在极度密集、沿单视线多层高斯深度重叠时,忽略多基元间累积透射率变化(零遮挡假设)会在高剪枝率(≥70%)下产生性能下降。
- 视角分布采样依赖:算法在计算期望深度时依赖训练/代表视点的空间坐标分布,若视点分布不均,靠近稀疏视角边界的高斯可能出现评估偏差。
- 未来改进方向:可进一步探索结合超轻量局部分块光线采样,或利用低秩离线稀疏修正矩阵补偿基元耦合,在几乎不增加耗时的前提下增强极端场景的剪枝鲁棒性。
相关工作与启发¶
- vs LightGaussian: LightGaussian 仅在参数空间依据不透明度与体积等经验阈值进行离线粗筛,缺乏相机视角到屏幕空间的映射,高剪枝率下高频细节损失惨重;REFINE 解析推导了与渲染管线等价的投影能量,在保持同等极速的同时画质显著超越。
- vs PUP 3D-GS / MesonGS: PUP 等方法依赖逐帧前向光栅化遍历积累 Fisher 信息与像素贡献,单场景剪枝需消耗上万 GFLOPs,计算极其耗时;REFINE 利用解析解直接给出封闭解重要性打分,将算力开销降低了 3000 倍以上。
评分¶
- 新颖性: ⭐⭐⭐⭐ [成功通过投影解析与二阶矩阵统计特性避开前向渲染,思路巧妙自洽]
- 实验充分度: ⭐⭐⭐⭐⭐ [覆盖三大主流基准、详尽的复杂度分析、消融实验以及严谨的 Hessian 对角能量比定量检验]
- 写作质量: ⭐⭐⭐⭐⭐ [数学推导流畅规范,物理意义阐述透彻清晰]
- 价值: ⭐⭐⭐⭐⭐ [对于 3DGS 移动端轻量化与实时传输具有立竿见影的工程与学术价值]