基于信息论的计算成像系统设计与优化
成像系统中的编码器光学系统将物体映射为无噪图像噪声会将这些图像污染为测量值。我们的信息估计器仅利用这些含噪测量值和噪声模型来量化测量值区分不同物体的能力。作者Henry Pinkard, Leyla Kabuli, Eric Markley, Tiffany Chien, Jiantao Jiao, Laura Waller许多成像系统产生的测量结果人类从未见过或无法直接解读。你的智能手机在生成最终照片前会通过算法处理原始传感器数据。磁共振成像扫描仪采集频域测量值在医生查看前需要经过重建。自动驾驶汽车则直接用神经网络处理摄像头和激光雷达的数据。在这些系统中重要的不是测量结果看起来如何而是它们包含了多少有用信息。即使信息以人类无法解读的方式编码人工智能也能将其提取出来。然而我们很少直接评估信息内容。传统的指标如分辨率和信噪比分别评估质量的各个方面这使得比较在不同因素间进行权衡的系统变得困难。常见的替代方法是训练神经网络进行图像重建或分类但这将成像硬件的质量与算法的质量混为一谈。我们开发了一个框架能够根据信息内容直接评估和优化成像系统。在我们发表于 NeurIPS 2025 的论文中我们展示了这个信息度量可以预测成像系统在四个领域的性能并且优化该度量产生的设计能与最先进的端到端方法相媲美同时需要更少的内存、更少的计算量并且无需特定任务的解码器设计。为什么选择互信息互信息量化了测量结果在多大程度上减少了关于产生它的物体的不确定性。两个具有相同互信息的系统即使它们的测量结果看起来完全不同在区分物体的能力上也是等效的。这一个单一数值综合了分辨率、噪声、采样以及所有其他影响测量质量的因素。一张模糊、有噪点的图像如果保留了区分物体所需的关键特征它可能比一张丢失了这些特征的清晰、干净图像包含更多信息。信息将传统上分离的质量指标统一起来。它将噪声、分辨率和光谱灵敏度作为一个整体来考量而不是将它们视为独立的因素。以前将信息论应用于成像的尝试面临两个问题。第一种方法将成像系统视为无约束的通信信道忽略了透镜和传感器的物理限制这导致了极不准确的估计。第二种方法需要对被成像的物体进行显式建模限制了其通用性。我们的方法通过直接从测量值估计信息来避免这两个问题。从测量值估计信息估计高维变量之间的互信息是出了名的困难。样本需求随维度呈指数级增长且估计值存在高偏差和高方差。然而成像系统具有一些特性使得可以将这个难题分解为更简单的子问题。互信息可以写成I(物体测量值) H(测量值) - H(测量值|物体)第一项 H(测量值) 衡量了由物体差异和噪声共同引起的测量值总变异性。第二项 H(测量值|物体) 衡量了仅由噪声引起的变异性。互信息等于总测量变异与仅噪声变异之差。成像系统具有特性明确的噪声。光子散粒噪声服从泊松分布。电子读出噪声是高斯分布。这种已知的噪声物理特性意味着我们可以直接计算 H(测量值|物体)剩下的只需从数据中学习 H(测量值)。对于 H(测量值)我们用一个概率模型例如一个变换器或其他自回归模型来拟合一个测量值数据集。该模型学习所有可能测量值的分布。我们测试了三个在效率与准确性之间进行权衡的模型一个平稳高斯过程最快、一个完整高斯模型中等和一个自回归 PixelCNN最准确。该方法提供了真实信息的上界任何建模误差都只会导致高估而不会低估。在四个成像领域的验证如果信息估计能够捕捉到限制实际系统的因素那么它应该能够预测解码器的性能。我们在四个成像应用中检验了这种关系。信息估计可以预测彩色摄影、射电天文学、无透镜成像和显微镜领域的解码器性能。更高的信息量始终在下游任务中产生更好的结果。彩色摄影数码相机使用滤色器阵列限制每个像素只能检测特定波长的光。我们比较了三种滤色器设计传统的拜耳模式、随机排列和一种学习得到的排列。信息估计正确地对哪些设计能产生更好的色彩重建进行了排序这与神经网络去马赛克得出的排名一致且无需任何重建算法。射电天文学望远镜阵列通过结合全球各地站点的信号来实现高角分辨率。选择最优的望远镜站点位置在计算上是难以处理的因为每个站点的价值都依赖于其他所有站点。信息估计可以预测不同望远镜配置下的重建质量从而无需进行昂贵的图像重建即可实现站点选择。无透镜成像无透镜相机用光调制掩模取代传统光学元件。它们的测量结果在视觉上与场景毫无相似之处。信息估计可以预测在不同噪声水平下透镜、微透镜阵列和扩散器设计的重建精度。显微镜LED 阵列显微镜使用可编程照明来产生不同的对比度模式。信息估计与神经网络从细胞图像预测蛋白质表达的准确性相关从而无需进行昂贵的蛋白质标记实验即可进行评估。在所有情况下更高的信息量都意味着更好的下游任务性能。使用 IDEAL 设计系统信息估计不仅能评估现有系统。我们的“信息驱动的编码器分析学习”IDEAL方法利用信息估计的梯度上升来优化成像系统参数。IDEAL 通过信息估计的梯度反馈来优化成像系统参数无需解码器网络。计算成像设计的标准方法是端到端优化它联合训练成像硬件和一个神经网络解码器。这需要反向传播通过整个解码器造成了内存限制和潜在的优化困难。IDEAL 通过仅优化编码器来避免这些问题。我们在滤色器设计上对其进行了测试。从一个随机滤色器排列开始IDEAL 逐步改进设计。最终结果在信息内容和重建质量方面都与端到端优化相匹配。IDEAL 达到了与端到端优化相当的性能同时避免了训练期间的解码器复杂性。启示基于信息的评估为在实际条件下严格评估成像系统创造了新的可能性。当前的方法要么需要主观视觉评估要么需要部署时无法获得的真实数据要么使用孤立地衡量整体能力的指标。我们的方法仅从测量值就提供了一个客观、统一的度量。IDEAL 的计算效率为设计以前难以处理的成像系统提供了可能。通过避免解码器反向传播该方法降低了内存需求和训练复杂性。我们在后续工作中更广泛地探索了这些能力。该框架可能扩展到成像之外的其他感知领域。任何可以被建模为具有已知噪声特性的确定性编码的系统都可能从基于信息的评估和设计中受益包括电子、生物和化学传感器。本文基于我们发表在 NeurIPS 2025 上的论文《Information-driven design of imaging systems》。代码已在 GitHub 上提供。项目网站上提供了视频摘要。本文最初发表于 BAIR 博客并经作者许可在此发布。FINISHED更多精彩内容 请关注我的个人公众号 公众号办公AI智能小助手或者 我的个人博客 https://blog.qife122.com/对网络安全、黑客技术感兴趣的朋友可以关注我的安全公众号网络安全技术点滴分享
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2458480.html
如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!