Skip to Content
  • Article
  • Open Access

20 August 2026

29 Pages

Structure-Prior-Guided Multi-Stage Cross-Modal Collaborative Network for RGB-D Semantic Segmentation

,
,
and
School of Computer Science and Software Engineering, Southwest Petroleum University, Chengdu 610500, China
*
Author to whom correspondence should be addressed.
This article belongs to the Section AI in Imaging

Abstract

Red–green–blue and depth (RGB-D) semantic segmentation combines appearance cues from RGB images with geometric information from depth maps, but sensor noise, missing measurements, and boundary-inconsistent depth responses can introduce conflicting evidence during cross-modal fusion. We propose the Structure-Prior-Guided Network (SPGNet), a dual-branch, multi-stage framework that follows a correction-before-fusion strategy. At each feature scale, SPGNet estimates a learned structure prior from cross-modal agreement and discrepancy. The Cross-Modal Correction Module (CCM) uses this prior to regulate bidirectional information transfer, suppressing unreliable responses while retaining complementary cues. The Dual-branch Enhancement Fusion Module (DEF) then enhances the corrected RGB and depth features and integrates them through shared-representation-guided interaction, after which a lightweight multi-scale decoder produces the segmentation output. Under a unified training and evaluation protocol, SPGNet achieved three-run mean Intersection over Union (mIoU) scores of 50.845% on NYU Depth V2 and 48.457% on SUN RGB-D. Compared with the best reproduced baseline on each dataset, SPGNet improved mean mIoU by 2.111 and 0.899 percentage points, respectively. These results suggest that separating reliability-oriented correction from multimodal fusion can limit the propagation of unreliable cross-modal responses and improve indoor RGB-D semantic segmentation performance.

1. Introduction

Semantic segmentation, which assigns a semantic label to each image pixel, is a fundamental problem in indoor scene understanding and supports applications such as robotic navigation, scene reconstruction, and human–environment interaction [1,2,3]. Red–green–blue (RGB) images provide rich color, texture, and semantic context, but RGB-only parsing is often degraded by weak texture, illumination variation, occlusion, and cluttered indoor layouts. With the increasing availability of depth sensors, RGB and depth (RGB-D) semantic segmentation has become an effective way to improve indoor parsing by introducing geometric information complementary to RGB appearance [4,5,6]. Depth maps describe object contours, spatial layout, and structural discontinuities, which are especially useful for distinguishing regions that are visually similar but geometrically different [7,8].
However, introducing depth information does not automatically lead to better segmentation. The main difficulty lies in both modality heterogeneity and modality reliability. RGB and depth describe the same scene from different physical perspectives: RGB features emphasize photometric appearance, whereas depth features encode geometric structure and relative spatial layout. At the same time, depth maps captured in indoor scenes may contain sensor noise, missing values, local misalignment, and boundary-inconsistent responses [4,9,10]. If these unreliable responses are directly exchanged or fused with RGB features, local conflicts between appearance and geometry can be propagated into the final representation, causing fragmented object regions, blurred boundaries, and semantic leakage across adjacent categories. Figure 1 illustrates this issue through three representative failure modes of Direct Fusion: region fragmentation, boundary leakage, and cross-modal inconsistency. Here, Direct Fusion denotes a simple RGB-D feature integration baseline without the proposed pre-fusion correction. In the region-fragmentation example (left column), Direct Fusion fails to preserve the individual chair regions and the gaps between adjacent objects, whereas SPGNet recovers a more coherent arrangement. In the boundary-leakage example (middle column), Direct Fusion confuses much of the table region and introduces errors near the chair–table boundary; SPGNet produces a cleaner separation between these categories. In the cross-modal-inconsistency example (right column), Direct Fusion yields substantial class confusion between the table and chair regions, whereas SPGNet restores their dominant semantic regions more consistently with the annotation. Although some annotation boundaries are relatively coarse, these region-level observations are consistent with the interpretation that pre-fusion correction can limit the propagation of conflicting cross-modal responses; the controlled ablation in Section 4.4 provides the corresponding quantitative evidence.
Figure 1. Qualitative motivation examples from NYU Depth V2. The three columns illustrate region fragmentation, boundary leakage, and cross-modal inconsistency, respectively. Rows (1)–(4) show the red–green–blue (RGB) images, ground-truth annotations, Direct Fusion predictions, and Structure-Prior-Guided Network (SPGNet) predictions, respectively.
Existing RGB-D semantic segmentation methods have explored this problem from several directions. Early and convolutional neural network (CNN)-based methods commonly adopt dual-branch encoders, residual fusion, gated propagation, context aggregation, or geometry-aware convolution to combine RGB appearance and depth geometry [5,6,9,11]. Recent Transformer-based and hybrid methods further introduce token interaction, cross-modal feature rectification, asymmetric modality modeling, RGB-D representation pretraining, primary-modality guidance, and depth-guided self-attention [10,12,13,14,15,16]. These studies show that RGB and depth should not be treated as two identical inputs, and that their complementary information must be selected and aligned carefully. Although these methods have significantly improved RGB-D feature interaction, reliability modeling is usually coupled with the fusion or interaction process itself. In such designs, unreliable source-modality responses may be injected into the target branch before their local compatibility is explicitly estimated. This motivates a pre-fusion mechanism that first models feature-level RGB–depth agreement and discrepancy, and then uses the resulting structure prior to regulate the direction and strength of cross-modal transfer.
Motivated by this, this paper proposes the Structure-Prior-Guided Network (SPGNet), a dual-branch multi-stage framework for RGB-D semantic segmentation. The key idea of SPGNet is to decouple reliability-oriented cross-modal correction from representation-oriented multimodal fusion. At each feature scale, SPGNet first estimates a learned feature-level structure prior from RGB–depth consistency and discrepancy cues. Instead of relying on externally supervised edge maps or ground-truth structural labels, this prior is learned in the encoded feature space and reflects local cross-modal compatibility. Guided by this prior, the Cross-Modal Correction Module (CCM) performs gated bidirectional correction to regulate cross-modal transfer before fusion. The Dual-branch Enhancement Fusion Module (DEF) then enhances the corrected modality-specific features and integrates them through shared-representation-guided interaction. Finally, a lightweight multi-scale decoder aggregates the fused features for dense prediction.
This separation enables the network to reduce the influence of locally inconsistent modality responses before feature integration, while preserving both appearance semantics and geometric structure for indoor scene parsing. Experiments on NYU Depth V2 and SUN RGB-D show that SPGNet consistently improves over representative reproduced RGB-D segmentation baselines under the unified evaluation protocol, and ablation studies further support the complementary roles of CCM and DEF.
The main contributions of this paper are summarized as follows:
  • We propose SPGNet, a multi-stage RGB-D semantic segmentation framework that decouples reliability-oriented cross-modal correction from representation-oriented multimodal fusion.
  • We design CCM to estimate a learned feature-level structure prior from RGB–depth consistency and discrepancy cues, and to use this prior for gated bidirectional cross-modal correction before fusion.
  • We design DEF to enhance the corrected RGB and depth features and to integrate them through shared-representation-guided interaction, preserving modality-specific cues while forming a unified multimodal representation.

3. Materials and Methods

3.1. Overall Framework

Given an RGB image I r ∈ R 3 × H × W and a depth map I d ∈ R 1 × H × W , the depth channel is replicated to obtain I ˜ d ∈ R 3 × H × W . SPGNet predicts
Y = Φ ( I r , I ˜ d ) ,
where Y ∈ R K × H × W contains the final logits and K includes the dataset-specific ignored label index. Supervision uses Y g t , with unlabeled pixels excluded as defined in Section 3.5.
As shown in Figure 2, SPGNet contains two encoders, four stage-wise Cross-Modal Correction Modules (CCM), four Dual-branch Enhancement Fusion Modules (DEF), and a multi-scale decoder. At each scale, CCM uses a learned structure prior to regulate cross-modal transfer, after which DEF enhances and fuses the corrected features.
Figure 2. Overall architecture of the Structure-Prior-Guided Network (SPGNet). Red–green–blue (RGB) and depth features are extracted by heterogeneous encoders, corrected by the Cross-Modal Correction Module (CCM) at each stage, enhanced and fused by the Dual-branch Enhancement Fusion Module (DEF), and aggregated by the multi-scale decoder.
The RGB and depth encoders use Pyramid Vision Transformer v2 B2 (PVT-v2-B2) and 34-layer Residual Network (ResNet34), respectively, with ImageNet-pretrained weights. This pairing combines long-range RGB context with the local geometric features of ResNet34, whose four stages are aligned to PVT-v2-B2 through lightweight projections. For inputs divisible by 32, stage i ∈ { 1 , 2 , 3 , 4 } has resolution ( h i , w i ) = ( H / 2 i + 1 , W / 2 i + 1 ) and channels ( 64 , 128 , 320 , 512 ) , giving
F r i , F d i ∈ R C i × h i × w i , ( C 1 , C 2 , C 3 , C 4 ) = ( 64 , 128 , 320 , 512 ) .
At each stage, the two features are first corrected and then fused:
( F ^ r i , F ^ d i ) = CCM i ( F r i , F d i ) ,
F f i = DEF i ( F ^ r i , F ^ d i ) ,
where F ^ r i and F ^ d i are the CCM-processed RGB and depth features, and F f i ∈ R C i × h i × w i is the fused feature at stage i. The four fused features are aligned to the first-stage resolution and aggregated by the decoder to obtain Y ˜ ∈ R K × h 1 × w 1 :
Y ˜ = Decoder ( F f 1 , F f 2 , F f 3 , F f 4 ) .
The decoder output is then upsampled to the input resolution:
Y = Up ori ( Y ˜ ) .
Section 3.2 and Section 3.3 describe CCM and DEF, and Section 3.4 specifies the decoder implementation.

