arXiv:2504.14131cs.CVcs.LG2025-04

用改进U-Net直接从高光谱图像生成化学分布图,效果优于传统方法。

Transforming Hyperspectral Images Into Chemical Maps: A Novel End-to-End Deep Learning Approach

  • 用改进U-Net和自定义损失函数,端到端生成化学图谱。
  • 测试误差比PLS低7%,空间相关性达99.91%。
  • 结果物理合理,不超0-100%范围,适合材料与食品分析。

当前从高光谱图像生成化学图谱的方法多基于偏最小二乘(PLS)回归,产生像素级预测但忽略空间上下文,噪声大。本研究提出一种端到端深度学习方法,采用改进U-Net与自定义损失函数,直接从高光谱图像生成化学图谱,跳过传统像素级分析的中间步骤。在猪肉腹部样本的真实数据集上,与传统PLS回归对比,该方法在均脂肪含量预测任务中测试集均方根误差降低7%。同时,U-Net生成的化学图谱中99.91%的方差具有空间相关性;而PLS生成图谱仅2.37%方差具空间相关性,表明其像素预测高度独立。此外,PLS生成结果中存在超出物理合理范围(0-100%)的预测,而U-Net能有效约束输出在此区间内。结果表明,U-Net在化学图谱生成上显著优于PLS。

原文摘要 · Abstract (English)

Current approaches to chemical map generation from hyperspectral images are based on models such as partial least squares (PLS) regression, generating pixel-wise predictions that do not consider spatial context and suffer from a high degree of noise. This study proposes an end-to-end deep learning approach using a modified version of U-Net and a custom loss function to directly obtain chemical maps from hyperspectral images, skipping all intermediate steps required for traditional pixel-wise analysis. This study compares the U-Net with the traditional PLS regression on a real dataset of pork belly samples with associated mean fat reference values. The U-Net obtains a test set root mean squared error that is 7% lower than that of PLS regression on the task of mean fat prediction. At the same time, U-Net generates fine detail chemical maps where 99.91% of the variance is spatially correlated. Conversely, only 2.37% of the variance in the PLS-generated chemical maps is spatially correlated, indicating that each pixel-wise prediction is largely independent of neighboring pixels. Additionally, while the PLS-generated chemical maps contain predictions far beyond the physically possible range of 0-100%, U-Net learns to stay inside this range. Thus, the findings of this study indicate that U-Net is superior to PLS for chemical map generation.

化学图谱高光谱U-Net深度学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。