用改进的导数训练方法,提升组织光传输模拟的精度,助力光学超声成像重建。
DeepLight: A Sobolev-trained Image-to-Image Surrogate Model for Light Transport in Tissue
- 采用索博列夫训练增强神经网络对光传输导数的拟合能力
- 在分布内和分布外样本上均降低泛化误差,提升重建精度
- 适用于高维物理模拟,特别适合解决逆问题的高效建模
在光学超声成像中,通过反演光传输恢复组织吸收系数仍是一个挑战性问题。现有变分反演方法依赖于精确且可微的光传输模型。由于神经代理模型能快速、可微地模拟复杂物理过程,被视为解决此类逆问题的有力候选。然而,这类代理模型的导数通常无法准确匹配底层物理算子的真实导数,而准确的导数对求解逆问题至关重要,导数误差会严重阻碍高质量重建。为此,本文提出一种用于组织光传输的代理模型,采用索博列夫训练以提高模型导数的准确性。所用的索博列夫训练形式适用于一般高维模型。结果表明,该方法不仅提升了导数精度,还降低了分布内与分布外样本的泛化误差,显著增强了代理模型在下游任务中的实用性,尤其在求解逆问题方面具有重要价值。
原文摘要 · Abstract (English)
In optoacoustic imaging, recovering the absorption coefficients of tissue by inverting the light transport remains a challenging problem. Improvements in solving this problem can greatly benefit the clinical value of optoacoustic imaging. Existing variational inversion methods require an accurate and differentiable model of this light transport. As neural surrogate models allow fast and differentiable simulations of complex physical processes, they are considered promising candidates to be used in solving such inverse problems. However, there are in general no guarantees that the derivatives of these surrogate models accurately match those of the underlying physical operator. As accurate derivatives are central to solving inverse problems, errors in the model derivative can considerably hinder high fidelity reconstructions. To overcome this limitation, we present a surrogate model for light transport in tissue that uses Sobolev training to improve the accuracy of the model derivatives. Additionally, the form of Sobolev training we used is suitable for high-dimensional models in general. Our results demonstrate that Sobolev training for a light transport surrogate model not only improves derivative accuracy but also reduces generalization error for in-distribution and out-of-distribution samples. These improvements promise to considerably enhance the utility of the surrogate model in downstream tasks, especially in solving inverse problems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。