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

面向遥感分割的图信息瓶颈

Graph Information Bottleneck for Remote Sensing Segmentation

School of computer and Information Engineering, Central South University of Forestry and Technology(计算机与信息工程学院,中央南大学林业科技学院);Mohamed bin Zayed University of Artificial Intelligence(穆罕默德·本·扎耶德人工智能大学)

Yuntao Shou, Wei Ai, Tao Meng, Nan Yin

AI总结:

针对现有遥感分割方法对不规则物体建模灵活性不足、图对比学习易引入任务无关冗余信息的问题,提出结合图信息瓶颈的SC-ViG架构,替换UNet卷积模块实现分割,性能优于现有最先进方法。

摘要:

遥感分割在环境保护、城市变化检测等领域有着广泛的应用。尽管基于深度学习的遥感分割方法(如CNN和Transformer)已取得一定成效,但它们在对不规则物体建模时灵活性不足。此外,现有的图对比学习方法通常采用最大化互信息的方式来保持不同图视图之间的节点表示一致性,这可能导致模型学习到与任务无关的冗余信息。为解决上述问题,本文将图像视为图结构,提出了一种用于遥感分割的简单对比视觉图神经网络(SC-ViG)架构。具体而言,我们构建了节点掩码和边掩码的图视图以获取最优图结构表示,该表示能够自适应地学习是否对节点和边进行掩码。此外,本文创新性地将信息瓶颈理论引入图对比学习中,在最大化任务相关信息的同时最小化与任务无关的冗余信息。最后,我们用SC-ViG模块替换UNet中的卷积模块,以完成遥感图像的分割和分类任务。在公开真实数据集上开展的大量实验表明,本文方法的性能优于现有最先进的遥感图像分割方法。

英文摘要:

Remote sensing segmentation has a wide range of applications in environmental protection, and urban change detection, etc. Despite the success of deep learning-based remote sensing segmentation methods (e.g., CNN and Transformer), they are not flexible enough to model irregular objects. In addition, existing graph contrastive learning methods usually adopt the way of maximizing mutual information to keep the node representations consistent between different graph views, which may cause the model to learn task-independent redundant information. To tackle the above problems, this paper treats images as graph structures and introduces a simple contrastive vision GNN (SC-ViG) architecture for remote sensing segmentation. Specifically, we construct a node-masked and edge-masked graph view to obtain an optimal graph structure representation, which can adaptively learn whether to mask nodes and edges. Furthermore, this paper innovatively introduces information bottleneck theory into graph contrastive learning to maximize task-related information while minimizing task-independent redundant information. Finally, we replace the convolutional module in UNet with the SC-ViG module to complete the segmentation and classification tasks of remote sensing images. Extensive experiments on publicly available real datasets demonstrate that our method outperforms state-of-the-art remote sensing image segmentation methods.

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

↑