3.2. Cross-Modal Correction Module

CCM estimates a structure prior before fusion to regulate bidirectional transfer and limit the propagation of unreliable depth responses, boundary inconsistencies, and RGB–depth misalignment.
Given the RGB feature F r and depth feature F d at the same scale, CCM aims to generate the updated bimodal features F ^ r and F ^ d :
( F ^ r , F ^ d ) = CCM ( F r , F d ) .
As summarized in Figure 3, CCM constructs a prior from cross-modal consistency and discrepancy, generates branch-specific gates, and then performs bidirectional cross-attention.
Figure 3. Structure of the Cross-Modal Correction Module (CCM): (a) overall architecture; (b) fixed-edge and aligned-feature construction of the structure prior; and (c) branch-specific gate generation from the target and prior-modulated source features.

3.2.1. Structure Prior Generation

Figure 3b maps F r , F d ∈ R C × h × w to the structure prior P by modeling both cross-modal agreement and discrepancy in the encoded feature space.
The implementation first extracts a fixed structural response from each modality. Let S ( F ) denote channel averaging followed by convolution with fixed horizontal and vertical Sobel kernels and gradient-magnitude computation. The modality-specific edge responses are
E r = S ( F r ) , E d = S ( F d ) ,
where E r , E d ∈ R 1 × h × w . The edge-consistency response is computed explicitly as
C s = 1 − | E r − E d | E r + E d + ϵ ,
with ϵ = 10 − 6 . To measure feature discrepancy, the two modalities are independently aligned by bias-free 1 × 1 convolutions, A r = ϕ r ( F r ) and A d = ϕ d ( F d ) , and their channel-averaged absolute difference is computed as
D s = 1 C ∑ c = 1 C | A r , c − A d , c | .
Thus, C s , D s ∈ R 1 × h × w . The two edge responses, consistency response, and discrepancy response are concatenated into a four-channel descriptor and mapped to the channel-wise structure prior:
P = σ F p ( [ E r , E d , C s , D s ] ) ,
where [ · ] denotes channel concatenation, F p ( · ) comprises two 3 × 3 Conv–BN–ReLU layers followed by a 1 × 1 Conv–BN projection, and P ∈ [ 0 , 1 ] C × h × w . The consistency and discrepancy quantities are explicit responses; the mapping from their four-channel descriptor to P is learned.

3.2.2. Bidirectional Cross-Modal Correction

The prior P then guides symmetric depth-to-RGB and RGB-to-depth correction before fusion.
First, the two input feature streams are normalized to obtain
F ¯ r = BN ( F r ) , F ¯ d = BN ( F d ) .
Taking the update of the RGB branch as an example, this paper treats the RGB feature as the target branch and the depth feature as the source branch, and obtains the following representations through 1 × 1 convolutional mappings:
T r = ψ r ( F ¯ r ) , S d = ψ d ( F ¯ d ) .
Here, T r , S d ∈ R C × h × w are the target RGB and source depth representations.
Then, the structure-prior map is used to perform structure-aware modulation on the source-branch feature:
S d ′ = S d ⊙ ( 1 + α r P ) ,
where ⊙ denotes element-wise multiplication, P ∈ [ 0 , 1 ] C × h × w is produced by the Sigmoid output described in the previous subsection, and α r is a learnable scalar initialized to 0. Therefore, this modulation term is an identity mapping at the beginning of training.
Because α r is unbounded, explicit attenuation is performed by the following Sigmoid gate rather than by this modulation term.
Next, the target feature, the prior-modulated source feature, and their difference are concatenated and fed into a gating unit to generate a cross-modal gating map, following the gate-generation pattern in Figure 3c:
G r = σ F g r ( [ T r , S d ′ , T r − S d ′ ] ) ,
where σ ( · ) denotes the Sigmoid function, and G r ∈ [ 0 , 1 ] C × h × w .
The Sigmoid gate attenuates or preserves the modulated source response; the same construction is used in the reverse direction.
After gating, the filtered cross-modal feature is obtained as
S ˜ d = S d ′ ⊙ G r .
Thus, CCM filters the source response before cross-modal transfer.
Then, lightweight cross-attention is performed by using the RGB feature as the query and the gated depth feature as the key and value. To control the computational cost, average-pooling downsampling is applied only to the key-value branch in the implementation:
S ˜ d , ↓ = AvgPool r i ( S ˜ d ) ,
where r i denotes the key-value downsampling rate of CCM at the current stage, and S ˜ d , ↓ ∈ R C × h i ↓ × w i ↓ denotes the pooled key-value feature with spatial size determined by r i . Then,
Δ F r = CA r Q = T r , K = S ˜ d , ↓ , V = S ˜ d , ↓ .
Specifically, CA r ( · ) generates the query Q r from T r and generates the key and value K d , V d from S ˜ d , ↓ :
Q r = W q r T r , K d = W k r S ˜ d , ↓ , V d = W v r S ˜ d , ↓ ,
The 1 × 1 projections produce Q r ∈ R N i d i × h × w and K d , V d ∈ R N i d i × h i ↓ × w i ↓ . After attention over flattened spatial dimensions, an output projection restores Δ F r ∈ R C × h × w .
The depth update symmetrically treats depth as the target and RGB as the source:
T d = ψ d ′ ( F ¯ d ) , S r = ψ r ′ ( F ¯ r ) ,
S r ′ = S r ⊙ ( 1 + α d P ) ,
G d = σ F g d ( [ T d , S r ′ , T d − S r ′ ] ) ,
S ˜ r = S r ′ ⊙ G d ,
S ˜ r , ↓ = AvgPool r i ( S ˜ r ) ,
Δ F d = CA d Q = T d , K = S ˜ r , ↓ , V = S ˜ r , ↓ .
Here, T d , S r ∈ R C × h × w , G d ∈ [ 0 , 1 ] C × h × w , and α d is initialized to 0. The projection dimensions match the RGB update: Q d ∈ R N i d i × h × w , K r , V r ∈ R N i d i × h i ↓ × w i ↓ , and Δ F d ∈ R C × h × w .
Finally, the bimodal features are updated in a residual manner:
F ^ r = F r + γ r Δ F r , F ^ d = F d + γ d Δ F d ,
where γ r and γ d are learnable residual coefficients, both initialized to 0.
The zero initialization of α r and α d makes the prior modulation initially an identity operation, while the zero initialization of γ r and γ d guarantees that the complete CCM initially preserves its two input features. The residual scales then learn how strongly the cross-modal corrections should be introduced before DEF.

3.3. Dual-Branch Enhancement Fusion Module

Given F ^ r , F ^ d ∈ R C × h × w , DEF enhances each modality, constructs a shared reference, applies shared-guided cross-attention, and outputs F f ∈ R C × h × w for the decoder (Figure 4).
Figure 4. Structure of the Dual-branch Enhancement Fusion Module (DEF): (a) overall architecture and (b) Feature Enhancement Unit. The operator legend applies to both panels; ⊕ denotes residual addition and C denotes channel concatenation.

3.3.1. Modality Feature Enhancement

As shown in Figure 4b, DEF first enhances the RGB and depth features independently to preserve their modality-specific characteristics. For an arbitrary input feature F ∈ R C × h × w , the feature enhancement unit can be expressed as
E ( F ) = F enh [ D 3 × 3 ( P ( F ) ) , D 5 × 5 ( P ( F ) ) ] + F ,
Here, P is a 1 × 1 projection, D 3 × 3 and D 5 × 5 are C-channel depthwise separable convolutions, and F enh : R 2 C × h × w → R C × h × w fuses their concatenated outputs before residual addition.
The two kernels capture complementary local contexts while the residual path preserves the input.
Therefore, the corrected RGB and depth features are enhanced as
F r enh = E ( F ^ r ) , F d enh = E ( F ^ d ) .

3.3.2. Shared-Guided Cross-Modal Interaction

