LLM2D
一种用于高光谱图像中砖块级油水分类的随机森林和CNN混合框架
A Hybrid Random Forest and CNN Framework for Tile-Wise Oil-Water Classification in Hyperspectral Images
作者: Mehdi Nickzamir, Seyed Mohammad Sheikh Ahamdi Gandab
发布日期: 2/4/2025
arXiv ID: oai:arXiv.org:2502.00232v1

摘要

arXiv:2502.00232v1 交叉公告类型: 摘要:提出了一种新的混合随机森林和卷积神经网络(CNN)框架,用于高光谱图像(HSI)中的油水分类。为了解决保持空间上下文的挑战,将图像分割成较小的、非重叠的块,这些块作为训练、验证和测试的基础。随机森林在像素级分类中表现出色,优于XGBoost、注意力机制U-Net和HybridSN等模型。然而,随机森林会丢失空间上下文,限制了其充分利用高光谱数据中空间关系的能力。为了提高性能,我们在随机森林生成的概率图上训练了一个CNN,利用CNN整合空间上下文的能力。混合方法在召回率上提高了7.6%(达到0.85),F1分数提高了2.4%(达到0.84),AUC提高了0.54%(达到0.99),相比基准模型有了显著提升。这些结果突显了结合概率输出和空间特征学习对于高光谱图像情境感知分析的有效性。