首页|结合等变交叉正则化的轻量化双分支建筑物变化检测网络

结合等变交叉正则化的轻量化双分支建筑物变化检测网络

扫码查看
空间细节信息和语义上下文信息在建筑物变化检测研究中扮演着至关重要的角色.然而,在当下主流的单分支网络架构中,同时获取这两种信息会面临计算成本和模型大小方面的严峻挑战.为应对这一挑战,本文提出一种全新的轻量化双分支网络架构用于高效特征提取,并引入等变交叉正则化模块以增强特征表达,从而实现精细化的建筑物变化检测.具体地,轻量化双分支网络架构由细节分支、语义分支和双信息交互融合模块组成,能同时高效地捕获空间细节信息和语义上下文信息,以生成细粒度的深度语义变化特征图;同时,等变交叉正则化模块从隐式监督角度出发,在不增加网络参数的前提下,从语义和空间层面对变化特征图进行一致性约束,从而提高网络在变化建筑物尺度和边缘上的感知能力;为了验证方法的有效性,本文选用现有优秀的轻量化和非轻量化变化检测网络作为对比方法,分别在WHU、LEVIR两个数据集上开展对比实验.结果表明,在仅需2.27 M参数和4.25 G浮点运算的轻量化前提下,本文方法在两个数据集上分别实现了87.03%、83.41%的交并比精度,综合性能显著优于现有的轻量化和非轻量化变化检测方法.研究成果有望为未来的轻量化建筑物变化检测研究提供科学参考.
A lightweight dual-branch network incorporating equivariant cross-regularization for building change detection
Recently, deep learning has witnessed rapid advancements in building change detection based on remote sensing images. Much of this progress can be attributed to the simultaneous capture of spatial details and contextual semantics within single-branch architectures, thereby generating fine-grained and high-level semantic changed feature maps. Nevertheless, capturing spatial details necessitates convolutional layers with wide channels, while understanding contextual semantics requires a network with sufficient depth. Once simultaneously meeting both requirements in single-branch network architectures inevitably faces challenges in terms of computation costs and model sizes.To address these challenges, this study proposes a lightweight dual-branch network architecture for efficient feature extraction and introduces an equivariant cross-regularization module for enhanced feature expression, aimed at achieving effective building change detection. Specifically, the dual-branch network consists of a detail branch, a semantics branch, and a detail-semantics aggregation module. The detail branch adopts three simple convolution layers with wide channels and small receptive fields to maintain high-resolution spatial feature maps of changed buildings. Concurrently, the semantics branch employs a fast down-sampling strategy based on a stem block, six gather-and-expansion blocks, and a context embedding block to efficiently capture high-level semantic feature maps of changed buildings. The detail-semantics aggregation module serves as a bridge, mitigating the gaps in spatial resolution and semantic level gaps between the two types of feature maps to generate fine-grained and high-level semantic changed features. Additionally, the equivariant cross-regularization module constrains the changed feature at both the semantic and spatial levels without inflating network parameters while enhancing the model's sensitivity to the scale and boundaries of changed buildings. To evaluate the effectiveness of the proposed method, we compare it with numerous state-of-the-art lightweight and non-lightweight change detection networks using the WHU and LEVIR datasets. The results demonstrate that with just 2.27 M parameters and 4.25 G floating-point operations, our approach attains intersection-over-union accuracies of 87.03% and 83.41% on two datasets, respectively, surpassing both lightweight and non-lightweight change detection networks in comprehensive performance metrics. Furthermore, ablation experiments on the LEVIR dataset are conducted to analyze the effectiveness of our proposed dual-branch and modules. The results demonstrate that the dual-branch network incorporating the detail-semantics aggregation module effectively integrates the advantages of the detail branch and the semantics branch to generate fine-grained and high-level semantic changed features. On the other hand, integrating the equivariant cross-regularization module into the dual-branch network effectively enhances the network's capacity for identifying the scale and boundaries of changed buildings effectively.

building change detectionlightweight networkspatial detail informationcontextual semantic informationdetail branchsemantics branchdetail-semantics aggregation moduleequivariant cross-regularization module

戴延帅、慎利、刘仕川、董宽林、李志林

展开 >

西南交通大学地球科学与工程学院,成都 611756

西南交通大学高速铁路安全运营空间信息技术国家地方联合工程实验室,成都 611756

建筑物变化检测 轻量化网络 空间细节信息 语义上下文信息 细节分支 语义分支 双信息交互融合模块 等变交叉正则化模块

国家自然科学基金项目国家自然科学基金重大项目

4220151442394063

2024

地理信息世界
中国地理信息产业协会 黑龙江测绘地理信息局

地理信息世界

CSTPCD
影响因子:0.826
ISSN:1672-1586
年,卷(期):2024.31(3)
  • 10