A shared reference is constructed directly from the corrected bimodal features:
F m = F m ( [ F ^ r , F ^ d ] ) ,
where F m ( · ) : R 2 C × h × w → R C × h × w is implemented by a convolutional fusion block, and therefore F m ∈ R C × h × w .
Thus, F m retains CCM-corrected shared cues, whereas F r enh and F d enh retain modality-specific responses.
Subsequently, convolutional projections are applied to the enhanced RGB feature, the enhanced depth feature, and the shared fused feature, respectively, and all three outputs are kept in R C × h × w :
R = R r ( F r enh ) , D = R d ( F d enh ) , M = R m ( F m ) .
Next, two-branch cross-attention interaction is performed by using the RGB-branch feature and the depth-branch feature as queries, respectively, and the shared fused feature as the key and value:
R ca = SGCA r def Q = R , K = M ↓ , V = M ↓ + R ,
D ca = SGCA d def Q = D , K = M ↓ , V = M ↓ + D ,
Only the shared key/value feature is average-pooled: M ↓ ∈ R C × h i ↓ × w i ↓ is determined by r i ′ , while the R and D queries retain resolution h × w . The Q / K / V projections use 16 N i channels, scaled dot-product attention aggregates V, and an output projection restores C channels, giving R ca , D ca ∈ R C × h × w .
Using M as the common key/value reference aligns both branches without direct RGB–depth attention.
To provide an additional bypass from the corrected DEF inputs, this paper adds the interaction results to the input features of DEF in a residual manner:
R ′ = R ca + F ^ r , D ′ = D ca + F ^ d .
The query-side residuals in the attention equations and the additional corrected-feature bypasses preserve modality information.
Finally, the output feature at the current scale is obtained through the final fusion head, corresponding to the “FinalFusion Head” in Figure 4a:
F f = F final ( [ R ′ , D ′ ] ) ,
where the convolutional head F final : R 2 C × h × w → R C × h × w produces the decoder input.

3.4. Decoder and Multi-Scale Fusion

The decoder integrates { F f 1 , F f 2 , F f 3 , F f 4 } , where F f i ∈ R C i × h i × w i and ( C 1 , C 2 , C 3 , C 4 ) = ( 64 , 128 , 320 , 512 ) , using one prediction branch without auxiliary supervision.
First, fused features from the second to fourth stages are upsampled to the first-stage resolution and then concatenated with F f 1 along the channel dimension:
F m s = Concat F f 1 , Up 2 → 1 ( F f 2 ) , Up 3 → 1 ( F f 3 ) , Up 4 → 1 ( F f 4 ) .
Here, Up i → 1 ( · ) denotes bilinear interpolation that upsamples the feature of the i-th stage to h 1 × w 1 . Under the stage-resolution relation h i = h 1 / 2 i − 1 and w i = w 1 / 2 i − 1 , the corresponding scale factors are 2 , 4 , 8 for the second to fourth stages, respectively. Therefore, F m s ∈ R ( 64 + 128 + 320 + 512 ) × h 1 × w 1 = R 1024 × h 1 × w 1 .
Then, a convolutional fusion layer and a classification layer are used to output logits at the first-stage resolution:
Y ˜ = F cls F fuse ( F m s ) .
F fuse compresses the concatenated features and F cls produces Y ˜ ∈ R K × h 1 × w 1 .
Finally, the logits are resized to the original input size H × W to obtain the final output:
Y = Up ori ( Y ˜ ) ,
where Y ∈ R K × H × W contains the final logits.

3.5. Loss Function

Let Y be the output logits and Y g t ∈ { 0 , 1 , … , K − 1 } H × W the ground truth, with label 0 ignored. The valid pixels are
Ω v = { p ∣ Y p g t ≠ 0 } .
The overall loss function is defined as
L = L s e g ,
where L s e g denotes the pixel-level cross-entropy loss over valid pixels:
L s e g = − 1 | Ω v | ∑ p ∈ Ω v log exp ( Y Y p g t , p ) ∑ k = 0 K − 1 exp ( Y k , p ) .
K = 41 for NYU Depth V2 and K = 38 for SUN RGB-D, corresponding to 40 and 37 valid classes plus the ignored label. The loss is applied only to the final output.

3.6. Implementation Specification of CCM and DEF

Table 1 summarizes the four-scale tensor dimensions and hyperparameters for a 480 × 640 input. Unless stated otherwise, CCM and DEF convolutional blocks use Conv–BN–ReLU, while alignment, Q / K / V , Sigmoid, and output projections follow the listed rows. The coefficients α r , α d , γ r , γ d are initialized to 0, making CCM an identity mapping at initialization. The prior hidden dimension is C p = max ( ⌊ C / 2 ⌋ , 16 ) ; the CCM head dimension is d i = max ( ⌊ C i / ( 2 N i ) ⌋ , 16 ) , and the DEF head dimension is 16. Attention projections use N i d i channels in CCM and 16 N i in DEF before being mapped back to C channels.
Table 1. Notation, dimensions, and hyper-parameter specifications of the Cross-Modal Correction Module (CCM) and Dual-branch Enhancement Fusion Module (DEF) used in this work. Subscripts r and d denote the red–green–blue (RGB) and depth branches, respectively. C ∈ { 64 , 128 , 320 , 512 } denotes the output channels of stages 1–4; h × w denotes the corresponding spatial resolution; N i denotes the number of attention heads, d i denotes the CCM channels per head, d = 16 denotes the fixed DEF channels per head, and r i / r i ′ denote the key/value downsampling ratios in cross-attention.

3.7. Datasets and Experimental Settings

Experiments are conducted on two public RGB-D indoor semantic segmentation benchmarks, namely NYU Depth V2 [2] and SUN RGB-D [3]. Both datasets provide aligned RGB images, depth maps, and pixel-level semantic annotations, and cover diverse indoor scenes with different sensor characteristics and depth response patterns.
NYU Depth V2 contains 1449 indoor scene images. Following the standard split, 795 images are used for training and 654 images are used for testing. The dataset is evaluated under the NYU40 setting with 40 valid semantic classes. SUN RGB-D contains 10,335 RGB-D samples collected from different sources, including NYU Depth V2, Berkeley B3DO, and SUN3D. Following the official split, 5285 images are used for training and 5050 images are used for testing. In this study, SUN RGB-D is evaluated using 37 valid semantic classes. For both datasets, label 0 denotes unlabeled or void pixels and is excluded from both loss computation and metric computation.
All experiments were implemented in Python 3.9 using PyTorch 2.8.0 and NumPy 1.26.4 and conducted on a workstation equipped with an NVIDIA GeForce RTX 5070 Ti GPU. The optimization settings for SPGNet were based on published configurations in recent RGB-D semantic segmentation studies and adapted to our unified experimental protocol. We used AdamW with an initial learning rate of 6 × 10 − 5 and a weight decay of 0.01 , matching the NYU Depth V2 configuration reported for PrimKD [15]. The use of polynomial learning-rate decay with a power of 0.9 and a warm-up phase was informed by AsymFormer [13]; in our protocol, the warm-up period was set to 5 epochs. To maintain a consistent training length across the reproduced experiments, all models were trained for 100 epochs. The batch sizes were set to 2 for NYU Depth V2 and 4 for SUN RGB-D to fit the available GPU memory. Within each dataset, the same optimization settings were used for SPGNet and the four controlled SUN RGB-D ablation configurations.
The stage-wise hyperparameters of CCM and DEF were specified according to the encoder feature dimensions and the computational cost of spatial attention. For encoder channels of { 64 , 128 , 320 , 512 } , the numbers of attention heads were set to { 1 , 2 , 4 , 4 } . The key/value downsampling ratios were { 8 , 4 , 4 , 2 } for CCM and { 8 , 4 , 2 , 1 } for DEF. Thus, stronger key/value reduction was applied at high-resolution stages to control the attention cost, whereas lower-resolution stages retained more key/value locations. The DEF per-head dimension was fixed at 16 as a compact implementation choice. The same CCM and DEF configuration was used for both datasets.
The RGB image was used as a three-channel appearance input, while the single-channel depth map was expanded to three channels to match the input structure of the depth encoder. During training, random scale resizing, random horizontal flipping, and random cropping were applied. The random scale was sampled from { 0.5 , 0.75 , 1.0 , 1.25 , 1.5 , 1.75 } , after which each sample was cropped or padded to 480 × 640 . During testing, the samples were processed at the same spatial resolution. RGB images were normalized using the ImageNet mean and standard deviation. For the depth modality, the depth map was first scaled by 1 / 255 and then normalized using mean values of ( 0.48 , 0.48 , 0.48 ) and standard deviation values of ( 0.28 , 0.28 , 0.28 ) . The network was trained using the standard cross-entropy loss with label 0 ignored, without auxiliary supervision.
The archived metric exports used for the three-run summaries did not retain a common global seed across Python 3.9, NumPy 1.26.4, and PyTorch 2.8.0; consequently, no seed is assigned retrospectively to these reported runs. They are treated as three independent stochastic training runs, and the sample standard deviations summarize their observed run-to-run variability. In SPGNet, the PVT-v2-B2 RGB encoder and ResNet34 depth encoder were initialized through the pretrained weights provided by their  timm  constructors. Newly introduced convolutional, linear, and normalization layers retained the initialization applied by their PyTorch or  timm  constructors, except for the CCM prior-modulation coefficients α r , α d and residual coefficients γ r , γ d , which were explicitly initialized to zero as described in Section 3.2.
The reproduced comparison methods include ACNet [27], SA-Gate [9], PrimKD [15], SGACNet [37], ShapeConv [11], and AsymFormer [13]. For each dataset, the proposed method and all reproduced baselines were evaluated using the same input resolution, dataset-specific label-processing procedure, training length, and evaluation code.

