arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.19082cs.LGcs.AIcs.CVcs.GRcs.HC

用于误差感知场景表示网络的正则化多解码器集成

Regularized Multi-Decoder Ensemble for an Error-Aware Scene Representation Network

The Ohio State University(俄亥俄州立大学)

Tianyu Xiong, Skylar W. Wurster, Hanqi Guo, Tom Peterka, Han-Wei Shen

AI总结:

针对科学可视化中 SRN 推理时无法评估坐标级重建误差的问题,提出共享特征网格的多解码器集成及方差正则化损失,实现可靠不确定性估计与更准确重建。

摘要:

特征网格场景表示网络(SRN)已被应用于科学数据,作为用于分析和可视化的紧凑函数代理。由于 SRN 是黑盒有损数据表示,评估预测质量对于科学可视化应用至关重要,以确保科学家能够信任正在可视化的信息。目前,现有架构不支持推理时的重建质量评估,因为在缺少真值数据的情况下无法评估坐标级误差。我们提出了一种参数高效的多解码器 SRN(MDSRN)集成架构,该架构由一个共享特征网格和多个轻量级多层感知机解码器组成。MDSRN 可以针对给定输入坐标生成一组合理预测,以计算均值作为多解码器集成的预测,并计算方差作为置信度分数。坐标级方差可以与数据一起渲染,以告知重建质量,或者被集成到不确定性感知体可视化算法中。为防止量化方差与预测质量之间的不一致,我们提出了一种用于集成学习的新型方差正则化损失,它促使正则化多解码器 SRN(RMDSRN)获得与真实模型误差密切相关的更可靠方差。我们在多种标量场数据集上,全面评估了 Monte Carlo Dropout、Mean Field Variational Inference、Deep Ensemble 和 Predicting Variance 与所提出的 MDSRN 和 RMDSRN 相比的方差量化质量和数据重建质量。我们证明,在相同神经网络参数预算下,RMDSRN 在不确定性 SRN 中实现了最准确的数据重建以及具有竞争力的方差-误差相关性。

英文摘要:

Feature grid Scene Representation Networks (SRNs) have been applied to scientific data as compact functional surrogates for analysis and visualization. As SRNs are black-box lossy data representations, assessing the prediction quality is critical for scientific visualization applications to ensure that scientists can trust the information being visualized. Currently, existing architectures do not support inference time reconstruction quality assessment, as coordinate-level errors cannot be evaluated in the absence of ground truth data. We propose a parameter-efficient multi-decoder SRN (MDSRN) ensemble architecture consisting of a shared feature grid with multiple lightweight multi-layer perceptron decoders. MDSRN can generate a set of plausible predictions for a given input coordinate to compute the mean as the prediction of the multi-decoder ensemble and the variance as a confidence score. The coordinate-level variance can be rendered along with the data to inform the reconstruction quality, or be integrated into uncertainty-aware volume visualization algorithms. To prevent the misalignment between the quantified variance and the prediction quality, we propose a novel variance regularization loss for ensemble learning that promotes the Regularized multi-decoder SRN (RMDSRN) to obtain a more reliable variance that correlates closely to the true model error. We comprehensively evaluate the quality of variance quantification and data reconstruction of Monte Carlo Dropout, Mean Field Variational Inference, Deep Ensemble, and Predicting Variance compared to the proposed MDSRN and RMDSRN across diverse scalar field datasets. We demonstrate that RMDSRN attains the most accurate data reconstruction and competitive variance-error correlation among uncertain SRNs under the same neural network parameter budgets.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