对抗样本噪声的遥感深度学习分类网络模型

研究背景
深度学习方法的性能主要取决于两个因素:训练数据量的大小和标签标注的准确性。相比于数据量不足的问题,标签标注噪声问题更难解决。在遥感数据集中,标签噪声普遍存在,主要原因包括以下几点:一是遥感图像的土地类型复杂,需要专业知识才能准确标注;二是多个专家对同一图像标注时可能存在不一致;三是低成本的自动化标注或非专业人员标注往往导致标注结果不可靠。标签噪声会通过影响网络的损失值,使网络参数朝错误方向更新,从而降低分类性能。
针对标签噪声问题的处理方法主要分为两类:一类是针对标签本身,另一类是针对损失值。针对标签的处理方法包括样本筛选和噪声转移矩阵的计算。样本筛选方法通过选择干净样本进行训练,避免噪声样本对网络的负面影响。噪声转移矩阵方法则通过估计类别间的转换概率,纠正噪声标签为干净标签。然而,这些方法的效果依赖于样本选择策略或噪声分布的准确性。
针对损失值的处理方法包括设计鲁棒性损失函数和对损失值赋权重。鲁棒性损失函数通过减少噪声样本对损失值的影响,提高网络的抗噪性能,但在复杂数据集上表现有限。赋权重的方法通过调整样本的损失权重,降低噪声样本的影响,同时提高干净样本的权重,从而提升网络性能。
在遥感图像分类任务中,标签噪声问题的研究较少。为此,本文提出了一种通用的抗噪声网络框架——Weight Loss Net(WLN),通过结合注意力机制动态调整样本权重,降低噪声样本的影响,并在公开数据集上验证了其有效性。
数据源及覆盖区域
本文采用了Inria Aerial Image Labeling Dataset(以下简称Inria数据集)作为实验数据源。该数据集解决了遥感领域中航空图像像素级自动标注的问题,图像分辨率为30cm,标签分为建筑类别和非建筑类别。数据集覆盖了不同城市区域,包括建筑物密集的大城市和建筑物稀少的小镇。
Inria训练数据集包含180幅5000×5000大小的图像,覆盖Austin、Chicago、Kitsap County、Western Tyrol和Vienna五个地区,总面积为405平方公里。标签由180张单通道图像组成,其中255表示建筑类别,0表示非建筑类别。由于测试集标签不可用,本文将原训练集按8:1:1的比例划分为训练集、验证集和测试集,三者相互独立且不重叠。

实验噪声设置
标签噪声是指实例的标注类别与实际类别不一致。在像素级遥感土地覆被标注中,常见的标签噪声包括:1)多标注噪声,标签面积大于实际面积;2)少标注噪声,标签面积小于实际面积;3)错标注噪声,将对象标注为错误类别;4)漏标注噪声,对象的标签完全缺失。
为了模拟标签噪声,本文采用卷积核对标签进行图像膨胀和腐蚀操作。膨胀操作模拟多标注噪声和错标注噪声,腐蚀操作模拟少标注噪声和漏标注噪声。根据卷积核大小,设置了三个噪声等级:卷积核为9表示Noise level 1,卷积核为17表示Noise level 2,卷积核为25表示Noise level 3。此外,设置了五个噪声率(0%、25%、35%、45%、50%)以测试网络在不同噪声水平下的表现。

WLN 网络结构

WLN网络由分割子网络和注意力子网络组成。分割子网络用于生成分割结果并计算损失值;注意力子网络通过注意力机制计算样本的损失权重值。为解决腐蚀噪声中类别失衡的问题,本文引入类别平衡系数,与损失值和损失权重值结合,作为最终损失值,用于反向传播更新网络参数。
分割子网络采用U-Net模型,也可替换为其他分割模型。注意力子网络采用SE模块,通过捕获特征通道间的依赖性,为每个通道分配权重,从而增强有用特征,抑制无用特征。
实验结果及分析
实验通过对WLN和U-Net模型在不同噪声类型和水平下的表现进行定量和定性分析,验证了WLN的抗噪性能。
(1)膨胀噪声
在膨胀噪声下,随着噪声率和噪声等级的增加,U-Net的精度显著下降,而WLN能够保持较高的精度。例如,在Noise level 3和噪声率为50%时,U-Net的OA下降12.7%,而WLN仅下降0.2%。

(2)腐蚀噪声
在腐蚀噪声下,U-Net在高噪声率和高噪声等级时表现较差,而WLN的稳定性更好。例如,在Noise level 3和噪声率为50%时,U-Net的OA下降8.4%,而WLN仅下降1.5%。

提取细节比较
通过目视解译,WLN在膨胀和腐蚀噪声下的地物提取细节明显优于U-Net,尤其是在高噪声水平下,WLN能够更准确地识别建筑物边界。


结论
本文提出了一种通用的抗噪声网络框架WLN,通过结合分割子网络和注意力子网络,有效降低了标签噪声对遥感图像分类的影响。实验结果表明,WLN在不同噪声类型和水平下均表现出较高的精度和鲁棒性,优于传统方法。未来工作将进一步研究类别平衡系数的优化及其在其他任务中的应用。
原文链接
Lin, C., Guo, S., Chen, J., Sun, L., Zheng, X., Yang, Y., & Xiong, Y. (2021). Deep Learning Network Intensification for Preventing Noisy-Labeled Samples for Remote Sensing Classification. Remote Sensing, 13(1689), 1–19.
Enjoy Reading This Article?
Here are some more articles you might like to read next: