This work proposes an Adaptive Multi-feature Balanced network (AMBNet) for semantic segmentation in complex urban remote sensing scenarios. To fully exploit optical images and Digital Surface Models (DSM) data obtained from remote sensing sensors, a Depth Feature Extraction and Balancer (DFEB) module is devised to estimate and balance the depth information of all pixels by capturing detailed structural compositions of the ground surface. After that, a Parallel Multi-Stage Segmentator (PMSS) comprised of a dual-branch Encoder and Decoder with skip connections is constructed to perform effective segmentation by exploiting the balanced DSM (BDSM) and optical information. As a result, the proposed AMBNet can make effective use of optical images to complete depth information, so as to achieve multimodal information-assisted semantic segmentation for complex remote sensing scenes. Comprehensive experiments performed on the ISPRS Vaihingen and Potsdam remote sensing datasets confirm the segmentation performance of the proposed method.
1 Introduction
Semantic segmentation plays a pivotal role within the realm of remote-sensing image processing. Nevertheless, the semantic segmentation of remote sensing images is fraught with distinct technical intricacies when contrasted with its counterpart designed for natural images. For instance, extensive shadows cast by edifices can obscure ground areas whereas the dense canopy of vegetation impedes the identification of vehicles and structures. These intricacies necessitate the employment of more sophisticated techniques in the domain of remote sensing semantic segmentation. Driven by recent breakthroughs in the field of deep learning (DL), a spectrum of DL-based segmentation models have been devised for remote sensing applications [11, 5, 24, 18, 26]. These models can be divided into two main categories according to their network structures, namely the CNN-based networks [18, 17] and the Transformer-based networks [26, 4]. The former offers superior computational efficiency during training and test at the cost of a limited local receptive field, hindering their capability of effectively discerning high-level object semantics. In contrast, the latter enhances the segmentation performance through the utilization of Vit-based encoders [6], effectively capturing intricate long-range dependencies. Nevertheless, Transformer-based models come at a formidable computational cost.
In the meantime, recent researches have suggested that augmenting the input data with multimodal information, such as multispectral imagery (MSI) [14, 25], hyperspectral imagery (HSI) [20, 28], Digital Surface Models (DSm) [23, 30] and light detection and ranging (LiDAR) [21, 24], together with optical images, can significantly enhance the semantic segmentation accuracy, particularly for objects exhibiting similar chromatic attributes on the terrain [7, 29]. For instance, FuseNet [9] and vFuseNet [2] enhanced the encoding of ground elevation data by seamlessly integrating optical images with DSM. Furthermore, CMFNet [19] employed a crossmodal multi-scale Transformer to perform a comprehensive fusion of multimodal data. While the aforementioned methods demonstrate effectiveness, they are not without limitations. Firstly, data labeling poses a challenge as not all remote sensing images across different regions have accessible multi-source data for reference. Secondly, the fusion of multi-source hyperspectral data introduces complexities such as data registration, error correction, and consistency issues. The uncertainty and noise from diverse data sources may propagate into the fusion results, impacting the model’s stability. Thirdly, handling shadows presents difficulties, as they often involve lighting changes, demanding high adaptability from the model. Accurately removing or correcting shadows can prove challenging. Furthermore, these methods assume the reliability of Digital Surface Model (DSM) data, which can be a practical concern. The precision of DSM data depends on various environmental factors, including the topography of the study area. To address this concern, this work proposes a depth estimation method that leverages detailed information from optical sources to generate more accurate and reliable DSM data. This refined approach, referred to as balanced DSM (BDSM), serves as the foundation for subsequent multimodal fusion processes.
Monocular depth prediction endeavors to extrapolate the spatial depth particulars of a scene from a solitary remote sensing image. This depth information can furnish the model with richer contextual insights, facilitating a more profound comprehension of the remote sensing imagery. Simultaneously, it serves to alleviate the impact of terrain irregularities and shadows, proving particularly advantageous in addressing challenges posed by uneven topography, mountainous terrains, and shaded regions. The derived depth information can be employed as an ancillary input for semantic segmentation tasks, enhancing the model’s capacity to discriminate between diverse categories of ground objects. The outcomes of depth prediction may be harnessed to generate sophisticated semantic features, elevating the accuracy of ground object boundary delineation in segmentation tasks.
Multimodal fusion approaches have been developed for many years in the field of computer vision. For example, intensive investigations have been devoted to deriving multi-feature information from spatial geometry and focal length changes. Some pioneering works [3] have been established by utilizing multi-feature fusion to aid semantic segmentation. More specifically, different algorithms are employed in feature-based image data fusion to extract features from various data sources before fusing these features. For instance, color transformation [15] and spatial-to-frequency domain transformation [16] are utilized to extract geometric and spectral characteristics of the target, including range, shape, neighborhood, texture, relative positioning, and spectral information [8]. Compared with natural images, multimodal fusion for remote sensing images encounters more challenges due to the following facts: (1) Remote sensing images are mostly collected by unmanned aerial vehicles or satellites in orbit in the form of a top view of limited angles; (2) The objects in remote sensing images are more complex, including varying scales, irregular shapes or boundaries. These problems require more sophisticated network designs for remote sensing multi-feature fusion.
To address these challenges, an Adaptive Multi-Feature Balancing Network (AMBNet) is proposed for semantic segmentation of complex urban remote sensing scenes in this work. It utilizes a deep feature extraction and balancer (DFEB) for depth information extraction and balance before exploiting a parallel multi-stage segmentator (PMSS) for multimodal data fusion. Specifically, DFEB utilizes a height estimator and feature merge balancer to improve the quality of DSM by combining DSM and ground features for optical images, using Gaussian filtering [13] to smooth the DSM data. In particular, the proposed DFEB can effectively reduce noise while preserving image details. To assess the effectiveness of the feature fusion process, we propose to evaluate the similarity between remote sensing images and the corresponding enhanced BDSM features using performance indicators such as SSIM, perceptual hashing, LBP, and NMSE. After that, the generated BDSM and optical images are passed into the PMSS equipped with parallel ResNet Layers and decoder layers with skip connections for the final segmentation prediction. By leveraging depth estimation and multimodal data fusion, the proposed AMBNet opens up new prospects for semantic segmentation of remote sensing images. Extensive experiments are conducted on ISPRS Vaihingen and Potsdam remote sensing datasets to verify the performance of the proposed AMBNet in multimodal remote sensing segmentation tasks.
2 Proposed AMBNet
2.1 Adaptive Multi-Feature Balancing Network (AMBNet)
Figure 1 illustrates the schematic of the proposed AMBNet consisting of two novel modules, namely DFEB and PMSS. To explore multimodal information more efficiently, the proposed AMBNet is optimized in an end-to-end manner. As shown in Figure 1, both optical and DSM are input into AMBNet in which the DFEB module generates the BDSM by exploiting auxiliary crossmodal information. More specifically, the proposed DFEB integrates optical and DSM information using a variety of techniques, including Laplacian pyramid decomposition [27], Gaussian blur, and weighted fusion. Multiple fusion evaluation metrics can be employed, including SSIM, Perceptual Hashing, LBP analysis, and NMSE, to facilitate the feature matching between the fused feature map and the pre-processed optical image, thereby assessing the quality of the fusion process. Given the inherent imprecision and lack of fine-grained detail in much of the DSM data, especially with regard to objects such as vehicles, the richness of detailed feature information is primarily embedded in the optical image. However, the original optical image is burdened with excessive high-frequency information that does not always align with the fusion outcome. To address this challenge, comparison is drawn between the Laplacian pyramid-pre-processed results and the fused DSM data. Finally, the resulting BDSM is fed into PMSS along with the raw optical images for final semantic segmentation. The proposed PMSS undertakes multi-dimensional feature extraction from both the enhanced DSM information and the original optical images, culminating in the production of the final semantic segmentation output after upsampling.
2.2 Deep Feature Extraction and Balancer (DFEB)
The proposed DFEB module, as depicted in Figure 2, embodies two distinct components, namely the height estimator and the feature merge balancer. The former is designed for estimating the height of ground objects, consisting of a classical encoder-decoder structure that can output highly recognizable estimated DSM (EDSM) based on optical images. Upon receiving EDSM and DSM, the latter derives reliable BDSM by weighing EDSM and DSM with the Laplacian pyramid.
Height Estimator: Stemmed from its foundational ResNet [10], the height estimator capitalizes on the pivotal MidasNet [22] while introducing feature screening together with channel attention mechanisms [12]. Furthermore, the amalgamation of feature channels is optimized through the back-propagation process. The structure of the height estimator is shown in Figure 3. It is comprised of four ResNet layers, four skip connections, four corresponding Decoder layers and a channel selection module. More specifically, ResNet layers processes the optical image input denoted as X ∈ ℝH×W×3. The feature maps of four different scales are obtained by four ResNet layers that also serve as skip connections. Four decoder layers recover the spatial and contextual information to generate the corresponding depth feature maps denoted as 𝒬 ∈ ℝH×W×64. Finally, a channel attention selection module is implemented to assign weighting coefficients to useful information. We denote by Fse(•) the squeeze and excitation function, and the final EDSM can be generated as follows:
More specifically, the channel attention mechanism incorporated in the framework of monocular depth estimation draws its inspiration from SENet (Squeeze and Excitation Network), wherein the conventional Rectified Linear Unit (ReLU) activation function is substituted with the Tank function. This strategic replacement allows for the comprehensive utilization of both positive and negative tensor information present in the depth feature map. It is imperative to note that this is a channel-centric attention model specifically designed to capture and quantify the significance of individual feature channels. Consequently, it facilitates the selective amplification or attenuation of specific channels tailored to different task requirements. Following the convolutional operation, a branch for bypass is introduced, commencing with the Squeeze operation. This operation condenses the spatial dimensions into feature values, effectively collapsing each two-dimensional feature map into a scalar value. Essentially, it approximates a pooling operation with a global receptive field. Remarkably, the total count of feature channels remains unaltered throughout this process.
2.3 Feature Merge Balancer (FMB)
As shown in Figure 4, the proposed Feature Merge Balancer employs a multi-faceted approach to fuse the EDSM and DSM data by exploiting the Laplacian pyramid method and the Gradient domain fusion (GDF). More specifically, three fusion images denoted as IMG1, IMG2 and IMG3 are generated and compared against the RGB image separately in terms of four metrics, namely Structural Similarity Index (SSIM), Perceptual Hashing, LBP, and NMSE. After that, the model derives the Overall Similarity (OS) from the four resulting indicators before the optimal fusion solution is selected. In particular, since SSIM is specifically designed to quantify the similarity between two images, it assesses the perceived quality of an image by considering various aspects of the structural information and luminance conditions that are important factors for human visual perception. Finally, adaptive weights are assigned to the four indicators to provide comprehensive evaluation on the Overall Similarity.
The Laplacian pyramid shown in Figure 5 is a multi-scale image representation. Firstly, the original image is downsampled for multiple times to generate a series of images of different resolutions, each of which is a blurred version of the original image, which develops a Gaussian pyramid [1]. We denote by G0 and Gi the input image and the i-th level image of the Gaussian pyramid, respectively. It is worth noting that the higher-level images have lower resolution.
Next, we create a Laplacian pyramid by subtracting the Gaussian image of an adjacent level from one level in the Gaussian pyramid. This step generates a series of images, each of which contains details of the original image at different spatial frequencies. This process can be mathematically modeled as:
where Expand (•) models the process of upsampling, mapping the pixel at position (x, 𝓎) in the source image to the position (2x + 1, 2𝓎 + 1) of the target image. Furthermore,⊗ stands for the convolution operator.
To fuse two images, the specific structure described above performs the same operation on the Laplacian pyramids of two inputs before the Laplacian images at the corresponding levels are added together with weights. The weights are usually determined by a fusion mask that defines the image’s information that should dominate at each level. Note that the variance is large for the current local block in the detail areas of the Laplacian image. In contrast, the variance is small for the smooth areas.
2.4 Parallel Multi-Stage Segmentator (PMSS)
The proposed PMSS is designed to perform the multimodal feature fusion and subsequently, generate the final segmentation map. As shown in Figure 6, two parallel encoders comprised of multiple ResNet Layers are utilized to encode BDSM and optical images. More specifically, the encoders are divided into four blocks with the output feature maps being denoted as Mopticai ∈ RH×W×Ci and MDepth ∈ RH×W×Ci where C ∈ {64,128, 256, 512} for i = 1, 2, 3,4. After each downsampling operation, the BDSM is fused into the optical encoder as the auxiliary information to the next downsampling operation. Mathematically, the resulting post-fusion feature map MFusion can be modeled as follows:
To augment the preservation of intricate long-term details, skip connections have been strategically incorporated, facilitating seamless information flow from the encoder to the decoder components. Ultimately, the process culminates with the fusion of depth information and the upsampled feature maps, which is shown to be a critical precursor for improving the segmentation performance in the following operation.
2.5 Loss and Activation functions
The following loss function LFMB is employed in this work:
where {wi} with i = 1,2,3,4 stands for the weighting coefficients designed to reflect the relative importance of the corresponding performance metric during the training process.
In addition, the softmax function is used as the activation function in the classification process, which maps the output of multiple neurons to the probability within the interval of (0,1) for multi-class classification. Furthermore, the softmax function is used in conjunction with the cross entropy loss function to avoid the problem of numerical overflow.
The following softmax activation function Si and softmax loss function Lsft are employed in this work:
where pi is the output value at the i-th node while C is the total number of output nodes or classes. Furthermore, [P]i is the i-th value of the output P, which represents the probability that this sample is classified into the i-th category. Finally, qi = 1 if the sample actually belongs to the i-th category. Otherwise, we set qi = 0.
Using the softmax function Si as the probability distribution of pi, we can obtain the Cross Entropy (CE) loss function as follows:
If there are multiple samples, the CE loss function can be computed as the mean value over all samples.
Finally, the following overall loss function combines LFMB and LCE to enable diverse optimization objectives.
where γ ∈ [0,1] is a weighting coefficient. Adjustment of γ allows the network to learn various aspects of knowledge. As a result, the network can enhance image fusion while simultaneously improving pixel-level classification accuracy.
3 Fusion Quality Metrics
3.1 SSIM (Structural Similarity)
SSIM is a measure designed to detect the similarity of two given images of the same size by comparing the brightness, contrast, and structure of the two images.
where I(x,y), c(x,y) and s(x,y) are the brightness contrast function, the contrast function and the structural contrast function, respectively. These functions are defined as follows:
where μx, σx and μxy are given below:
Finally, constants C, C2 and C3 are introduced for mathematical stability by preventing the division-by-zero problem in the denominator.
It is worth noting that these formulae possess inherent clarity, necessitating the computation of mean and variance for each of the two images independently, followed by the derivation of their covariance, which is subsequently incorporated into the SSIM formula for evaluation. The resultant computation output represents an image, effectively illustrating the aliasing artifacts present in both images.
3.2 Local Binary Pattern (LBP)
LBP is an image-processing metric for texture analysis. The calculation of the LBP features is based on the neighborhood of pixels, usually using 3 × 3 or 5 × 5 neighborhoods. We consider a central pixel with gray value of Ic surrounded by K pixels whose gray values are denoted as IP1,IP2,…, IPK. Then, the LBP value of the central pixel is computed by comparing its gray value against those of its neighboring pixels. If the gray value of the neighboring pixel is greater than or equal to the central pixel, then we count the pixel in the LBP calculation with pre-defined weighting. Otherwise, the contribution of the neighboring pixel to the LBP value of the central pixel is ignored. Mathematically, the computation of the LBP value can be expressed as follows:
where
Finally, a K-bit binary LBP code is formed by connecting these binary codes together.
After obtaining the LBP code, we can utilize the histogram intersection kernel for similarity measurement. More specifically, we denote by H1 and H2 two histograms of LBP features with each comprising 𝓁 bins, e.g. 𝓁 = 256 for 8-bit LBP representation. The calculation of the histogram intersection kernel is delineated as follows:
Note that Eq. (18) stands for the similarity between two histograms by measuring their overlaps. A larger similarity value indicates that the texture features of the two images are more similar.
In summary, SSIM and LBP serve as two image quality evaluation metrics grounded in human perception. They are primarily employed for assessing the structural similarity and perceptual quality of images, showcasing relevance to the accuracy of semantic segmentation in deep learning, particularly in the context of multi-feature fusion for remote sensing scenarios.
4 Experiments Configuration
4.1 Datasets
The experiments involves the utilization of the well-known Potsdam and Vaihingen datasets provided by ISPRS for training and validation purposes. These datasets are commonly considered for urban classification and 3D building reconstruction projects. They incorporate a digital terrain model (DSM) derived from high-resolution orthophotos and dense image-matching technology. Both dataset areas encompass urban environments, with Vaihingen representing a compact village featuring numerous standalone structures and small multi-story buildings, while Potsdam epitomizes a historical city characterized by grand edifices, narrow thoroughfares, and a densely populated layout. Each dataset has undergone manual categorization into the five most prevalent land cover classes. To evaluate the efficacy of our AMBNet, the DSM data from both the Potsdam and Vaihingen datasets is chosen as comparative experiments. More specifically, we divided the Vaihingen dataset into the training and test sets as follows:
Training set: 1, 3, 23, 26, 7, 11, 13, 28, 17, 32, 34, 37;
test set : 5, 21, 15, 30.
Furthermore, the Potsdam dataset was divided into the training and test sets as follows:
Training set: 6_10, 7_10, 2_12, 3_11, 2_10, 7_8, 5_10, 3_12, 5_12, 7_11, 7_9, 6_9, 7_7, 4_12, 6_8, 6_12, 6_7, 4_11;
Test set: 2_11, 3_10, 4_10, 5_11, 6_11, 7_12.
4.2 Segmentation Evaluation Method
Overall accuracy (OA), mean MIoU, and F1 Score are used for semantic segmentation accuracy analysis. We first define “Precision” as the proportion of all samples with positive predictive values and positive real values whereas “Recall” the proportion of samples with positive predicted values among all samples with positive real values:
where TPc, FPc and FNc denote true positives, false positives and false negatives for the c-th class, respectively.
Subsequently, OA, IoU and F1 Score can be defined as follows:
where TNc is the true negatives for the c-th class. mloU and mF1 Score are the mean of IoU and F1 Score of the five main categories, respectively.
5 Results and Discussions
5.1 Fusion Example Derived from Laplacian Pyramid
Figure 7 illustrates an example derived from the Laplacian pyramid fusion in the proposed FMB. As shown, the DSM and EDSM (depth estimation) information is input into the Laplacian pyramid. The input is then represented in multi-scale space before image fusion is performed between adjacent layers. Finally, the fused image is generated by summing the output from each level of the Laplacian pyramid.
5.2 Depth Estimation Example
The proposed AMBNet consists of two parts, DFEM and PMSS. The former estimates the height and produces the BDSM while the latter performs the semantic segmentation by exploiting optical and BDSM. In this section, we will demonstrate and analyze the performance of depth estimation and semantic segmentation.
In DFEB, the height estimator can effectively explore the detailed information from the optical images before the feature merge balancer improves the quality of DSM. Figure 8 shows the results of DFEB where two red boxes are added to highlight the differences. It is observed from the right box that the car elevation information is missing in the original DSM information. This makes it difficult to recognize the small vehicles on the road, and subsequently challenging to semantically segment the vehicles. In sharp contrast, the vehicle information is correctly captured by the height estimator as shown in Figure 8(c). Furthermore, the left box represents the height information of the building. It is observed that the DSM information is very irregular at the boundary of the building. After exploiting the optical information, the EDSM at the same location became more regular. In conclusion, the height information is substantially enhanced after the Laplacian pyramid and weighted fusion being applied, which is evidenced from the fact that the BDSM in Figure 8(d) has more details than DSM and is smoother than EDSM.
Figure 9 presents the three-dimensional visualization, showcasing the negative influence that terrain has made on DSM information as well as the improvement on the DSM quality provided by the proposed AMBNet. Note that the DSM values of ground objects on the earth surface are heavily affected by the terrain. As a result, if a single image for semantic segmentation contains a large ground area range, it is possible for ground objects of lower relative heights to possess higher DSM values. For instance, buildings and roads may have similar DSM values if the road is on a hillside, which can pose a major challenge to multimodal tasks especially when the ground objects exhibit similar colors. Furthermore, some trees may have low height information, which may result in mis-classification as low vegetation. In summary, inaccurate DSM information may incur poor segmentation performance. In this work, the proposed DFEB improves the DSM accuracy by exploiting the ground optical information as compensation. This is evidenced by comparing Figure 9(b) and Figure 9(d) in which the optimization effect of DFEB can be observed. More specifically, in the red box of Figure 9(b), a higher DSM value can be observed, which may be due to the terrain or one of the taller trees. However, in the ground truth, we can observe that most of the ground objects in the red box area are trees. As a result, inaccurate DSM information may mislead segmentation models. To cope with this problem, the proposed DFEB provides improved BDSM as indicated in the areas highlighted by the red box in Figure 9(d), which helps improve the semantic segmentation performance.
5.3 Semantic Segmentation Result Analysis
5.3.1 Single-modal models versus AMBNet
We benchmark the proposed AMBNet against four state-of-the-art models, namely DABNet, ABCNet, MAResUnet and UNetformer on Potsdam and Vaihingen datasets. The experimental results on the Vaihingen dataset are summarized in Table 1 and visualized in Figure 11. It is evidenced from Table 1 and Figure 11 that the proposed AMBNet achieved the best segmentation performance for most categories, which confirmed the effectiveness of the proposed height estimation and multimodal fusion method. In particular, AMBNet provided the most significant improvement on the car class with an increase of 4.4% as compared to the existing method DABNet. This can be explained by the fact that the increased details of BDSM can more effectively distinguish car from road. Furthermore, the classification accuracy for road and building has been improved by 2.88% and 0.13%, respectively as these two categories have the most uniform and significant elevation information. For instance, the DSM values of buildings are generally higher while those of roads are generally lower. As a result, the enhanced DSM, namely EDSM can improve these two categories. In terms of the overall performance, AMBNet excelled in terms of three key overall performance metrics, namely OA, MIoU, and mF1. Remarkably, the AMBNet achieved OA of 92.06%, MIoU of 83.33%, and mF1 of 90.67%, which stands for an increase of 1.02%, 2.46% and 1.55% respectively, as compared to the corresponding performance of other comparative methods. These results confirmed that the proposed AMBNet achieved better generalization performance by enhancing the DSM with more details.
Furthermore, the experimental results on the Potsdam dataset are summarized in Table 2 and visualized in Figure 12. In particular, Figure 12 presents two visualization examples of the segmentation results generated by all five methods with prediction differences being highlighted with rectangle boxes. The first example shown in the first row indicate is very challenging as the road and the building have similar optical characteristics. Since DABNet and UNetformer do not effectively use the DSM information, the road was mis-classified as a building. Furthermore, the segmentation performance on road by ABCNet and MAResUnt was not satisfactory. In contrast, the proposed AMBNet achieved much better segmentation results with smoother borders and fewer impurities compared with other methods.
The second example shown in the second row in Figure 11 is another challenging semantic segmentation case. Some of the leaves of the trees in this area have fallen, which caused severe interference. Meanwhile, the buildings exhibit almost identical colors as the road while being partially covered by trees. It can be seen from Figure 11 that only the proposed AMBNet produced good semantic segmentation of the buildings under trees. In contrast, other methods could not accurately distinguish tree and building. Furthermore, another box shows a path through the grass. In this area, despite that the colors of road or low vegetation are clearly distinguishable, their DSM values are similar. It is observed that with the help of BDSM, AMBNet achieved the best segmentation of the narrow path in the middle of grass. Table 2 also confirms AMBNet’s excellent semantic segmentation performance in categories such as car, low veg, tree, building and road. In summary, the proposed AMBNet demonstrated improved performance by exploiting enhanced DSM as compared to other existing semantic segmentation methods.
5.3.2 Existing multi-modal models versus AMBNet
To demonstrate the efficacy of our multi-modal AMBNet, we conducted a comparative analysis with other multi-modal semantic segmentation models, namely CMGFNet, ESANet, FUSENet, and CMFNet. Figure 13 presents the test results on the Vaihingen dataset. Notably, distinguishing between rooftop parking lots and roads poses a significant challenge due to their similar colors and structural features, as illustrated in the purple box where cars were parked on the roof. This segmentation and recognition difficulty is particularly pronounced in most models, with only AMBNet exhibiting commendable results, outperforming others that struggle with low recognition rates. Further examination of the results reveals that AMBNet excelled in recognizing vehicles, trees, and shrubs. As detailed in Table 3, AMBNet attained exceptional accuracy across all categories, boasting an Overall Accuracy (OA) of 92.06% and a mean Intersection over Union (MIoU) of 83.33%. A higher MIoU indicates clearer object outline edges.
5.3.3 Abalation experiments of AMBNet
This section reports the ablation experiments conducted on the AMBNet model, featuring three essential components, namely DFEB, FEB, and PMSS. The objective of these experiments is to scrutinize the individual contributions of each structure to the model’s performance, providing insights into the functioning of the model.
Figure 14 depicts the results of the ablation experiment, revealing the distinct significance of each module within AMBNet. In Figure 14 (b), the impact of removing DFEB is evident, leading to confusion in the recognition of vehicles within the purple box and irregular outlines for buildings in the black box. In Figure 14 (c), the consequences of removing FMB are highlighted. Without FMB’s feature selection for fusion, the absence of the channel attention mechanism may result in the model confusing roofs with roads, manifesting as large areas of white space (representing roads) in the purple box. Table 4 presents the data from the ablation experiment. It is notable that upon deleting DFEB, FMB, and PMSS, the accuracy was reduced to 91.45%, 90.64%, and 90.84%, respectively. While deleting a specific module may enhance accuracy in a particular category, this improvement comes at the cost of reduced accuracy in other categories, indicating an imbalance in the model’s recognition capabilities.
The preceding ablation experiments comprehensively illustrated the significance and individual contributions of each module. Notably, FMB and DFEB emerged as particularly crucial components, with FMB playing a pivotal role in the effective fusion of channel multi-dimensional features, and DFEB contributing significantly to the optimization of DSM information.
6 Conclusion
In this work, an Adaptive Multi-feature Balanced Network named AMBNet has been proposed to perform depth estimation and multi-feature fusion for remote sensing images. Specifically, the proposed AMBNet utilizes a Depth Feature Extraction Module (DFEB) to accurately estimate ground object heights, leading to the creation of a more precise Digital Surface Model (DSM) referred to as BDSM. This enhanced DSM mitigates the adverse effects of terrains on raw DSM data, providing improved and reliable multimodal auxiliary information. Additionally, AMBNet incorporates a Parallel Multi-Stage Network (PMSS) to harness the combined power of BDSM and optical images for semantic segmentation. The results demonstrate that AMBNet excels in handling building shadows and detecting smaller ground targets hidden by canopies, showcasing commendable performance in semantic segmentation. These findings are substantiated through comprehensive experiments conducted on the Vaihingen and Potsdam datasets.
There are several extensions of this study that can be further explored. First, it is of great practical interest to further improve the height estimation performance to generate more accurate EDSM. Furthermore, it is interesting to consider how to efficiently utilize the EDSM information as the monocular depth estimation module produces high-dimensional output of multiple channels. Finally, end-to-end designs of semantic segmentation and other downstream tasks will be explored in future research.















