arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2307.10853cs.CV

探索弱监督变化检测中的有效先验与高效模型

Exploring Effective Priors and Efficient Models for Weakly-Supervised Change Detection

Wuhan University(武汉大学);Ant Group(蚂蚁集团)

Zhenghui Zhao, Lixiang Ru, Chen Wu

AI总结:

针对弱监督变化检测中图像级标注与像素级预测不一致的问题,提出膨胀先验解码器、标签门控约束及基于transformer的TransWCD模型,在WHU-CD数据集上取得显著性能提升,部分指标超越全监督方法。

摘要:

弱监督变化检测(WSCD)旨在仅利用图像级标注检测像素级变化。得益于其标注高效性,WSCD近年来受到越来越多的关注。然而,当前WSCD方法常面临变化遗漏与虚检的挑战,即图像级标注与像素级预测之间存在不一致。具体而言,变化遗漏指WSCD模型未能预测出任何变化像素,即便图像级标签表明存在变化,变化虚检则是相反的情况。为解决这一挑战,本研究在WSCD中利用全局尺度和局部尺度先验,提出两个组件:膨胀先验(DP)解码器和标签门控(LG)约束。DP解码器对带有变化图像级标签的样本进行解码,跳过带有未变化标签的样本,并将其替换为全未变化的像素级标签。LG约束源自变化表征与图像级标签之间的对应关系,当模型错误预测变化状态时对其进行惩罚。此外,我们开发了TransWCD,这是一个简单但强大的基于transformer的模型,展现了弱监督学习在变化检测中的潜力。通过将DP解码器和LG约束集成到TransWCD中,我们构建了TransWCD-DL。我们提出的TransWCD和TransWCD-DL在WHU-CD数据集上的F1分数分别比当前最优方法显著提升了+6.33%和+9.55%,部分性能指标甚至超过了若干全监督变化检测(FSCD)对比方法。代码将发布于https://github.com/zhenghuizhao/TransWCD。

英文摘要:

Weakly-supervised change detection (WSCD) aims to detect pixel-level changes with only image-level annotations. Owing to its label efficiency, WSCD is drawing increasing attention recently. However, current WSCD methods often encounter the challenge of change missing and fabricating, i.e., the inconsistency between image-level annotations and pixel-level predictions. Specifically, change missing refer to the situation that the WSCD model fails to predict any changed pixels, even though the image-level label indicates changed, and vice versa for change fabricating. To address this challenge, in this work, we leverage global-scale and local-scale priors in WSCD and propose two components: a Dilated Prior (DP) decoder and a Label Gated (LG) constraint. The DP decoder decodes samples with the changed image-level label, skips samples with the unchanged label, and replaces them with an all-unchanged pixel-level label. The LG constraint is derived from the correspondence between changed representations and image-level labels, penalizing the model when it mispredicts the change status. Additionally, we develop TransWCD, a simple yet powerful transformer-based model, showcasing the potential of weakly-supervised learning in change detection. By integrating the DP decoder and LG constraint into TransWCD, we form TransWCD-DL. Our proposed TransWCD and TransWCD-DL achieve significant +6.33% and +9.55% F1 score improvements over the state-of-the-art methods on the WHU-CD dataset, respectively. Some performance metrics even exceed several fully-supervised change detection (FSCD) competitors. Code will be available at https://github.com/zhenghuizhao/TransWCD.

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

↑