SH-Swin:面向高光谱木材树种识别的轻量级结构化光谱交互方法

《Smart Agricultural Technology》:SH-Swin: Lightweight Structured Spectral Interaction for Hyperspectral Wood Species Identification

【字体: 大 中 小 】 时间:2026年07月23日 来源:Smart Agricultural Technology 7.1

编辑推荐:

  高光谱成像为木材树种识别提供了一种非破坏性途径,但相邻光谱波段高度相关且具物理有序性。当基于Transformer的模型在无结构约束下混合这些通道时,会引入冗余计算并降低可解释性。研究人员提出SH-Swin,一种结构化光谱头分组Swin Transformer

  
高光谱成像为木材树种识别提供了一种非破坏性途径,但相邻光谱波段高度相关且具物理有序性。当基于Transformer的模型在无结构约束下混合这些通道时,会引入冗余计算并降低可解释性。研究人员提出SH-Swin,一种结构化光谱头分组Swin Transformer(Structured Spectral Head-Group Swin Transformer),将光谱波段组织作为建模先验而非将所有波段视为同质通道。该方法结合连续性约束的光谱波段精炼与波长感知子空间划分,以减少冗余的跨波段交互;随后采用头分组窗口注意力(Head-Group window attention)将通道混合限制于结构化光谱子空间内,并通过光谱门控交换(Spectral Gating Exchange)实现子空间间的自适应通信。由UNet++生成的前景掩码进一步通过SegMask注意力与前景感知全局平均池化(FG-GAP)融入模型,在抑制背景干扰的同时增强目标区域表征。在包含10个树种的高光谱木材数据集上,SH-Swin将总体分类准确率从0.892提升至0.960,宏F1(Macro-F1)从0.891提升至0.960。在固定预处理设定下,SH-Swin分类器需6.794 GFLOPs与21.598 M参数,而基线Swin Transformer V1分类器为9.334 GFLOPs与27.597 M参数。消融研究表明,分组光谱建模降低了分类器侧冗余计算,而光谱门控交换与前景感知机制改善了跨子空间交互与鲁棒特征建模。这些结果支持SH-Swin作为一种在固定预处理设定下具 favorable 分类器侧计算效率的可解释分类器,在自动化木材识别与智能林产品加工中具有潜在应用价值。
研究背景与问题提出
木材作为重要生物质资源与高值材料,其树种身份决定商业价值、工程用途、合规性与溯源能力。传统木材识别依赖宏观与微观解剖特征,存在主观性强、需专家经验、样品制备繁琐等局限。近红外光谱(NIRS)、可见-近红外高光谱成像、DART-MS化学指纹、元素同位素与DNA条码等方法虽减少了对人工解剖的依赖,但仍受仪器尺寸、特征稳定性、参考数据库与细粒度溯源能力限制。人工智能方法将木材识别推向学习视觉与光谱表征,但多数深度网络将高光谱通道视为同质输入并在全波段维度密集混合特征,而高光谱相邻波段强相关且物理有序,这种无结构约束的交互会传播冗余信息、削弱局部波长连续性并降低可解释性。针对细粒度树种识别,核心问题是如何在约束光谱交互的同时保留跨波长区域的互补信息。该研究由Tianqi Zhu、Yuxiang Huang等发表于《Smart Agricultural Technology》,以云南普洱人工林10个树种横切面高光谱图像为对象,提出SH-Swin框架,旨在以结构化光谱先验降低冗余计算、增强前景表征并取得可解释分类性能。
主要关键技术方法
研究人员使用SOC710-VP成像系统在400–1000 nm范围以4.69 nm光谱分辨率采集128波段显微高光谱立方体,样本来自云南普洱人工林10个树种(树个体数3–12不等),按树个体层级以7:2:1划分训练、验证与测试集以避免同树块泄漏。离线阶段以UNet++由RGB合成图生成二值前景掩码(前景IoU 0.947),通过积分图滑动窗裁剪为512×512并拼接掩码通道;光谱端基于前景统计做边界不稳定度筛选(去除原始1–21与118–128波段)得96连续有效波段,再以SAM、Pearson、SSIM与FDDM四指标融合相似度做连续性邻接合并至64波段;模型端以Swin Transformer V1为骨干,引入波长感知四子空间划分、分组卷积Patch Embedding、光谱头分组窗口注意力(Head-Group W-MSA/SW-MSA)、子空间内分组QKV与输出投影、光谱门控交换(含通道重排与Sigmoid门控)、SegMask注意力屏蔽背景查询、前景感知全局平均池化(FG-GAP)分类聚合。
研究结果
3.1 与基线模型比较
在12个CNN、轻量网与Transformer基线(EfficientNetV2、ShuffleNet、MobileNetV3、DenseNet、ResNet50、ResNeXt50、VGG16/19、ViT、SwinV1/V2)同协议下,SH-Swin总体准确率(OA)0.960、Macro-F1 0.960居首;3D t-SNE显示其类簇最紧凑分离,表4中类别0/3/4/5/7/8的F1均高于SwinV1与多数基线,证明结构化光谱交互提升细粒度判别与跨类稳定性。
3.2 与高光谱专用模型比较
对比MambaHSI、HyPyraMamba、CSCANet、A2S2KResNet、SpectralFormer,HyPyraMamba以24.284 GFLOPs获OA 0.971最高精度,但SH-Swin以6.794分类器GFLOPs获OA 0.960,比CSCANet(74.766)、A2S2KResNet(82.415)、SpectralFormer(39.227)分类器算力更低,表明其在分类器侧精度-算力权衡占优,但非全指标绝对最优。
3.3 计算复杂度分析
3.3.1 流程复杂度:UNet++离线分割69.242 GFLOPs/9.160 M参数,SH-Swin分类器6.794 GFLOPs/21.598 M参数,全管道76.036 GFLOPs/30.758 M参数;轻量声明仅指固定预处理下分类器前向。
3.3.2 SH-Swin复杂度:分组Patch Embedding将投影成本由hwCBP2降至hwC(B/G)P2;光谱头窗口注意力将QKV与输出投影由4hwC2降至4hwC2/G,窗口项2M2hwC不变,故固定窗口下随谱组数G增大投影开销下降。
3.4 模型结构优化效应
3.4.1 消融与敏感分析:基线SwinV1(OA 0.892,9.334 GFLOPs)加SC+HGA后GFLOPs降至6.791但OA跌至0.875,证分组省算力却伤全局;加Spectral Gating Exchange后OA回至0.920且算力几乎不变;加SegMask Attention至0.947;换FG-GAP达0.960。敏感分析显示64波段/4子空间为平衡配置,32波段欠压缩、80波段收益递减,2子空间精度略高但参数升、8子空间过碎降效。
3.4.2 优化后性能:SH-Swin各类别误分率较SwinV1降低,概率混淆矩阵对角占优,Grad-CAM响应集中于木材前景区。
3.4.3 可解释性:SHAP蜂群图显示波段61与43贡献略高但不依赖单波段,依赖图表明高反射区间波段61/63/56呈正SHAP响应,与Grad-CAM前景聚焦互为印证。
讨论部分总结
讨论指出高光谱通道不应始终视作同质输入,常规密集混频虽利全局建模但传播冗余并稀释波长结构;SC+HGA单独用降算力却损精度,说明邻域波长子空间内含足量语义但需受控交换;Spectral Gating Exchange以近零额外分类器代价恢复子空间互补性;前景机制(SegMask+FG-GAP)逐级净化背景统计。比较与SHAP/Grad-CAM一致表明保序光谱组织与受限跨子空间交互兼顾判别与可解释。局限含单区域人工林、仅VNIR(400–1000 nm)未含SWIR、UNet++使全管道非轻量、Dog-bone Wood召回偏低,未来可引VNIR-SWIR与木质素化学先验。
结论部分翻译
本研究提出SH-Swin,一种面向高光谱木材树种识别的结构化光谱头分组Swin Transformer。该方法通过整合波长感知分组光谱建模与前景感知表征学习,解决光谱冗余、无结构全局光谱交互与背景干扰问题,其核心原则是保留物理有意义的光谱组织并启用相邻光谱子空间间受控语义交互。SH-Swin将ShuffleNet的分组学习与通道重排哲学经结构化光谱头分组机制引入Swin Transformer,在相邻波长感知子空间内独立建模以减少冗余全局光谱混合;光谱门控交换实现受控跨子空间信息交换并恢复必要语义协作;SegMask注意力与FG-GAP将UNet++前景掩码嵌入注意力与聚合阶段以增强抗背景鲁棒性。在所评估高光谱木材数据集上,SH-Swin取得OA 0.960与Macro-F1 0.960,优于Swin Transformer及多数CNN、Transformer与高光谱专用模型且与若干专用方法竞争;固定预处理下分类器侧算力6.794 GFLOPs,UNet++离线分割全管道76.036 GFLOPs。消融证SC+HGAalone省算力但损性能,Spectral Gating Exchange+SegMask+FG-GAP渐进增效,敏感分析表明主配置为精度-分类器算力平衡而非全域最优。SHAP与Grad-CAM证其关注多代表光谱区且稳聚木材前景。SH-Swin为高分光谱木材识别提供可解释结构化光谱交互框架,在固定预处理下兼顾精度与分类器效率,对自动木材分析具设计范式意义。
相关新闻
生物通微信公众号
微信
新浪微博

热点排行

    今日动态 | 人才市场 | 新技术专栏 | 中国科学人 | 云展台 | BioHot | 云讲堂直播 | 会展中心 | 特价专栏 | 技术快讯 | 免费试用

    版权所有 生物通

    Copyright© eBiotrade.com, All Rights Reserved

    联系信箱:

    粤ICP备09063491号