3.8. Evaluation Metrics

This paper evaluates RGB-D semantic segmentation performance using six metrics: mean Intersection over Union (mIoU), mean F1-score, mean precision, mean recall, Overall Accuracy (OA), and Cohen’s Kappa. mIoU measures the overlap between predicted and ground-truth regions at the class level. Mean precision reflects the average correctness of class-specific positive predictions, mean recall reflects the average coverage of ground-truth pixels across semantic classes, and the mean F1-score balances the two quantities at the class level. OA measures the proportion of correctly classified evaluated pixels, while Kappa further accounts for chance agreement between predictions and ground-truth annotations.
During evaluation, the segmentation head produces K = 41 logits for NYU Depth V2 (labels 0–40) and K = 38 logits for SUN RGB-D (labels 0–37). Pixels with label 0, representing void or padded regions, are removed before confusion-matrix accumulation and are ignored by the cross-entropy loss. For each valid class c ∈ { 1 , … , K − 1 } , IoU, precision, recall, and F1 are computed from the confusion matrix; mIoU, mean precision, mean recall, and mean F1 are uniformly macro-averaged over labels 1–40 for NYU Depth V2 and labels 1–37 for SUN RGB-D. Masks are resized using nearest-neighbor interpolation and undergo the same spatial transformations as their paired RGB and depth inputs. OA and Kappa are computed from the same confusion matrix over the evaluated non-ignored pixels.
NYU Depth V2 and SUN RGB-D contain 40 and 37 valid categories, respectively, under different processed-label protocols. This study uses the complete-class macro averages above as its primary class-balanced comparison and does not claim that SPGNet improves every individual category or that its gains are concentrated in a predefined object-size group. The archived results contain the aggregate metrics reported in Table 2 and Table 3, but not the per-run class-wise confusion matrices or full-resolution predictions required to reconstruct a consistent complete per-class comparison. Moreover, per-class IoU alone does not identify small- versus large-object behavior because the same semantic category can occupy substantially different image areas. A complete class-wise comparison and instance-scale stratification would therefore require renewed inference under a predefined common protocol and are left as limitations of the present evaluation.
Table 2. Quantitative comparison on NYU Depth V2 under the NYU40 processed-label protocol. All metrics are reported as the mean ± sample standard deviation over three independent training runs. mIoU, F1, and OA denote mean Intersection over Union, mean F1-score, and Overall Accuracy, respectively. The best results are in bold.
Table 3. Quantitative comparison on SUN RGB-D under the 37-class processed-label protocol. All metrics are reported as the mean ± sample standard deviation over three independent training runs. mIoU, F1, and OA denote mean Intersection over Union, mean F1-score, and Overall Accuracy, respectively. The best results are in bold.
Let M ∈ R K × K denote the confusion matrix accumulated over the complete test set, where rows correspond to predicted labels and columns correspond to ground-truth labels. For each valid class c, the true-positive, false-positive, and false-negative counts are T P c = M c c , F P c = ∑ j ≠ c M c j , and F N c = ∑ i ≠ c M i c , respectively. Class-wise precision, recall, and F1 are then computed as
P c = T P c T P c + F P c + ϵ , R c = T P c T P c + F N c + ϵ , F 1 c = 2 P c R c P c + R c + ϵ ,
where ϵ = 10 − 10 prevents division by zero. Consistent with the implementation, mean precision, mean recall, and mean F1 are macro-averaged over the K − 1 valid semantic classes:
P ¯ = 1 K − 1 ∑ c = 1 K − 1 P c , R ¯ = 1 K − 1 ∑ c = 1 K − 1 R c , F 1 ¯ = 1 K − 1 ∑ c = 1 K − 1 F 1 c .

4. Results and Discussion

Figure 5 reports the unsmoothed per-epoch training and validation losses of SPGNet over the complete 100-epoch schedule. On NYU Depth V2, the training loss decreases from 3.656 to 1.764, while the validation loss decreases from 3.540 to 1.851. On SUN RGB-D, the corresponding losses decrease from 3.135 to 1.063 and from 2.686 to 1.300, respectively. For both datasets, the losses decline rapidly during the early epochs and then decrease more gradually, while the validation curves remain stable during the later stage without sustained divergence. These trends indicate stable optimization convergence under the adopted training protocol.
Figure 5. Unsmoothed per-epoch training and validation loss curves of SPGNet over 100 epochs on (a) NYU Depth V2 and (b) SUN RGB-D.

4.1. Results on NYU Depth V2

Table 2 presents the quantitative comparison between the proposed method and representative RGB-D semantic segmentation methods on NYU Depth V2 under the unified training and evaluation protocol. Across three independent runs, SPGNet obtains the highest mIoU, F1, mean precision, mean recall, OA, and Kappa values of 50.845 ± 0.126 %, 64.985 ± 0.139 %, 68.390 ± 1.331 %, 64.117 ± 0.326 %, 76.497 ± 0.126 %, and 73.956 ± 0.073 %, respectively. Compared with AsymFormer [13], the second-best method for mIoU and F1, SPGNet improves mean mIoU and mean F1 by 2.111 and 1.681 percentage points. For OA and Kappa, PrimKD [15] gives the second-best three-run means, and SPGNet exceeds it by 0.969 and 1.263 percentage points, respectively.
The gains are also consistent when compared with other convolutional or gating-based RGB-D fusion methods. Using the three-run means, SPGNet improves mIoU over ShapeConv [11], SA-Gate [9], SGACNet [37], and ACNet [27] by 3.635, 3.566, 5.627, and 6.526 percentage points, respectively. These results suggest that applying structure-prior-guided correction before dual-branch enhancement fusion is beneficial under this evaluation setting, while the magnitude of the improvement remains moderate against the strongest transformer-based baseline.
For the additional component metrics, SPGNet achieves the highest three-run mean precision and mean recall of 68.390% and 64.117%, respectively. Compared with AsymFormer, the strongest baseline on these two metrics, SPGNet improves mean precision by 3.149 percentage points and mean recall by 1.350 percentage points. The simultaneous improvements in precision and recall are consistent with the higher mean F1-score obtained by SPGNet on NYU Depth V2.

4.2. Results on SUN RGB-D

