1. Introduction
Against the backdrop of intensifying global climate change, floods exhibit notable characteristics such as increased frequency, expanded affected areas, and aggravated disaster losses, posing severe challenges to existing disaster response systems. Traditional flood monitoring relies on sparse hydrological stations with limited coverage, which makes it difficult to accurately determine the extent of inundation. Moreover, optical satellite remote sensing is susceptible to cloud interference during cloudy and rainy weather, which drastically reduces its monitoring effectiveness. Remote sensing images, with their high spatial resolution, rapid revisit capability, and all-weather monitoring ability, can penetrate clouds and rain to acquire surface information. Additionally, they can accurately extract waterbody extents, and dynamically track flood evolution processes, thereby providing critical data support for pre-disaster warnings, emergency responses during disasters, and post-disaster damage assessments. Therefore, research on water body extraction from high-resolution remote sensing images is urgently needed to address increasingly severe flood disasters and enhance disaster prevention, mitigation, and relief capabilities, with substantial practical implications and scientific value.
Current remote sensing water body identification technologies have formed an intergenerational evolution system from traditional threshold segmentation and traditional machine learning to deep learning [
1]. Traditional threshold methods, based on the physical characteristics of low reflectance of water bodies in the near-infrared band, achieve rapid segmentation using indices such as Normalized Difference Water Index (NDWI) and Modified Normalized Difference Water Index (MNDWI); however, they are susceptible to interference from shadows and spectral confusion with land features [
2,
3]. Traditional machine learning methods, such as Random Forest (RF) and Support Vector Machine (SVM), improve classification accuracy by manually designing multidimensional features, such as spectral and texture features [
4,
5,
6,
7,
8,
9]. Yue et al. employed a supervised random forest classifier to automatically generate high-quality training samples from multisource water products, enabling large-scale water body extraction without manual intervention [
10]. Nagaraj R systematically compared the performance of eight machine learning classifiers for water body extraction from high-resolution remote sensing images, providing a systematic performance benchmarking reference for the application of machine learning methods to small water body extraction and dynamic water resource monitoring [
11]. Gümüşçü İ compared the water segmentation performance of SVM and RF algorithms under different Sentinel-2 band combinations, revealing the influence mechanisms of different geographical environments on classification accuracy [
12]. Traditional threshold methods and machine learning methods represented by RF and SVM still have certain limitations: in complex environments, cloud occlusion, terrain shadows, and the phenomena of “same object with different spectra” and “different objects with the same spectrum” for water bodies lead to severe misclassifications; for extracting elongated rivers and small/micro water bodies, due to small target pixel proportions, sample imbalance, and the tendency of deep features to be overwhelmed, issues such as fragmentation and omission frequently occur; the water–land boundary is complex and blurred, making it difficult for models to achieve fine delineation.
Deep learning methods, primarily based on Fully Convolutional Networks, such as U-Net and DeepLabV3+, perform end-to-end automatic feature learning through multiple layers of nonlinear transformations. They can automatically learn hierarchical features from low-level (edges and textures) to high-level (semantic concepts) from massive data, overcoming the bottlenecks of traditional remote sensing information extraction, that is, “cumbersome manual feature design and ill-posed physical parameter inversion.” These methods considerably improve the accuracy of land cover classification, recall rate of target recognition, and automation level of change detection, laying an architectural foundation for high-precision feature identification from high-resolution imagery, including water body extraction, road segmentation, and object detection [
13,
14].
In recent years, deep learning models for water body information extraction have evolved along two main lines. The first is architecture optimization represented by the U-Net and DeepLab series, which incorporates encoder–decoder structures, dilated convolutions, and attention mechanisms, continuously pushing the boundaries of boundary refinement and multi-scale target recognition in complex scenarios [
15,
16,
17]. Wagner F H extended U-Net into an instance segmentation model, U-Net-Id, demonstrating the instance-level segmentation capability of the U-Net architecture for information extraction from high-resolution satellite imagery, providing a reference for the evolution from semantic segmentation to instance segmentation [
18]. Zhu Yuying et al. introduced Dense Cross-layer Skip Connections (DCSC) to replace the original skip connections of U-Net, effectively reducing information loss during feature propagation through the cross-scale fusion and optimization of multi-level semantic features [
19]. Li Z. systematically studied an integrated framework of active learning, incremental learning, transfer learning, and DeepLabV3+, finding that only 10–15% labeled samples were needed to achieve a Mean
IoU of 0.90, reducing annotation costs to some extent [
20]. Abramova A. compared the performance of U-Net, U-Net++, Attention U-Net, and R2 U-Net for Arctic lake segmentation, showing that the classic U-Net (
IoU = 0.88) outperformed more complex variants, indicating that architectural improvements do not always translate into actual performance gains [
21]. Dias J.A.C. systematically compared U-Net, DeepLabV3+, and SegFormer for cloud segmentation, finding that U-Net achieved the best balance between accuracy and efficiency, DeepLabV3+ had the fastest processing speed but slightly lower accuracy, and SegFormer achieved the highest accuracy but had the longest inference time [
22].
The second line combines the encoder structure of FCNs with new paradigms such as transformers [
23,
24], leading to models such as Swin-Unet and the Segment Anything Model (SAM) [
25,
26,
27], which achieved further improvements in long-range dependency modeling and few-shot generalization capabilities. Wan Z. provided the first systematic review of SAM’s application in remote sensing water body extraction, comprehensively analyzing challenges, adaptation strategies, task performance, and future directions [
28]. Chang Z used Swin Transformer as the backbone framework to construct CW-SwinUNet, surpassing state-of-the-art (SOTA) methods in the semantic segmentation of ultra-high-resolution remote sensing images, validating the effectiveness of the Transformer-UNet hybrid architecture [
29].
In terms of data processing, rough set theory provides an important mathematical framework for handling imprecise and incomplete information and knowledge, exhibiting unique theoretical characteristics when dealing with uncertainty and distinguishing it from traditional uncertainty analysis methods such as probability statistics, fuzzy set theory, and evidence theory. Rough set theory does not rely on prior knowledge external to the dataset (e.g., probability distribution functions or membership parameters). Clearer decision rules are obtained through attribute reduction, thereby maintaining higher objectivity in the knowledge discovery process. The rough set theory has been widely used in many fields, including machine learning, data mining, image processing, pattern recognition, decision support, and analysis. Andrzej introduced granular computing into the approximate reasoning process of rough set theory, enriching the ability of intelligent systems (ISs) to generate reasonable solutions [
30]. Amjad et al., based on the fundamental properties of rough sets, established a framework for h-approximation spaces, providing more flexible and precise technical means for data approximation [
31]. Nguyen proposed a generalized weighted neighborhood rough set model (GWNRSs) that addresses boundary region objects often overlooked in traditional models, achieving good results in classification accuracy and reduction size [
32]. C.V. et al. optimized a rough-set-based rule classifier and integrated data augmentation techniques, improving the automatic fault detection accuracy of photovoltaic modules on drones [
33]. Tarek et al. introduced a novel optimizer based on rough set theory to optimize the deep feature extraction process, demonstrating excellent performance in thermal face recognition tasks, effectively handling challenges such as noise, occlusion, and illumination changes [
34]. This research has opened new directions for the application of rough set theory in deep learning and computer vision.
Rough set theory, as an effective mathematical tool for handling imprecise and uncertain information, has been increasingly applied to remote sensing classification and hydrological feature extraction in recent years. Unlike traditional statistical methods that rely on prior probability distributions, rough set theory is entirely data-driven and requires no external knowledge, making it particularly suitable for processing the high-dimensional, noisy, and heterogeneous spectral data typical of remote sensing imagery [
31]. The tolerance rough set, an extension of classical rough set theory, replaces the strict equivalence relation with a tolerance (similarity) relation, enabling direct processing of continuous-valued attributes without discretization—a critical advantage for remote sensing spectral feature analysis. In the domain of wetland and water-related land-cover classification, the tolerant rough set has been integrated with neural networks to preprocess training data, effectively reducing the influence of noisy samples and improving network training success rates [
35]. More recently, Dong et al. [
36] introduced a granular-ball rough set algorithm for feature variable selection in hydrological and water quality datasets, combining it with k-nearest neighbor analysis to enhance predictive performance while significantly reducing model complexity. Their work demonstrated that granular-ball rough sets offer a more flexible approximation of data distributions compared to traditional neighborhood-based approaches. Barman et al. [
37] proposed a two-step classification method for hyperspectral remote sensing images that first performs band selection using neighborhood rough set theory to mitigate the “curse of dimensionality,” followed by a mathematical morphology-based classification. Additionally, recent studies have explored fuzzy rough set-based approaches for quantifying local and overall fuzzy relations between variables in spatial data, offering new perspectives for hydrological spatial analysis [
38]. Variable-precision rough sets have also been applied to land cover discrimination in wetland inventories, further demonstrating the flexibility of rough set variants in handling classification uncertainty in complex aquatic environments [
39]. These studies collectively underscore the potential of rough set theory—particularly its tolerance, neighborhood, granular-ball, and fuzzy variants—as a powerful preprocessing and feature selection tool for enhancing the accuracy and efficiency of water body extraction from high-resolution remote sensing imagery.
In remote sensing-based water body extraction, commonly adopted sample denoising and optimization strategies encompass conventional filtering-based denoising, data augmentation, active learning, self-training, co-training, and transfer learning. Conventional filtering methods—including mean, median, and bilateral filters—can effectively attenuate image noise; however, their fixed kernel sizes are inherently ill-suited to capturing the spatially heterogeneous noise distributions encountered in complex land-cover mosaics, frequently resulting in blurred water boundaries and the loss of narrow or small water features. Data augmentation, which diversifies training samples via geometric transformations and color perturbations, can improve generalization to a limited extent; however, it cannot fundamentally eliminate label noise or low-quality samples, and in water extraction settings marked by severe class imbalance, simplistic augmentation may inadvertently exacerbate the inherent skew. Active learning, which iteratively queries the most informative unlabeled samples for expert labeling, offers substantial savings in annotation effort, yet its effectiveness is critically contingent upon the quality of the initial labeled set and the design of the acquisition function, rendering it vulnerable to erroneous query loops under elevated noise levels. Self-training and co-training, as representative semi-supervised paradigms, exploit unlabeled data through pseudo-label generation and cross-model collaboration. However, in cross-regional applications where significant domain shifts exist between source and target areas, the cumulative propagation of mislabeled pseudo-labels can severely degrade final segmentation performance. Transfer learning, while capable of alleviating target-domain label scarcity by leveraging source-domain knowledge, incurs a markedly elevated risk of negative transfer when the spectral signatures of water bodies differ substantially across domains—for instance, between shaded mountainous streams and open plain lakes.
In stark contrast, the tolerance rough set offers a compelling alternative characterized by four distinctive advantages. First, it is entirely data-driven and requires no prior probabilistic assumptions or membership functions, thus preserving a higher degree of objectivity in knowledge discovery. Second, by substituting the strict equivalence relation of classical rough sets with a tolerance (similarity) relation, it directly processes continuous spectral values without the need for discretization, thereby circumventing information loss that commonly compromises conventional denoising pipelines. Third, it simultaneously performs attribute reduction and noise elimination while maintaining the full classification capability of the original decision system—a property that is particularly valuable for sample purification in high-dimensional remote sensing feature spaces. Fourth, it yields compact, interpretable decision rules that explicitly justify the retention or rejection of individual samples, ensuring full traceability and reproducibility of the data-preprocessing workflow. Collectively, these merits motivate our selection of the tolerance rough set as the foundational preprocessing mechanism, aiming to suppress noise at the data source, upgrade sample quality, and furnish a robust data substrate for stable training and precise inference of the subsequent deep learning architecture.
Despite the progress achieved by existing Fully Convolutional Network (FCN) series models in water body extraction, several critical limitations persist. U-Net variants suffer from a limited receptive field, causing fragmentation of narrow rivers and vulnerability to shadow-induced false positives. Transformer-based architectures (e.g., Swin-Unet) model long-range dependencies but incur substantial computational overhead (>50 M parameters), hindering rapid deployment. Lightweight networks achieve high speed at the cost of accuracy in complex scenes involving water–shadow confusion and fine boundary delineation [
40,
41,
42,
43].
To address these intertwined challenges—limited receptive fields, semantic ambiguity, insufficient boundary refinement, and the accuracy–efficiency trade-off—our FCN-RAM offers three distinctive advantages: (1) semantic-guided dual-dimensional attention recalibration, which dynamically enlarges the receptive field and suppresses shadow-induced noise without inference overhead; (2) tolerance rough set-based sample purification, a model-agnostic preprocessing that eliminates inconsistent training samples and mitigates class imbalance; (3) superior accuracy–efficiency trade-off, achieving state-of-the-art accuracy across resolutions (0.3–10 m) with only 32.38 M parameters and 43 images/second inference speed. Collectively, these innovations bridge the gap between academic precision and operational feasibility for automated flood surveillance.
Among various attention mechanisms, the Convolutional Block Attention Module (CBAM) has been widely adopted in remote sensing segmentation tasks due to its sequential application of channel and spatial attention for adaptive feature refinement. However, CBAM treats channel-wise and spatial-wise attention as largely independent operations, with spatial attention weights derived solely from channel-compressed shallow features via pooling operations. Oktay et al. proposed Attention U-Net, which integrates learnable attention gates into the standard U-Net architecture to automatically focus on target structures of varying shapes and sizes, thereby improving segmentation accuracy without requiring external organ localization modules [
44]. This design inherently suffers from two critical limitations when applied to cross-level feature fusion in encoder–decoder architectures: (i) the spatial attention map is generated without semantic guidance from deeper layers, making it susceptible to low-level ambiguities such as shadows and textural noise; and (ii) the independent processing of the two dimensions fails to establish cross-dimensional interdependencies, thereby limiting its capacity to resolve the semantic gap between shallow spatial details and deep categorical information [
45,
46,
47,
48,
49].
To overcome these deficiencies, we propose a Residual Attention Module (RAM) that innovatively leverages deep semantic features as conditional guidance to jointly recalibrate both channel and spatial responses of shallow features. Unlike CBAM, RAM first extracts global channel descriptors from the concatenation of shallow and deep features, then computes channel attention weights that are contextually informed by high-level semantics. Concurrently, for spatial attention, RAM adopts a gated fusion strategy in which deep features serve as a querying gate to modulate the contribution of shallow spatial cues, rather than relying on naïve channel averaging. This design ensures that the resulting attention maps are semantically aware and cross-dimensionally coordinated, effectively bridging the representational gap between encoder and decoder features while suppressing task-irrelevant noise. Furthermore, a residual connection is embedded to preserve the original shallow information and facilitate gradient flow, a feature absent in standard CBAM implementations. Collectively, these architectural refinements enable RAM to achieve superior feature enrichment and more precise boundary delineation in high-resolution water body extraction, as quantitatively evidenced in our experimental section.
To address these issues, this study employs a tolerance rough set algorithm for sample data preprocessing to reduce the impact of noise and improve the sample data quality. A Residual Attention Module (RAM) is introduced to optimize the U-Net architecture for water body extraction from high-resolution remote sensing images. Guided by deep features, the RAM attention module adaptively enhances effective features in both the channel and spatial dimensions, establishes interdependencies between these two dimensions, fuses weighted shallow features with corresponding deep features, progressively enriches the representation of effective features, and simultaneously suppresses irrelevant information.
The main contributions of this paper are as follows:
- (1)
An RAM is introduced to optimize the U-Net architecture, constructing a deep learning, fully convolutional network classification and recognition model (FCN-RAM). The model integrates feature information from both channel and spatial dimensions, enriches the semantic information representation of shallow features, and suppresses irrelevant noise, thereby improving the classification and recognition capabilities of the model. Three remote sensing datasets with different resolutions (GLH-Water, GID, and ESWD) were selected for high-resolution remote sensing information extraction. Compared with commonly used mainstream model architectures under the same conditions, the results show that the FCN-RAM model has significant advantages in terms of both sample training efficiency and extraction accuracy.
- (2)
Using the tolerance rough set algorithm to preprocess the sample data reduces the impact of noise, thereby significantly improving the quality of the data samples, providing high-quality sample data for model training, and consequently enhancing the model accuracy and information extraction efficiency.
The remainder of this paper is organized as follows.
Section 2 describes the three experimental datasets and the associated data preprocessing procedures, with particular emphasis on the tolerance rough set-based sample purification algorithm, followed by a detailed exposition of the proposed FCN-RAM architecture, including the RAM design and its integration with the U-Net backbone.
Section 3 presents comprehensive experimental results on three benchmark datasets, encompassing comparative analyses against both alternative attention modules and mainstream network architectures.
Section 4 discusses the methodological implications, model limitations, and avenues for future improvement. Finally,
Section 5 summarizes the key conclusions of this study.
3. Results
3.1. Ablation Study on the Similarity Threshold δa in Tolerance Rough Set Preprocessing
The tolerance rough set preprocessing module introduced in
Section 2.1.2 involves a critical parameter—the similarity threshold
δa—which determines the tolerance relation between sample pairs and consequently governs the granularity of sample purification. To systematically investigate the sensitivity of model performance to this parameter and to identify its optimal value, we conducted a series of ablation experiments on the GLH-Water dataset, varying
δa across a range of candidate values while keeping all other experimental configurations identical.
Following the parameter selection practices in neighborhood rough set-based remote sensing classification studies, we set δa to vary within [0.01, 0.20] with an interval of 0.01, yielding 20 candidate configurations. For each δa value, the tolerance rough set preprocessing was applied to the training samples according to Equation (2), followed by FCN-RAM model training and evaluation on the validation set. The Overall Accuracy (OA) and Intersection over Union (IoU) were recorded as the primary performance metrics. To ensure statistical reliability, each configuration was repeated three times with different random seeds, and the average results are reported.
Figure 5 presents the variation in
OA and
IoU with respect to
δa on the GLH-Water validation set. As
δa increases from 0.01 to 0.20, both
OA and
IoU exhibit a clear inverted U-shaped trajectory. At extremely low
δa values (
δa ≤ 0.03), the tolerance relation is overly strict, causing the algorithm to retain nearly all samples—including those with minor spectral inconsistencies—resulting in minimal noise reduction and suboptimal performance (
OA ≈ 94.2%,
IoU ≈ 84.5%). As
δa increases to the moderate range of 0.06–0.10, the tolerance relation becomes sufficiently permissive to identify and eliminate inconsistent samples while preserving the integrity of the majority of clean samples. Within this range, the model achieves its peak performance, with
OA reaching 98.6% and
IoU attaining 90.8% at
δa = 0.08. When
δa exceeds 0.12, the tolerance relation becomes overly loose, causing the algorithm to treat spectrally distinct samples as “tolerant” and thereby removing samples that contain legitimate spectral variability. This over-purification leads to a loss of discriminative information and a subsequent decline in performance (
OA ≈ 95.7%,
IoU ≈ 86.3% at
δa = 0.20).
Based on the above results, δa = 0.08 was selected as the optimal similarity threshold for the GLH-Water dataset, as it consistently yielded the highest validation performance across all three random seeds. To further validate the generalizability of this parameter choice, we applied the same δa = 0.08 configuration to the GID and ESWD datasets without retuning. The resulting performance (OA = 97.37% on GID, 95.12% on ESWD) remained highly competitive, suggesting that δa = 0.08 serves as a robust default setting across different resolutions and land-cover characteristics. However, we note that optimal δa may vary with dataset-specific properties such as spectral heterogeneity and signal-to-noise ratio; a systematic tuning on each new dataset is therefore recommended for practical deployment.
3.2. Internal Ablation Study of the Residual Attention Module (RAM)
To validate the individual contributions of each core component within the proposed RAM, we conducted internal ablation experiments on the GLH-Water dataset with five configurations: (1) U-Net baseline, (2) U-Net with channel attention only (CA), (3) U-Net with spatial attention only (SA), (4) U-Net with full RAM but without residual connection (RAM w/o Res), and (5) full FCN-RAM. All configurations were trained under identical conditions and results are summarized in
Table 4.
Channel attention alone contributes an 8.35% IoU gain over U-Net, slightly outperforming spatial attention (+6.84%), indicating that spectral recalibration is particularly critical for water–land discrimination. The combination of both attention mechanisms yields a 13.90% improvement—exceeding the sum of their individual contributions—confirming a synergistic effect between the two dimensions. Crucially, adding the residual connection further boosts IoU by +3.68% (from 87.14% to 90.82%), demonstrating that the identity mapping preserves fine-grained spatial details and stabilizes gradient flow. These results collectively validate that channel attention, spatial attention, and the residual connection each make indispensable contributions to the full FCN-RAM performance.
3.3. Statistical Significance Testing
To verify that the observed performance improvements of FCN-RAM over competing methods are statistically significant rather than attributable to random training variability, we conducted repeated training experiments and performed statistical significance tests following established practices in deep learning-based remote sensing studies.
Each model configuration—including U-Net baseline, U-Net + CA, U-Net + SA, U-Net + RAM w/o Res, and full FCN-RAM—was trained five times with different random seeds while keeping all other hyperparameters identical. For each training run, we recorded the IoU and F1-score on the GLH-Water test set. All five runs for each configuration were conducted under the same hardware and software environment to ensure fair comparison.
Given that the five runs for each configuration were performed on the same test set under identical conditions, we employed paired t-tests (two-tailed, α = 0.05) to compare the performance of FCN-RAM against each baseline and ablation variant. The paired design controls for run-to-run variability and isolates the effect of architectural differences. For each pairwise comparison, the null hypothesis assumes no difference in mean performance between the two configurations; rejection of the null hypothesis (p < 0.05) indicates a statistically significant improvement.
Table 5 summarizes the mean and standard deviation of
IoU and
F1-score across five independent runs for each configuration, along with the
p-values from paired
t-tests comparing each variant against the full FCN-RAM.
All reported values represent the mean and standard deviation computed from five independent training runs with distinct random seeds. Pairwise comparisons between each ablation variant and the full FCN-RAM were conducted using two-tailed paired t-tests at a significance level of α = 0.05; p-values < 0.001 were considered to indicate extremely high statistical significance.
The experimental results lead to three principal observations. First, the standard deviations across repeated runs remain consistently within a narrow range (≤0.31% for IoU and ≤0.28% for F1-score), demonstrating that the training procedure exhibits stable convergence behavior and robust reproducibility irrespective of random initialization. Second, all pairwise comparisons against the full FCN-RAM yield p-values below the 0.001 threshold, confirming that the incremental performance gains conferred by each successive architectural component—channel attention, spatial attention, and the residual connection—are statistically reliable rather than fortuitous. Third, the monotonic decline in standard deviation from the baseline (0.31%) to the complete model (0.18%) suggests that the attention mechanisms and residual mapping not only elevate predictive accuracy but also enhance training stability by mitigating variance arising from initialization. Collectively, these statistical findings provide rigorous empirical evidence that the superior performance of FCN-RAM is not an artifact of stochastic training fluctuations, but rather a robust and reproducible advantage attributable to the architectural innovations introduced in this study.
3.4. Comparative Analysis of Different Attention Modules
The three attention modules—ECA, SimAM, and CBAM—were selected as comparative baselines because they collectively represent the three dominant design paradigms of lightweight attention mechanisms in remote sensing segmentation: channel-only attention (ECA), parameter-free 3D attention (SimAM), and sequential channel–spatial attention (CBAM). Together, they comprehensively bracket the methodological design space, making the comparison both representative and sufficient to demonstrate RAM’s novelty. While other attention modules exist (e.g., SE, GAM, Triplet Attention), they share similar design philosophies with the selected representatives; including them would add redundancy without proportionally increasing insight. Our ablation study in
Section 3.2 further isolates the contribution of each RAM component, complementing the module-level comparison.
To validate the effectiveness of the proposed fused attention module, the U-Net model was used as the baseline, and its performance after embedding the fused attention module was evaluated using three datasets. Furthermore, the superiority of the proposed method is verified by comparing the classification results with those of other attention modules. The attention modules compared include the following:
- (1)
ECA (Efficient Channel Attention): This module replaces the two fully connected layers in SENet with a one-dimensional convolution, achieving local cross-channel interaction without dimensionality reduction while significantly reducing the number of parameters. It is lightweight and efficient, suitable for embedding into the skip connections of U-Net, effectively enhancing the representation capability of river features while maintaining a low computational burden [
60].
- (2)
SimAM (Simple Attention Module): SimAM is a parameter-free 3D attention module. Based on neuroscience theory, it directly measures neuron importance through an energy function, enhancing sensitivity to local river boundaries and improving contour extraction clarity without increasing computational cost. As a lightweight plug-and-play component, it can be flexibly embedded after the convolutional layers of U-Net and is particularly suitable for feature refinement in the spatial attention branch [
61,
62].
- (3)
CBAM: This lightweight attention module was proposed in 2018. It performs attention operations separately in the spatial and channel dimensions. The module consists of two sub-modules: the Channel Attention Module (CAM) and the Spatial Attention Module (SAM), which are used to adaptively weight features in the channel and spatial dimensions, respectively [
63].
- (4)
RAM (proposed in this study): A residual attention module that comprehensively considers channel attention and spatial attention, designed to enhance semantic consistency during feature fusion across different levels.
The experimental results of each attention module on the GLH-Water, GID, and ESWD datasets are shown in
Table 6.
From the comparison of the accuracy results of each module (
Table 6), it can be seen that, compared with attention modules that focus only on a single dimension (spatial or channel), such as ECA, SimAM, and CBAM, the proposed RAM achieves superior performance on all three datasets (GLH-Water, GID, and ESWD) in terms of Overall Accuracy (OA),
F1-score, and IoU. Among the compared modules, the CBAM ranked second, whereas the ECA performed the worst. Specifically, on the GLH-Water dataset, the IoU accuracy of the RAM was 16.13% higher than that of the ECA. On the ESWD dataset, the overall accuracy of RAM was 10.10% higher than that of ECA. Furthermore, considering the spatial resolution of each dataset, the extraction accuracy of all modules is positively correlated with the image resolution; the GLH-Water dataset has the highest resolution and achieves the best extraction accuracy; the GID dataset is intermediate; and the ESWD dataset has the lowest resolution and consequently shows relatively less satisfactory performance. The comprehensive results indicate that the RAM integrates deep feature information from both channel and spatial dimensions. In the channel dimension, the module optimizes shallow features, enlarges the receptive field, and enriches semantic information. Guided by deep features, the spatial dimension enhances the key response regions of shallow features and suppresses irrelevant noise, thereby accelerating the fusion of dual-dimensional information. The RAM exhibits significant advantages in deep feature extraction and semantic segmentation of critical regions, with extraction accuracy and classification performance clearly superior to those of the other modules. Moreover, the RAM has good flexibility and compatibility; it can be naturally embedded into skip connection layers, improving the performance with almost no additional computational cost. It also exhibited good transferability and could be extended to other network architectures.
From the feature extraction result figures, it can be intuitively observed that the extraction performance of each module is generally consistent with the quantitative results in the accuracy comparison table. The RAM exhibited fewer omissions and misclassifications, maintained continuity in extracting elongated rivers, preserved complete boundary details, and was less affected by cloud cover and ground-object shadows, thereby achieving the best overall performance. The CBAM ranks second, being more susceptible to the influence of cross-river structures (e.g., bridges), and shows weaker capability in identifying small river branches. The ECA module suffers from more omissions and misclassifications, mainly concentrated in areas with fine, dense features, and severe ground shadows; its visual results are insufficient to meet practical application requirements. The comprehensive results indicate that the RAM has an adequate receptive field in high-resolution remote sensing images, well-maintained boundary refinement and spatial details, strong adaptability to multi-scale targets, and better overall classification performance than expected.
3.5. Comparative Analysis with Other Network Architectures
To validate the effectiveness of the proposed FCN-RAM model (which integrates the RAM attention module), comparative experiments were conducted with existing mainstream semantic segmentation models on three remote sensing image datasets. The classification performance of the improved model was evaluated against that of the commonly used U-Net architecture, and comparisons were made with other network architectures. The networks involved in the comparison include the following:
- (1)
Attention U-Net: U-Net is a symmetric U-shaped encoder–decoder architecture. Its core lies in skip connections, which concatenate high-resolution features from each encoder layer with the corresponding decoder layers, compensating for detail loss caused by downsampling and achieving precise pixel-level localization [
64]. Oktay et al. (2018) proposed Attention U-Net, which integrates learnable attention gates into the standard U-Net architecture to automatically focus on target structures of varying shapes and sizes, thereby improving segmentation accuracy without requiring external organ localization modules.
- (2)
ResNet: ResNet addresses the difficulty of training deep networks by introducing residual connections. Its core is the residual block: instead of directly fitting the desired underlying mapping
, the network fits a residual function
, and the output is expressed as
. This additive identity mapping allows gradients to backpropagate directly through shortcut connections, effectively alleviating gradient vanishing and enabling network depths of hundreds or even thousands of layers without performance degradation [
65]. ResNet has established the basic design paradigm of modern deep convolutional neural networks.
- (3)
Water-SCNet: Water-SCNet is a cascaded dual-network architecture designed for urban river segmentation and break repair. It uses a soft attention mechanism to focus on broken areas and infers river connections occluded by roads or bridges, achieving end-to-end processing from segmentation to topological repair [
66].
- (4)
FCN-RAM (proposed): The network model proposed in this paper that integrates the attention module. By introducing the Residual Attention Module (RAM) to optimize the U-Net architecture, it integrates feature information from both channel and spatial dimensions, uses deep features as guidance, enriches the semantic information representation of shallow features, suppresses irrelevant noise, and thereby improves the model’s classification and recognition capability.
The experimental results of different network architectures on the GLH-Water, GID, and ESWD test sets are shown in
Table 7.
From the accuracy comparison results of the different network architectures (
Table 3), it can be seen that the proposed FCN-RAM model architecture outperforms other commonly used model architectures on all three test sets (GLH-Water, GID, and ESWD) in terms of the Overall Accuracy (
OA),
F1-score, and IoU. Specifically, on the GLH-Water dataset, the
IoU accuracy of the FCN-RAM was 17.59% higher than that of standard U-Net. For the GID dataset, the overall accuracy of the FCN-RAM was 11.15% higher than that of the standard U-Net. Considering the spatial resolution of each test set, the extraction accuracy of different network architectures is positively correlated with the image resolution; the GLH-Water test set still achieves the best extraction accuracy, and the overall classification accuracies on the GID and ESWD test sets are comparatively lower. Overall, the FCN-RAM model exhibited good adaptability to waterbody segmentation tasks on high-resolution remote sensing images of different resolutions.
The recognition result maps of different network architectures on the GLH-Water, GID, and ESWD test sets are shown in
Figure 12,
Figure 13 and
Figure 14, respectively.
The feature extraction results of the test set clearly show that the FCN-RAM model rarely suffers from omissions or misclassifications. It clearly distinguishes land cover features with similar spectral characteristics, preserves details relatively completely, automatically identifies and eliminates cross-river structures (e.g., bridges), and extracts continuous and complete river information. The overall performance meets the requirements for high-resolution image information extraction. The Water-SCNet model also performed well, particularly in urban river extraction areas where the continuity of river features was well maintained. However, they are more severely affected by ground-object shadows, resulting in a certain proportion of misclassifications. The traditional U-Net and ResNet models suffer from a significant loss of key feature information, with relatively high rates of misclassification and omission.
3.6. Qualitative Comparison and Error Mechanism Analysis
To elucidate the failure mechanisms of competing methods and the advantages of FCN-RAM, we conducted a systematic visual analysis of segmentation results on challenging scenes from the GLH-Water test set, with error pixels categorized into three principal types, as summarized in
Table 8.
We selected four representative scenarios to analyze the root causes of misclassifications. (Case 1) urban river with bridge occlusion, (Case 2) mountainous region with terrain shadows, (Case 3) narrow river network with complex branching, and (Case 4) lakes with small islands and irregular shorelines.
Case 1—Bridge Occlusion: U-Net fragments the river at bridge crossings due to its limited receptive field, which fails to capture long-range contextual continuity. ResNet partially mitigates this but still lacks cross-scale semantic integration. Water-SCNet maintains continuity via its bridge-repair module but produces FPs on dark rooftops due to over-generalized connectivity rules. By contrast, FCN-RAM accurately segments continuous rivers while preserving bridge boundaries: the deep-feature-guided spatial attention enhances water-corridor activation and suppresses bridge regions, while the RAM residual connection preserves fine structural details through upsampling.
Case 2—Terrain Shadows: U-Net and ResNet misclassify large shadowed areas as water because their encoders learn spectral decision boundaries insufficient to separate water from shadows, lacking contextual cues that shadows are adjacent to topographic highs and have distinct textures. ECA performs similarly poorly due to limited spatial refinement; SimAM improves slightly but still yields scattered FPs. FCN-RAM substantially reduces shadow-induced errors: the spatial attention branch, guided by deep semantics, attends to flat/open water geometries rather than elongated shadow patterns, while the tolerance rough set preprocessing eliminates training samples with inconsistent spectral-label mappings in shadow-prone regions.
Case 3—Narrow River Network: U-Net fails to detect most narrow branches (FN) and fragments the main channel, due to class imbalance (water pixels are a small minority) and repeated pooling that progressively erodes thin structures. ECA and SimAM still struggle with the finest branches because their attention weights are computed on aggregated feature maps where thin-structure information has been diluted. FCN-RAM recovers most narrow tributaries with minimal fragmentation: the channel attention emphasizes water-discriminative spectral bands (e.g., NIR), while the residual connection ensures that high-resolution shallow features are preserved as an additive component, maintaining structural integrity through the network.
Case 4—Irregular Shorelines and Small Islands: U-Net produces smoothed shorelines and misclassifies small islands due to loss of high-frequency spatial details through pooling. Water-SCNet preserves more boundary detail but introduces jagged artifacts along shorelines. FCN-RAM achieves the closest boundary alignment: the spatial attention branch, guided by deep features, up-weights boundary-pixel contributions in shallow representations, sharpening edge responses; the residual connection preserves original high-frequency components, enabling accurate identification of small islands through textural differences captured by the channel–spatial synergy.
The case analyses reveal a consistent pattern across error types: U-Net and ResNet are vulnerable to shadow-induced FPs (Type I) due to absent semantic guidance; U-Net, ECA, and SimAM struggle with narrow water omissions (Type II) due to limited receptive fields and diluted shallow features; ResNet and U-Net exhibit coarse boundary delineation (Type III) from insufficient edge-preservation. FCN-RAM addresses all three deficiencies through deep-feature-guided spatial attention (suppresses shadow FPs), residual-preserved shallow features (recovers narrow water bodies), and channel–spatial synergy (enhances boundary discriminability). These mechanisms are architecturally complementary, providing a holistic improvement that surpasses the sum of individual enhancements—as quantitatively evidenced by the ablation study and statistically validated by significance testing.
3.7. Model Complexity and Inference Efficiency
To complement the accuracy-oriented evaluation, we assessed model complexity and inference efficiency. Parameter counts (Params) and FLOPs were computed using the thop library, and inference time was averaged over 100 forward passes on 512 × 512 patches using an NVIDIA RTX 3090 GPU under identical conditions, as summarized in
Table 9.
Among attention variants, RAM adds only 1.34 M parameters (+4.3%) and 1.14 G FLOPs (+4.0%) over U-Net, with a moderate 4.4 ms latency increase (+23.5%), yet delivers a substantial 17.58 percentage-point IoU gain (73.24 → 90.82%). CBAM incurs slightly higher parameter overhead (32.16 M) but yields markedly lower accuracy (84.23%), confirming that RAM achieves a superior accuracy–efficiency trade-off. Against full architectures, FCN-RAM outperforms ResNet and Water-SCNet by 13.50 and 7.87 IoU points, respectively, while using only 72.5% and 66.2% of their parameters and reducing inference latency by 3.7 ms and 11.4 ms. This efficiency stems from RAM’s lightweight design—attention weights are computed via compact fully connected layers and small-kernel convolutions—and its integration within skip connections without overburdening the decoder. Despite a modest complexity increase over U-Net, FCN-RAM achieves 43 images per second on an RTX 3090, well within the operational requirements for near-real-time flood monitoring from high-resolution satellite imagery.
4. Discussion
Compared with current mainstream semantic segmentation models, the proposed FCN-RAM model achieves significant improvements in overall accuracy (OA), F1 score, and IoU across three different high-resolution datasets. These results demonstrate that the proposed framework exhibits superior performance in floodwater extraction tasks, specifically in terms of enhancing the semantic information expressed in shallow features. The RAM operates by first extracting global information along the channel dimension. This information is then combined with semantic information from the corresponding deep features to generate attention weights, which serve to refine the shallow features and enrich their semantic content. Concurrently, attention weights along the spatial dimension further highlight key response regions within the shallow features while suppressing, to a limited extent, the original noisy regions.
Compared with common attention modules that focus exclusively on a single dimension (i.e., channel or space), such as ECA and SimAM, the proposed RAM exhibits greater flexibility. The RAM simultaneously integrates deep feature information from both channel and spatial dimensions. Additionally, it introduces a residual mapping branch to alleviate the vanishing gradient problem, which effectively enhances its capacity for fusing channel- and spatial-dimensional information. On the three tested datasets (GLH-Water, GID, and ESWD), the RAM outperforms other attention modules across all three metrics (OA, F1 score, and IoU), followed by the CBAM, with the ECA module performing the worst. Specifically, on the GLH-Water dataset, the IoU accuracy of the RAM is 16.13% higher than that of the ECA module; on the ESWD dataset, the OA of the RAM is 10.10% higher than that of the ECA module. It is worth noting that the CBAM also employs a dual-attention mechanism for feature refinement, which bears some similarity to our approach. The key distinction is that the RAM uses deep features as a guide to enhance key response regions in shallow features while suppressing irrelevant noise, thereby effectively reducing the semantic gap between shallow features and their corresponding deep features.
Compared with the baseline model, U-Net, the RAM significantly boosts the performance of the U-Net architecture for water body identification in high-resolution remote sensing images. The FCN-RAM model achieves an F1-score of 95.19% on the GLH-Water dataset, which is nearly 11% higher than that of U-Net. On the GID and ESWD datasets, the F1-scores of FCN-RAM reach 93.38% and 87.92%, respectively. These results collectively demonstrate that FCN-RAM delivers exceptionally robust performance in water body segmentation across diverse high-resolution remote sensing images.
In a comparison with State-of-the-Art Methods (2024–2025), to contextualize the competitiveness of FCN-RAM, we compare its performance against recent high-resolution water segmentation methods. AMU-Net achieved IoU scores of 93.6% and 95.02% on the GID and WHDLD datasets, respectively, surpassing our GID IoU of 87.58%, albeit with a substantially heavier architecture and higher computational cost. DRFormer reported F1-scores of 95.28–96.67% on multiple urban water datasets, while HyMambaNet achieved 81.30% IoU and 89.99% F1 on the LoveDA dataset—substantially lower than our GLH-Water results (90.82% IoU, 95.19% F1). HWBENet and WB-Former demonstrated competitive accuracy but were primarily validated on urban-centric or rural datasets, with reported metrics not directly comparable to our benchmarks. These comparisons, while constrained by the lack of standardized benchmarks, suggest that FCN-RAM achieves a performance that can compete with state-of-the-art approaches on diverse datasets, with the additional advantages of moderate architectural complexity and a model-agnostic preprocessing paradigm.
4.1. Adaptability and Deployment Limitations in Complex Flood Scenarios
While FCN-RAM performs robustly on the benchmark datasets, its deployment in real-world flood scenarios—particularly turbid flash floods and urban waterlogging—entails several critical limitations. In mountainous flash floods, high suspended sediment loads significantly alter water spectral reflectance, narrowing the separation between water and surrounding land. The current training datasets predominantly comprise clear-water scenes; performance on hyper-turbid floodwater remains untested. For urban waterlogging, the built environment introduces numerous sources of spectral confusion—dark asphalt, building shadows, and blue/green rooftops may all exhibit water-like reflectance. While the RAM suppresses shadow-induced false positives to some extent, urban flooding scenarios with heterogeneous water depths and dynamic evolution during storm events are not fully represented in our training data. Operational deployment faces additional constraints. First, reliance on high-resolution (≤1 m) optical imagery limits applicability in data-scarce regions; performance on medium-resolution (10–30 m) imagery may degrade due to mixed-pixel effects. Second, the tolerance rough set preprocessing requires sufficiently large and representative training sets; its effectiveness may be compromised in developing countries where labeled data are scarce. Third, the current model performs static segmentation without incorporating temporal information, precluding change detection between pre-flood and during-flood imagery to isolate inundation from permanent water bodies. To mitigate these limitations, several strategies are worth pursuing: augmenting training data with synthetically generated turbid water samples or public flood datasets (e.g., FloodPlanet); incorporating SAR data (e.g., Sentinel-1) as an additional modality to overcome cloud cover; and applying model compression techniques such as quantization and pruning for edge-device deployment.
4.2. Contribution of Tolerance Rough Set Preprocessing to Segmentation Accuracy
The proposed framework incorporates a tolerance rough set-based preprocessing module prior to model training, aiming to enhance sample quality by mitigating noise and redundant features. Although direct ablation experiments comparing model performance with and without this preprocessing step are not explicitly presented in the experimental section, its contributions can be assessed from multiple perspectives.
From a theoretical standpoint, the tolerance rough set algorithm operates on the decision table constructed from sample attributes, where the tolerance relation replaces the strict equivalence relation of classical rough sets to accommodate continuous spectral values. This mechanism enables the identification and elimination of inconsistent samples—those with similar conditional attributes (spectral features) but conflicting decision attributes (water/non-water labels)—which are particularly prevalent in regions affected by shadows, cloud remnants, or mixed water–land pixels. By removing such samples, the preprocessing effectively reduces the label noise that would otherwise mislead gradient descent during training, leading to more stable convergence and improved generalization.
From an empirical perspective, the consistently high performance of FCN-RAM across all three datasets—particularly the substantial improvements over U-Net in IoU (17.59% on GLH-Water, 15.66% on GID, and 13.63% on ESWD)—can be partially attributed to the enhanced data quality delivered by the rough set preprocessing. Notably, the improvement margin is most pronounced on the GLH-Water dataset, which has the highest spatial resolution (0.3 m) and thus the most heterogeneous spectral signatures. This observation suggests that the denoising effect of tolerance rough set is particularly beneficial when handling high-resolution imagery where fine details and noise are intimately mixed.
Furthermore, compared with conventional preprocessing techniques such as median filtering or data augmentation, the tolerance rough set offers distinct advantages in the context of water body extraction. As discussed in the Introduction, median filtering tends to blur narrow water channels due to its fixed kernel size, while data augmentation fails to address label noise at the sample level. In contrast, the rough set-based approach selectively prunes noisy samples based on their consistency with the overall decision system, thereby preserving the spatial integrity of water boundaries while improving the overall training signal quality.
A critical point worth emphasizing is that the preprocessing module is model-agnostic—it operates independently of the subsequent segmentation network. This means that the performance gains observed in FCN-RAM over other architectures (e.g., ResNet, Water-SCNet) are not merely architecture-specific, but are built upon a shared foundation of higher-quality training data. In other words, the rough set preprocessing establishes a cleaner and more consistent data basis that benefits all evaluated models, not exclusively the proposed FCN-RAM.
Nevertheless, a quantitative assessment of the isolated contribution of rough set preprocessing—via controlled experiments with and without this module—would provide more definitive evidence. Such ablation studies, as well as the extension of this preprocessing paradigm to other deep learning models and remote sensing tasks, are planned as part of our future work. The ablation study on δa further corroborates the contribution of tolerance rough set preprocessing. The observed inverted U-shaped performance curve is consistent with the theoretical expectation: excessively small δa fails to eliminate sufficient noisy samples, while excessively large δa removes too many samples and discards valuable spectral variability. The optimal range (δa = 0.06–0.10) corresponds to a tolerance level that effectively balances noise suppression and information preservation—a finding that aligns with the parameter sensitivity patterns reported in neighborhood rough set-based remote sensing studies. Notably, the performance improvement from δa = 0.01 (OA = 94.21%) to δa = 0.08 (OA = 98.61%) amounts to 4.40 percentage points, demonstrating that appropriate parameter tuning of the preprocessing module contributes substantially to the overall segmentation accuracy.
4.3. Future Research Directions
Furthermore, the RAM can be applied to each layer of the skip connection, improving model performance while reducing computational time. This modular flexibility implies that the RAM can be embedded into other network architectures in the future, offering good potential for extension. This study has several limitations that require further investigation. First, while the RAM was validated on the U-Net architecture, its effectiveness for other models with similar structures is still unknown and must be evaluated through additional experiments. Such experiments should account for factors including computational resources, the adaptability of the network architecture to specific data characteristics, and the associated parameter settings. Second, the current work was validated only on general water body extraction tasks; future work should extend to fine-scale water identification under more specific conditions. Finally, the FCN-RAM model has a flexible structure that can be extended to a multi-class semantic segmentation model for multi-classification tasks on remote sensing images. These are all worthwhile directions for future in-depth exploration.
Building on the above findings, we outline several concrete directions for future work. Direction 1 is multi-source data fusion: integrating SAR data with optical imagery through a dual-stream architecture with cross-modal attention would enable all-weather flood monitoring, which is particularly critical for cloudy regions such as tropical and monsoon climates. Direction 2 is spatiotemporal modeling: extending FCN-RAM to a spatiotemporal segmentation framework—by incorporating ConvLSTM or Transformer-based temporal attention modules—would enable tracking of flood evolution and differentiation of permanent water bodies from transient inundation. Direction 3 is lightweight model design: investigating network pruning, knowledge distillation, or neural architecture search could yield a lightweight variant of FCN-RAM suitable for UAV or edge-device deployment in rapid post-disaster reconnaissance. Direction 4 is domain adaptation for data-scarce regions: unsupervised domain adaptation and few-shot learning strategies could transfer the FCN-RAM model from data-rich source domains (e.g., GLH-Water) to target regions with limited or no labeled data. Direction 5 is multi-class segmentation: extending the current binary framework to fine-grained multi-class segmentation—distinguishing open water, flooded vegetation, inundated built-up areas, and debris deposits—would provide richer information for disaster response.