arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.01641cs.CV
收藏

AIO2:遥感图像分割中面向不完整标注深度学习的目标标签在线校正

AIO2: Online Correction of Object Labels for Deep Learning with Incomplete Annotation in Remote Sensing Image Segmentation

Technical University of Munich (TUM)(慕尼黑工业大学);German Aerospace Center (DLR)(德国航空航天中心)

Chenying Liu, Conrad M Albrecht, Yi Wang, Qingyu Li, Xiao Xiang Zhu

AI总结:

针对遥感分割中众包标签不完整且含噪的问题,提出 AIO2,通过 ACT 自适应触发校正、O2C 在线逐目标改标并结合 mean teacher,提升不同噪声水平下建筑物分割的鲁棒性。

摘要:

尽管遥感数据量与日俱增,地球观测中的深度学习仍面临缺乏用于监督优化的准确标注这一问题。OpenStreetMap 等众包项目将标注负担分配给其社区。然而,由于标签质量控制不足、标注人员缺乏,以及自然灾害和城市发展等因素导致的地球表面频繁变化,此类标注不可避免地会产生噪声。我们提出了自适应触发的在线逐目标校正(Adaptively trIggered Online Object-wise correction,AIO2),以解决由不完整标签集引发的标注噪声问题。AIO2 的特点包括:一个自适应校正触发(Adaptive Correction Trigger,ACT)模块,可在模型训练欠拟合或过拟合时避免进行标签校正;以及一种在线逐目标校正(Online Object-wise Correction,O2C)方法,该方法利用空间信息自动修改标签。AIO2 采用 mean teacher 模型来增强带噪标签训练的鲁棒性,既为 ACT 中的拟合稳定训练准确率曲线,也为 O2C 中的校正提供伪标签。此外,O2C 以在线方式实现,无需在每个训练 epoch 存储更新后的标签。我们在两个具有不同空间分辨率的建筑物足迹分割数据集上验证了所提方法。不同建筑物标签噪声程度下的实验结果证明了 AIO2 的鲁棒性。源代码将在 https://github.com/zhu-xlab/AIO2.git 提供。

英文摘要:

While the volume of remote sensing data is increasing daily, deep learning in Earth Observation faces lack of accurate annotations for supervised optimization. Crowdsourcing projects such as OpenStreetMap distribute the annotation load to their community. However, such annotation inevitably generates noise due to insufficient control of the label quality, lack of annotators, frequent changes of the Earth's surface as a result of natural disasters and urban development, among many other factors. We present Adaptively trIggered Online Object-wise correction (AIO2) to address annotation noise induced by incomplete label sets. AIO2 features an Adaptive Correction Trigger (ACT) module that avoids label correction when the model training under- or overfits, and an Online Object-wise Correction (O2C) methodology that employs spatial information for automated label modification. AIO2 utilizes a mean teacher model to enhance training robustness with noisy labels to both stabilize the training accuracy curve for fitting in ACT and provide pseudo labels for correction in O2C. Moreover, O2C is implemented online without the need to store updated labels every training epoch. We validate our approach on two building footprint segmentation datasets with different spatial resolutions. Experimental results with varying degrees of building label noise demonstrate the robustness of AIO2. Source code will be available at https://github.com/zhu-xlab/AIO2.git.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