To further evaluate the proposed method on a larger indoor scene benchmark, this paper reports comparative results on SUN RGB-D. Compared with NYU Depth V2, SUN RGB-D contains more samples from heterogeneous sources and therefore introduces greater variation in scene layout, sensor type, and depth quality. Table 3 summarizes the quantitative results under the corresponding SUN RGB-D processed-label protocol and training pipeline.
Across three independent runs, SPGNet achieves the highest mean mIoU and F1 values of 48.457 ± 0.148 % and 62.496 ± 0.076 %, respectively. Compared with AsymFormer [13], the second-best method on these metrics, SPGNet improves mean mIoU and mean F1 by 0.899 and 0.963 percentage points. Compared with ShapeConv [11], the corresponding improvements are 3.072 and 2.991 points. Its mean mIoU improvements over PrimKD [15], SA-Gate [9], SGACNet [37], and ACNet [27] are 2.422, 3.345, 4.188, and 4.511 percentage points, respectively. SPGNet also obtains the highest three-run mean OA and Kappa values of 81.183% and 77.900%, exceeding AsymFormer by 0.544 and 0.628 percentage points, respectively.
For the additional component metrics, SPGNet achieves the highest three-run mean precision and mean recall of 64.362% and 61.257%, respectively. Compared with AsymFormer, SPGNet improves mean precision by 0.223 percentage points and mean recall by 1.092 percentage points. These simultaneous gains are consistent with the higher mean F1-score obtained by SPGNet on SUN RGB-D.
To assess statistical significance, two-sided Welch tests were applied to the three independent runs of SPGNet and AsymFormer. On NYU Depth V2, the mIoU difference of 2.111 percentage points has a 95 % confidence interval of 1.682–2.540 ( p < 0.001 ), and the F1 difference of 1.681 points has a 95 % confidence interval of 1.006–2.356 ( p = 0.005 ). On SUN RGB-D, the corresponding differences are 0.899 points in mIoU ( 95 % confidence interval: 0.347–1.450; p = 0.013 ) and 0.963 points in F1 ( 95 % confidence interval: 0.237–1.690; p = 0.028 ). Thus, the advantage over the strongest reproduced baseline is retained across the three runs on both datasets. Given the small number of runs, these inferential results are interpreted together with the effect sizes and run-wise variability.
The SUN RGB-D results follow the same trend as the NYU Depth V2 results: SPGNet consistently improves over the reproduced baselines, while the margin over the strongest transformer-based baseline remains relatively small. This observation suggests that the proposed correction-and-fusion design is useful across the two evaluated indoor RGB-D benchmarks, but the results should be interpreted together with the following model-scale analysis.
Figure 6 and Figure 7 present qualitative comparisons on NYU Depth V2 and SUN RGB-D, respectively.
Figure 6. Qualitative comparison on NYU Depth V2. From left to right, the columns show the RGB image, depth map, ground truth, ShapeConv, AsymFormer, and SPGNet predictions.
Figure 7. Qualitative comparison on SUN RGB-D. From left to right, the columns show the RGB image, depth map, ground truth, ShapeConv, AsymFormer, and SPGNet predictions.
In the first NYU Depth V2 example, ShapeConv produces fragmented labels on the tall cabinet and around the bed–floor transition, while AsymFormer introduces a small isolated false-positive region above the furniture. SPGNet yields a more contiguous cabinet region and a cleaner bed contour against the wall and floor. In the second and fourth examples, the bed occupies a large portion of the scene and is surrounded by pillows, nearby furniture, and weak wall–floor depth transitions. SPGNet preserves a more complete bed region with less label leakage into the surrounding background. In the dining-room example, the proposed model also maintains a coherent tabletop region and separates much of the adjacent furniture, although thin chair structures and small objects remain difficult for all three methods.
A similar pattern is visible on SUN RGB-D. In the bathroom and bedroom examples, SPGNet better preserves the dominant object extent at the bathtub–wall and bed–background interfaces, where the depth maps contain sharp transitions together with locally noisy or weak responses. In the living-room example, the sofa region remains more spatially coherent, whereas the comparison predictions contain more local boundary irregularities. The final doorway scene illustrates a harder case involving narrow foreground structures and partial occlusion: SPGNet recovers more of the small foreground regions but still exhibits errors near ambiguous boundaries. These observations are consistent with the intended roles of CCM and DEF: CCM can reduce the influence of locally inconsistent cross-modal responses before fusion, while DEF can combine the corrected modalities to refine object extent and boundaries. However, the end-to-end visual comparisons alone do not isolate the causal contribution of either module; that evidence is provided by the controlled module ablation in Section 4.4.
Representative Failure Cases. To complement the qualitative comparisons above, Figure 8 presents two representative failure cases that illustrate recurring error mechanisms under challenging indoor conditions.
Figure 8. Representative failure cases of SPGNet. From left to right, the columns show the RGB image, ground-truth annotation, and SPGNet prediction. Black pixels in the ground truth correspond to ignore index 0 and are excluded from the loss and quantitative evaluation. The top row illustrates a cluttered scene with extensive ignored regions, dense small objects, and occlusion; the bottom row illustrates boundary displacement and loss of thin structures in a dim bathroom scene.
The first case contains densely arranged furniture, small objects, and substantial occlusion. A large part of the visually complex background is assigned to the ignored category in the reference annotation, whereas SPGNet predicts semantic labels for several visible scene elements in that area. Because predictions at ignored pixels are excluded from the quantitative metrics, these colored regions should not be interpreted directly as segmentation errors. Within the labeled regions, however, the prediction remains locally fragmented around overlapping tables, chairs, and background furnishings. This example indicates that densely packed small structures, severe occlusion, and ambiguous interfaces can still cause confusion between adjacent object categories, while also demonstrating how incomplete annotations can complicate qualitative interpretation.
In the second case, SPGNet recovers the dominant semantic regions of the bathroom but loses several fine structural details. The predicted cyan region extends irregularly across the upper boundary of the dark-blue bathtub region, producing local boundary displacement and label leakage. Thin structures in the reference annotation, including the narrow fixture and bathtub rim, are omitted or absorbed into adjacent regions, and the object interfaces on the right are oversmoothed. Weak illumination, reflective or transparent surfaces, and thin metallic structures provide ambiguous appearance and geometric cues. When both modalities contain weak or inconsistent boundary evidence, the learned structure prior may therefore fail to preserve narrow structures or regulate cross-modal transfer accurately near object interfaces. These observations are consistent with the limitations discussed in Section 5 and motivate boundary-aware supervision, explicit uncertainty modeling, and broader failure analysis on held-out data.
The boundary observations in Figure 6, Figure 7 and Figure 8 are qualitative and should not be interpreted as a quantitative boundary-specific advantage. Boundary IoU and Boundary F1 depend on the boundary-extraction operator, matching tolerance or band width, treatment of ignored pixels, and normalization with respect to image resolution. These choices are particularly consequential for annotations containing coarse interfaces or extensive ignored regions. The archived evaluations do not retain full-resolution predictions for all reproduced methods; therefore, a controlled boundary comparison would require renewed inference for every method after fixing one common extraction, tolerance, ignored-pixel, and resolution-normalization protocol. This additional experiment is not included in the present revision, which retains the standard benchmark metrics and limits its boundary claims accordingly.
Visualization of the Learned Structure Prior. To examine the intermediate representation used by CCM, Figure 9 visualizes representative Stage 1 structure-prior maps on NYU Depth V2.
Figure 9. Visualization of learned Stage 1 structure priors generated by the Cross-Modal Correction Module (CCM) on NYU Depth V2. Each row shows the RGB image, the corresponding depth input, and the learned structure-prior response. For visualization, the Stage 1 prior tensor is averaged across channels, bilinearly upsampled to the input resolution, independently linearly scaled between its 1st and 99th percentiles, and shown without spatial smoothing. Brighter colors indicate stronger relative responses within the same map and should not be interpreted as absolute values comparable across samples.
The displayed response maps exhibit spatially organized patterns around prominent scene structures, including the piano and bench contours in the first example, the closet opening and nearby object boundaries in the second, and the shelf–aisle layout in the third. These responses are not identical to either raw RGB texture or depth intensity; instead, they summarize the learned prior constructed from joint consistency and discrepancy cues in the two encoded modalities. The maps therefore provide qualitative evidence that CCM forms a structure-sensitive intermediate representation before feature fusion. This visualization alone does not prove suppression of unreliable information, but, together with the controlled CCM ablation reported in Section 4.4, it supports the interpretation that the learned prior supplies spatial guidance for regulating subsequent cross-modal transfer.

4.3. Robustness Analysis

To isolate robustness to unreliable depth, we additionally evaluated the compared methods on the full NYU Depth V2 test set while keeping the RGB images and ground-truth labels unchanged. Only the normalized depth input was perturbed. Zero-mean Gaussian noise with standard deviations of 0.05 , 0.10 , and 0.20 was added to the depth tensor. For the misregistration test, only the depth map was translated by 1, 3, or 5 pixels along a cardinal direction selected independently for each test sample, with replicated boundary values. One fixed trained checkpoint per method was evaluated once for every condition; no retraining or test-time adaptation was performed. Table 4 reports mIoU, and the values in parentheses denote the absolute change in percentage points relative to the clean result from the same checkpoint. Because the archived robustness outputs did not retain the random perturbation seed, this table should be interpreted as a single-realization sensitivity analysis rather than a variance estimate over perturbation realizations.
Table 4. Depth-only robustness comparison on NYU Depth V2. RGB images and labels remain unchanged, while only depth is corrupted by Gaussian noise or spatially shifted. Parentheses indicate the mIoU change from the clean result of the same fixed checkpoint; the best score under each condition is in bold.
SPGNet retains the highest absolute mIoU under every evaluated depth perturbation. Its mIoU decreases from 50.78% on clean inputs to 50.22%, 49.41%, and 48.64% as the depth-noise level increases, remaining 0.12 percentage points above AsymFormer at σ = 0.20 . Under depth-only shifts of 1, 3, and 5 pixels, SPGNet obtains 50.79%, 50.78%, and 50.78% mIoU, respectively, and exceeds AsymFormer by 2.15 points at 5 pixels. These absolute results show that SPGNet preserves the strongest segmentation performance across the tested depth corruptions. Nevertheless, AsymFormer exhibits smaller relative decreases under the stronger noise settings; therefore, the evidence supports the highest retained absolute accuracy for SPGNet rather than the smallest degradation in every condition. Large contiguous missing-depth regions remain outside the scope of this experiment.

4.4. Ablation Study

To evaluate the contribution of each key component, this paper conducts step-by-step ablation experiments on SUN RGB-D under the unified training setting. All ablation settings use the same heterogeneous dual-branch encoder and differ only in whether CCM and DEF are enabled. The compared settings include Baseline, which uses neither CCM nor DEF; Baseline+CCM, which introduces only the cross-modal correction module; Baseline+DEF, which introduces only the dual-branch enhancement fusion module; and Full, which uses both CCM and DEF. In particular, Baseline+DEF is the fusion-only configuration: DEF remains enabled while CCM is removed, so the dual-modal features are enhanced and fused without preceding structure-prior-guided correction. The PVT-v2-B2 RGB encoder, ResNet34 depth encoder, multi-scale decoder, training schedule, and evaluation protocol are kept fixed across all four configurations. Therefore, the observed performance differences isolate the contributions of CCM and DEF rather than changes in the feature extractors. The quantitative results are reported in Table 5, and a representative visual comparison is shown in Figure 10.
Table 5. Ablation study on SUN RGB-D under the 37-class processed-label protocol.
Figure 10. Visual ablation comparison on SUN RGB-D. The panels show the RGB image, ground truth, Baseline, Baseline+CCM, Baseline+DEF, and the full SPGNet prediction.
Adding CCM alone improves mIoU from 45.395% to 46.375% (+0.980) and F1 from 59.309% to 60.271% (+0.962). Adding DEF alone improves mIoU to 45.817% (+0.422) and F1 to 59.651% (+0.342), although its OA and Kappa are slightly lower than those of the Baseline. Thus, DEF alone mainly improves overlap- and F1-related metrics on SUN RGB-D, while its effect on pixel-level accuracy is weaker when CCM is absent.
With both modules enabled, the full model achieves 48.418% mIoU, 62.418% F1, 81.217% OA, and 77.953% Kappa. Compared with the Baseline, the gains are 3.023, 3.109, 1.468, and 1.714 percentage points, respectively. More directly, comparing the full model with the fusion-only Baseline+DEF configuration measures the incremental benefit of adding CCM before DEF under otherwise identical settings. Adding pre-fusion correction increases mIoU from 45.817% to 48.418% (+2.601), F1 from 59.651% to 62.418% (+2.767), OA from 79.483% to 81.217% (+1.734), and Kappa from 76.009% to 77.953% (+1.944). On SUN RGB-D under the unified protocol, this controlled comparison shows that applying CCM before DEF improves final segmentation performance relative to the fusion-only configuration within the evaluated architecture. The full model outperforms both single-module variants on all four metrics, indicating that cross-modal correction and dual-branch enhancement fusion provide complementary contributions when used jointly. The visual example in Figure 10 further illustrates that the full model produces more complete object regions than the Baseline in this SUN RGB-D scene.

4.5. Model Complexity Analysis

To better characterize the trade-off between accuracy and model scale across different methods, this section re-profiles the RGB-D semantic segmentation methods appearing in the main quantitative comparisons using the local profiling framework, as shown in Table 6. All models are evaluated in inference mode with batch size 1 at 480 × 640 , using three-channel RGB images and one-channel depth maps; the depth map is repeated internally only when a model interface requires three channels. Parameter counts are computed directly from the instantiated models, and computational costs are reported as FLOPs under the common convention that one multiply–accumulate operation corresponds to two floating-point operations. For SPGNet, no layer is frozen during training and all model parameters are passed to the AdamW optimizer; therefore, its reported parameter count is also its trainable parameter count. End-to-end latency and throughput for every method are measured on an NVIDIA GeForce RTX 5070 Ti GPU under this common input protocol. CUDA synchronization is applied before and after each timed measurement, and timing begins only after 50 warm-up iterations. For each method, the reported latency is averaged over 1500 timed forward passes from three independent measurement runs, and frames per second (FPS) is computed as the reciprocal of the mean per-image latency. Timing covers the model forward pass to full-resolution logits and excludes model construction and host-to-device transfer. Peak allocated GPU memory is measured after warm-up by resetting the CUDA peak-memory statistics for each independent run, and the three run-wise peaks are averaged. Since fvcore does not support every operator used by these models, the FLOP values are partial profiling results and are intended for model-scale reference rather than exact hardware-independent complexity comparison. The profiled backbone column reflects the local adapter instantiation used for complexity measurement, while the mIoU columns follow the main quantitative comparison tables.
Table 6. Locally profiled complexity, runtime, and three-run mean Intersection over Union (mIoU) comparison of red–green–blue and depth (RGB-D) semantic segmentation methods appearing in the main quantitative experiments. Floating-point operations (FLOPs) are reported at 480 × 640 ; unsupported operators make the FLOP counts partial profiling results. End-to-end latency and frames per second (FPS) are measured for all methods on an NVIDIA GeForce RTX 5070 Ti with batch size 1 under the common input and timing protocol described in the text. All 122.21 M parameters reported for SPGNet are trainable.
In terms of parameter count, SPGNet contains a heterogeneous encoder, four-stage CCM, four-stage DEF, and a multi-scale fusion decoder, resulting in 122.21 M trainable parameters. This places it in the relatively large-model group among the locally instantiated models in this table. Its parameter count is higher than those of ACNet, SA-Gate, SGACNet, ShapeConv, and AsymFormer, but lower than that of PrimKD with MiT-B4 under the current local profiling setup.
In terms of computational cost, SPGNet requires 99.91 G partially profiled FLOPs under the local setup. Because unsupported operators differ across architectures, these FLOP values are provided as implementation-specific diagnostics and are not used for strict cross-model ranking. The common runtime measurements show that SPGNet requires 34.64 ms per image (28.87 FPS). It is slower than ACNet, SA-Gate, SGACNet, ShapeConv, and AsymFormer, but faster than PrimKD under the tested hardware and implementation environment. These results provide a hardware-specific indication of practical deployment cost. Combined with the accuracy results in Table 2 and Table 3, SPGNet achieves the highest mIoU on both NYU Depth V2 and SUN RGB-D among the compared methods, while requiring a relatively large parameter count and mid-range measured latency. Therefore, the proposed method should be regarded as an accuracy-oriented model rather than a lightweight design. Future work may further explore lightweight attention projections and channel transformations in CCM and DEF.
To isolate the computational overhead associated with CCM and DEF, the four controlled ablation configurations are also profiled under the same hardware, input resolution, batch size, inference mode, and timing protocol. As reported in Table 7, Baseline+CCM and Baseline+DEF differ from the common Baseline only by the corresponding module, while the full SPGNet enables both modules. This controlled profiling separates module-related changes in parameter count, FLOPs, latency, and peak allocated GPU memory from changes in the encoders, decoder, or input protocol. Runtime and peak-memory values are interpreted per configuration rather than added across rows because kernel execution and memory reuse are not strictly additive.
Table 7. Controlled module-overhead profiling of the Cross-Modal Correction Module (CCM) and Dual-branch Enhancement Fusion Module (DEF). All configurations use identical encoders and decoder and are profiled on an NVIDIA GeForce RTX 5070 Ti with batch size 1 and 3 × 480 × 640 RGB and repeated-depth inputs.

5. Conclusions

This paper proposes the Structure-Prior-Guided Network (SPGNet) for RGB-D semantic segmentation, with the aim of improving multimodal feature interaction under cross-modal response inconsistency, unreliable depth-response propagation, and limited multi-scale fusion. SPGNet adopts a heterogeneous dual-encoder architecture to model RGB appearance semantics and depth geometric structures separately, and applies the Cross-Modal Correction Module (CCM) and the Dual-branch Enhancement Fusion Module (DEF) sequentially at four scales. CCM constructs structure priors from RGB–depth consistency and discrepancy cues, and uses these priors for structural modulation, gated selection, and bidirectional cross-modal correction before fusion. DEF further enhances the corrected dual-branch features and performs shared-guided interaction before the multi-scale fusion decoder aggregates features from different levels for pixel-level prediction.
Under the dataset-specific processed-label protocols used in this paper, SPGNet obtains three-run mean mIoU, F1, OA, and Kappa values of 50.845%, 64.985%, 76.497%, and 73.956% on NYU Depth V2 and 48.457%, 62.496%, 81.183%, and 77.900% on SUN RGB-D, respectively. These results are favorable among the compared representative RGB-D segmentation methods in the reported evaluation setting. The ablation experiments further show that, under the same heterogeneous dual-branch encoder, jointly enabling CCM and DEF improves all four metrics on SUN RGB-D, which is consistent with the complementary roles of pre-fusion structural correction and subsequent enhancement fusion. The complexity analysis indicates that these accuracy gains are accompanied by a relatively large parameter count and computational cost, positioning SPGNet as an accuracy-oriented model rather than a lightweight design.
This study has several limitations. First, although the added depth-only Gaussian-noise and spatial-shift experiments evaluate SPGNet under controlled depth corruption while preserving the RGB input, they do not cover large contiguous missing-depth regions. The effects of severe depth dropout and more substantial RGB–depth misregistration therefore require further study. Second, the current experiments are limited to indoor RGB-D datasets; therefore, generalization to outdoor scenes with different sensing ranges, illumination conditions, depth statistics, and class distributions remains unverified. Transferability to other spatially aligned multimodal segmentation settings, such as RGB–thermal or RGB–LiDAR segmentation, also remains to be established. Because CCM and DEF operate on paired stage-wise feature maps rather than raw sensor-specific values, the framework could in principle be adapted by replacing the depth encoder and relearning modality-specific structure priors. However, differences in sensor alignment, noise characteristics, and spatial sparsity require task-specific validation. Third, SPGNet is an accuracy-oriented rather than lightweight model. It contains 122.21 M trainable parameters, and Table 6 reports 99.91 G FLOPs for the profiled portion of the model at an input resolution of 480 × 640 . The parameter count is independent of input resolution, whereas the computational and memory costs increase as the resolution grows. For convolutional components with fixed channel dimensions and kernels, the computational cost grows approximately linearly with the number of spatial locations. In CCM and DEF, the query retains the full spatial resolution at each stage, while the key and value maps are average-pooled by a ratio r along both spatial dimensions. Thus, for a stage feature map with N = h i w i query locations, the key/value sequence contains approximately N / r 2 locations, and the cross-attention score computation and score-matrix memory scale as O ( N 2 / r 2 ) , assuming fixed channel dimensions. Consequently, if the input height and width are both increased by a factor of s, convolutional cost grows approximately as s 2 , whereas this attention term grows approximately as s 4 for a fixed r. The stage-wise key/value downsampling ratios used in CCM ( { 8 , 4 , 4 , 2 } ) and DEF ( { 8 , 4 , 2 , 1 } ) reduce, but do not eliminate, this resolution-dependent growth. Finally, scenes involving severe occlusion, weak or ambiguous boundaries, reflective or transparent surfaces, or strong RGB–depth misalignment may remain challenging. The diagnostic examples in Figure 8 concretely illustrate fragmentation under clutter and occlusion, as well as boundary displacement and loss of thin structures under weak and reflective visual conditions. Although controlled depth noise and shift perturbations are now evaluated, these individual scene and sensor failure modes still require more systematic investigation on held-out data. Complete per-class behavior, instance-scale-stratified performance, and boundary-focused accuracy under a predefined extraction and tolerance protocol also remain to be evaluated.
Future work will focus on three directions. First, the parameter overhead introduced by attention projections and multi-channel convolutions in CCM and DEF can be reduced to improve deployment feasibility in real-time or resource-constrained scenarios. This direction will include evaluating lightweight convolutional networks and hierarchical Transformer architectures as alternative depth encoders to characterize the accuracy–efficiency trade-off and the generality of the heterogeneous encoder design. Second, robustness to controlled levels of depth degradation and the generalization ability of structure-prior-guided correction can be evaluated on more indoor and outdoor RGB-D datasets, cross-domain settings, and other spatially aligned multimodal segmentation tasks. Finally, more fine-grained visualization and error-case analyses can be incorporated to better explain the roles of structure priors, gated selection, and enhancement fusion for different scene categories and boundary regions. This direction will use a predefined protocol to report complete per-class results, stratify performance by instance scale and boundary difficulty, and quantify boundary accuracy under explicitly stated extraction operators and tolerance widths.

Author Contributions

Conceptualization, Y.Y. and F.M.; methodology, Y.Y. and Z.Z.; software, Y.Y., Z.Z. and S.D.; validation, Y.Y., Z.Z. and S.D.; formal analysis, Y.Y. and Z.Z.; investigation, Y.Y. and S.D.; resources, Y.Y. and S.D.; data curation, Y.Y. and S.D.; writing—original draft preparation, Y.Y.; writing—review and editing, Y.Y., Z.Z. and F.M.; visualization, Y.Y. and Z.Z.; supervision, F.M.; project administration, F.M. All authors have read and agreed to the published version of the manuscript.

Funding

This research received no external funding.

Institutional Review Board Statement

Not applicable.

Data Availability Statement

The NYU Depth V2 and SUN RGB-D datasets analyzed in this study were obtained from their official project websites: NYU Depth V2 (https://cs.nyu.edu/~fergus/datasets/nyu_depth_v2.html) and SUN RGB-D (https://rgbd.cs.princeton.edu/) (both accessed on 6 August 2026). The corresponding original dataset publications are cited in Refs. [2,3]. The source code and trained models of this study will be made available from the corresponding author upon reasonable request.

Acknowledgments

The authors would like to thank Southwest Petroleum University for providing the research environment that supported this work. The authors also gratefully acknowledge the teams and institutions responsible for collecting, annotating, publicly releasing, and maintaining the NYU Depth V2 and SUN RGB-D datasets, whose efforts have supported continued research in indoor RGB-D scene understanding. During the preparation of this manuscript, the authors used GPT-5.4 (OpenAI) for the purposes of assisting with English-language editing, grammatical correction, and improving the clarity of the manuscript. The authors have reviewed and edited the output and take full responsibility for the content of this publication.

Conflicts of Interest

The authors declare no conflicts of interest.

Abbreviations

The following abbreviations are used in this manuscript:
RGB-DRed–Green–Blue and Depth
SPGNetStructure-Prior-Guided Network
CCMCross-Modal Correction Module
DEFDual-branch Enhancement Fusion Module
mIoUMean Intersection over Union
F1F1-score
OAOverall Accuracy
ACNetAttention Complementary Network
SA-GateSeparation-and-Aggregation Gate
PrimKDPrimary-Modality-Guided Knowledge Distillation
SGACNetSpatial-Information-Guided Adaptive Context-Aware Network
ShapeConvShape-Aware Convolutional Layer
AsymFormerAsymmetrical Cross-Modal Representation Learning

References

  1. Long, J.; Shelhamer, E.; Darrell, T. Fully Convolutional Networks for Semantic Segmentation. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR); IEEE: New York, NY, USA, 2015. [Google Scholar]
  2. Silberman, N.; Hoiem, D.; Kohli, P.; Fergus, R. Indoor Segmentation and Support Inference from RGBD Images. In Proceedings of the European Conference on Computer Vision (ECCV); Springer: Berlin/Heidelberg, Germany, 2012. [Google Scholar]
  3. Song, S.; Lichtenberg, S.P.; Xiao, J. SUN RGB-D: A RGB-D Scene Understanding Benchmark Suite. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR); IEEE: New York, NY, USA, 2015. [Google Scholar]
  4. Gupta, S.; Girshick, R.; Arbeláez, P.; Malik, J. Learning Rich Features from RGB-D Images for Object Detection and Segmentation. In Proceedings of the European Conference on Computer Vision (ECCV); Springer: Berlin/Heidelberg, Germany, 2014. [Google Scholar]
  5. Hazirbas, C.; Ma, L.; Domokos, C.; Cremers, D. FuseNet: Incorporating Depth into Semantic Segmentation via Fusion-Based CNN Architecture. In Proceedings of the Asian Conference on Computer Vision (ACCV); Springer: Berlin/Heidelberg, Germany, 2016. [Google Scholar]
  6. Park, S.J.; Hong, K.S.; Lee, S. RDFNet: RGB-D Multi-Level Residual Feature Fusion for Indoor Semantic Segmentation. In Proceedings of the IEEE International Conference on Computer Vision (ICCV); IEEE: New York, NY, USA, 2017. [Google Scholar]
  7. Wang, W.; Neumann, U. Depth-Aware CNN for RGB-D Segmentation. In Proceedings of the European Conference on Computer Vision (ECCV); Springer: Berlin/Heidelberg, Germany, 2018. [Google Scholar]
  8. Jiao, J.; Wei, Y.; Jie, Z.; Shi, H.; Lau, R.W.H.; Huang, T.S. Geometry-Aware Distillation for Indoor Semantic Segmentation. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR); IEEE: New York, NY, USA, 2019. [Google Scholar]
  9. Chen, X.; Lin, K.Y.; Wang, J.; Wu, W.; Qian, C.; Li, H.; Zeng, G. Bi-Directional Cross-Modality Feature Propagation with Separation-and-Aggregation Gate for RGB-D Semantic Segmentation. In Proceedings of the European Conference on Computer Vision (ECCV); Springer: Berlin/Heidelberg, Germany, 2020; pp. 561–577. [Google Scholar] [CrossRef] [Scilit]
  10. Zhang, J.; Liu, H.; Yang, K.; Hu, X.; Liu, R.; Stiefelhagen, R. CMX: Cross-Modal Fusion for RGB-X Semantic Segmentation With Transformers. IEEE Trans. Intell. Transp. Syst. 2023, 24, 14679–14694. [Google Scholar] [CrossRef] [Scilit]
  11. Cao, J.; Leng, H.; Lischinski, D.; Cohen-Or, D.; Tu, C.; Li, Y. ShapeConv: Shape-Aware Convolutional Layer for Indoor RGB-D Semantic Segmentation. In Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV); IEEE: New York, NY, USA, 2021. [Google Scholar]
  12. Wang, Y.; Chen, X.; Cao, L.; Huang, W.; Sun, F.; Wang, Y. Multimodal Token Fusion for Vision Transformers. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR); IEEE: New York, NY, USA, 2022. [Google Scholar]
  13. Du, S.; Wang, W.; Guo, R.; Wang, R.; Tang, S. AsymFormer: Asymmetrical Cross-Modal Representation Learning for Mobile Platform Real-Time RGB-D Semantic Segmentation. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW); IEEE: New York, NY, USA, 2024. [Google Scholar]
  14. Yin, B.; Zhang, X.; Li, Z.; Liu, L.; Cheng, M.M.; Hou, Q. DFormer: Rethinking RGBD Representation Learning for Semantic Segmentation. In Proceedings of the International Conference on Learning Representations (ICLR), Vienna, Austria, 7–11 May 2024. [Google Scholar]
  15. Hao, Z.; Xiao, Z.; Luo, Y.; Guo, J.; Wang, J.; Shen, L.; Hu, H. PrimKD: Primary Modality Guided Multimodal Fusion for RGB-D Semantic Segmentation. In Proceedings of the ACM International Conference on Multimedia (ACM MM); Association for Computing Machinery: New York, NY, USA, 2024; pp. 1943–1951. [Google Scholar] [CrossRef] [Scilit]
  16. Yin, B.W.; Cao, J.L.; Cheng, M.M.; Hou, Q. DFormerv2: Geometry Self-Attention for RGBD Semantic Segmentation. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR); IEEE: New York, NY, USA, 2025; pp. 19345–19355. [Google Scholar] [CrossRef] [Scilit]
  17. Gupta, S.; Arbeláez, P.; Malik, J. Perceptual Organization and Recognition of Indoor Scenes from RGB-D Images. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR); IEEE: New York, NY, USA, 2013. [Google Scholar]
  18. Eigen, D.; Fergus, R. Predicting Depth, Surface Normals and Semantic Labels with a Common Multi-Scale Convolutional Architecture. In Proceedings of the IEEE International Conference on Computer Vision (ICCV); IEEE: New York, NY, USA, 2015. [Google Scholar]
  19. Deng, Z.; Todorovic, S.; Latecki, L.J. Semantic Segmentation of RGBD Images with Mutex Constraints. In Proceedings of the IEEE International Conference on Computer Vision (ICCV); IEEE: New York, NY, USA, 2015. [Google Scholar]
  20. Li, Z.; Gan, Y.; Liang, X.; Yu, Y.; Cheng, H.; Lin, L. LSTM-CF: Unifying Context Modeling and Fusion with LSTMs for RGB-D Scene Labeling. In Proceedings of the European Conference on Computer Vision (ECCV); Springer: Berlin/Heidelberg, Germany, 2016. [Google Scholar]
  21. Wang, J.; Wang, Z.; Tao, D.; See, S.; Wang, G. Learning Common and Specific Features for RGB-D Semantic Segmentation with Deconvolutional Networks. In Proceedings of the European Conference on Computer Vision (ECCV); Springer: Berlin/Heidelberg, Germany, 2016; pp. 664–679. [Google Scholar] [CrossRef] [Scilit]
  22. He, Y.; Chiu, W.C.; Keuper, M.; Fritz, M. STD2P: RGBD Semantic Segmentation Using Spatio-Temporal Data-Driven Pooling. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR); IEEE: New York, NY, USA, 2017. [Google Scholar]
  23. Qi, X.; Liao, R.; Jia, J.; Fidler, S.; Urtasun, R. 3D Graph Neural Networks for RGBD Semantic Segmentation. In Proceedings of the IEEE International Conference on Computer Vision (ICCV); IEEE: New York, NY, USA, 2017. [Google Scholar]
  24. Cheng, Y.; Cai, R.; Li, Z.; Zhao, X.; Huang, K. Locality-Sensitive Deconvolution Networks with Gated Fusion for RGB-D Indoor Semantic Segmentation. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR); IEEE: New York, NY, USA, 2017. [Google Scholar]
  25. Jiang, J.; Zheng, L.; Luo, F.; Zhang, Z. RedNet: Residual Encoder-Decoder Network for Indoor RGB-D Semantic Segmentation. arXiv 2018, arXiv:1806.01054. [Google Scholar]
  26. Xu, D.; Ouyang, W.; Wang, X.; Sebe, N. PAD-Net: Multi-Tasks Guided Prediction-and-Distillation Network for Simultaneous Depth Estimation and Scene Parsing. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR); IEEE: New York, NY, USA, 2018. [Google Scholar]
  27. Hu, X.; Yang, K.; Fei, L.; Wang, K. ACNet: Attention Based Network to Exploit Complementary Features for RGBD Semantic Segmentation. In Proceedings of the IEEE International Conference on Image Processing (ICIP); IEEE: New York, NY, USA, 2019. [Google Scholar]
  28. Zhou, H.; Qi, L.; Wan, Z.; Huang, H.; Yang, X. RGB-D Co-Attention Network for Semantic Segmentation. In Proceedings of the Asian Conference on Computer Vision (ACCV); Springer: Berlin/Heidelberg, Germany, 2020. [Google Scholar]
  29. Xing, Y.; Wang, J.; Zeng, G. Malleable 2.5D Convolution: Learning Receptive Fields along the Depth-axis for RGB-D Scene Parsing. In Proceedings of the European Conference on Computer Vision (ECCV); Springer: Berlin/Heidelberg, Germany, 2020. [Google Scholar]
  30. Lin, D.; Zhang, R.; Ji, Y.; Li, P.; Huang, H. SCN: Switchable Context Network for Semantic Segmentation of RGB-D Images. IEEE Trans. Cybern. 2020, 50, 1120–1131. [Google Scholar] [CrossRef] [Scilit] [PubMed]
  31. Lin, D.; Huang, H. Zig-Zag Network for Semantic Segmentation of RGB-D Images. IEEE Trans. Pattern Anal. Mach. Intell. 2020, 42, 2642–2655. [Google Scholar] [CrossRef] [Scilit] [PubMed]
  32. Wang, Y.; Huang, W.; Sun, F.; Xu, T.; Rong, Y.; Huang, J. Deep Multimodal Fusion by Channel Exchanging. In Proceedings of the Advances in Neural Information Processing Systems (NeurIPS); Curran Associates Inc.: Red Hook, NY, USA, 2020; Volume 33. [Google Scholar]
  33. Chen, L.Z.; Lin, Z.; Wang, Z.; Yang, Y.L.; Cheng, M.M. Spatial Information Guided Convolution for Real-Time RGB-D Semantic Segmentation. IEEE Trans. Image Process. 2021, 30, 2313–2324. [Google Scholar] [CrossRef] [Scilit] [PubMed]
  34. Seichter, D.; Köhler, M.; Lewandowski, B.; Wengefeld, T.; Gross, H.M. Efficient RGB-D Semantic Segmentation for Indoor Scene Analysis. In Proceedings of the IEEE International Conference on Robotics and Automation (ICRA); IEEE: New York, NY, USA, 2021. [Google Scholar]
  35. Yang, E.; Zhou, W.; Qian, X.; Yu, L. MGCNet: Multilevel Gated Collaborative Network for RGB-D Semantic Segmentation of Indoor Scene. IEEE Signal Process. Lett. 2022, 29, 2567–2571. [Google Scholar] [CrossRef] [Scilit]
  36. Zhou, W.; Yang, E.; Lei, J.; Wan, J.; Yu, L. PGDENet: Progressive Guided Fusion and Depth Enhancement Network for RGB-D Indoor Scene Parsing. IEEE Trans. Multimed. 2023, 25, 3483–3494. [Google Scholar] [CrossRef] [Scilit]
  37. Zhang, Y.; Xiong, C.; Liu, J.; Ye, X.; Sun, G. Spatial-Information Guided Adaptive Context-Aware Network for Efficient RGB-D Semantic Segmentation. IEEE Sens. J. 2023, 23, 23512–23521. [Google Scholar] [CrossRef] [Scilit]
  38. Yang, J.; Bai, L.; Sun, Y.; Tian, C.; Mao, M.; Wang, G. Pixel Difference Convolutional Network for RGB-D Semantic Segmentation. IEEE Trans. Circuits Syst. Video Technol. 2024, 34, 1481–1492. [Google Scholar] [CrossRef] [Scilit]
  39. Bai, L.; Yang, J.; Tian, C.; Sun, Y.; Mao, M.; Xu, Y.; Xu, W. DCANet: Differential Convolution Attention Network for RGB-D Semantic Segmentation. Pattern Recognit. 2025, 162, 111379. [Google Scholar] [CrossRef] [Scilit]
  40. Wu, Z.; Zhou, Z.; Allibert, G.; Stolz, C.; Demonceaux, C.; Ma, C. Transformer Fusion for Indoor RGB-D Semantic Segmentation. Comput. Vis. Image Underst. 2024, 249, 104174. [Google Scholar] [CrossRef] [Scilit]
  41. Ying, X.; Chuah, M.C. UCTNet: Uncertainty-Aware Cross-Modal Transformer Network for Indoor RGB-D Semantic Segmentation. In Proceedings of the European Conference on Computer Vision (ECCV); Springer: Berlin/Heidelberg, Germany, 2022. [Google Scholar] [CrossRef] [Scilit]
  42. Zhang, J.; Liu, R.; Shi, H.; Yang, K.; Reiß, S.; Peng, K.; Fu, H.; Wang, K.; Stiefelhagen, R. Delivering Arbitrary-Modal Semantic Segmentation. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR); IEEE: New York, NY, USA, 2023. [Google Scholar]
  43. Dong, S.; Feng, Y.; Yang, Q.; Huang, Y.; Liu, D.; Fan, H. Efficient Multimodal Semantic Segmentation via Dual-Prompt Learning. In Proceedings of the IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS); IEEE: New York, NY, USA, 2024; pp. 14196–14203. [Google Scholar] [CrossRef] [Scilit]
  44. Ge, M.; Su, W.; Gao, J.; Jia, G. CDMANet: Central Difference Mutual Attention Network for RGB-D Semantic Segmentation. J. Supercomput. 2025, 81, 242. [Google Scholar] [CrossRef] [Scilit]
  45. Cai, J.; Su, J.; Li, Q.; Yang, W.; Wang, S.; Zhao, T.; He, S.; Liu, W. Keep the Balance: A Parameter-Efficient Symmetrical Framework for RGB+X Semantic Segmentation. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR); IEEE: New York, NY, USA, 2025; pp. 10587–10598. [Google Scholar] [CrossRef] [Scilit]
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content.

Article Metrics

Citations

Article Access Statistics

Multiple requests from the same IP address are counted as one view.