Hierarchical Adaptive Feature Refinement Network for VHR Remote Sensing Image Segmentation
Abstract
Semantic segmentation of very-high-resolution (VHR) remote sensing imagery increasingly benefits from strong pretrained hierarchical encoders, yet exploiting their multi-stage representations remains difficult. Nearby regions demand different balances between fine detail and semantic context, aggressive task-specific transformations perturb useful pretrained features, and conventional semantic supervision provides limited structural guidance. We present HAFR-Net, a progressive refinement framework that adaptively organizes and conservatively refines hierarchical representations instead of replacing them with a monolithic decoder transformation. Heterogeneity-Guided Stage-Adaptive Fusion (HG-SAF) predicts dense stage weights conditioned on local feature variation. A Frequency-Residual Adapter (FRA) then injects frequency information through a bounded, zero-initialized residual branch that keeps the fused representation as its reference. A Confusion-Aware Tri-Prior Decoder (CATP) finally regularizes the prediction with boundary, objectness, and training-derived class-relation cues. Under a matched Swin-B training and single-scale inference protocol, HAFR-Net attains 84.12%, 87.86%, 55.17%, and 67.70% mIoU on ISPRS Vaihingen, ISPRS Potsdam, LoveDA, and OpenEarthMap, improving the matched UPerNet baseline by 0.55, 0.95, 1.55, and 1.84 percentage points, respectively. Controlled analyses further show consistent spatial reweighting beyond content-only routing, improved boundary and thin-structure accuracy over matched spatial and spectral alternatives, and reduced confusion on pre-declared class pairs.
Figures & tables
| Branch | Init. | First forward | Role |
|---|---|---|---|
| HG-SAF | last gate | uniform fusion | adaptive fusion |
| FRA | exact identity | residual refine. | |
| CATP fb. | near-ref. residual | struct. feedback |
| Method (Swin-B) | Vaih. | Pots. | LoveDA | OEM | Avg. |
|---|---|---|---|---|---|
| UPerNet [ 13 ] | 83.57 | 86.91 | 53.62 | 65.86 | 72.49 |
| FPN [ 12 ] | 83.21 | 86.54 | 53.18 | 65.41 | 72.09 |
| DeepLabv3+ [ 7 ] | 83.38 | 86.72 | 53.35 | 65.58 | 72.26 |
| SegFormer MLP [ 9 ] | 83.44 | 86.80 | 53.48 | 65.71 | 72.36 |
| UNetFormer dec. [ 14 ] | 83.79 | 87.14 | 54.02 | 66.18 | 72.78 |
| SE-gated fusion [ 31 ] | 83.68 | 87.05 | 53.88 | 66.02 | 72.66 |
| ISPRS Vaihingen | ISPRS Potsdam | |||||||||||
| Method | Imp.S. | Build. | L.Veg. | Tree | Car | mIoU | Imp.S. | Build. | L.Veg. | Tree | Car | mIoU |
| CNN-based | ||||||||||||
| UNet [ 5 ] | 90.1 | 88.0 | 69.1 | 78.4 | 64.5 | 78.02 | 82.8 | 89.9 | 73.8 | 75.6 | 83.2 | 81.06 |
| PSPNet [ 6 ] | 91.0 | 88.7 | 69.5 | 79.1 | 66.8 | 79.02 | 84.2 | 91.0 | 74.8 | 76.5 | 85.2 | 82.34 |
| DenseASPP [ 60 ] | 90.3 | 87.6 | 68.4 | 77.9 | 62.5 | 77.34 | 83.3 | 90.3 | 73.5 | 75.3 | 83.8 | 81.24 |
| MANet [ 28 ] | 91.2 | 89.2 | 70.1 | 79.4 | 68.9 | 79.76 | 84.8 | 91.8 | 75.4 | 77.1 | 86.2 | 83.06 |
| Method | Backgr. | Building | Road | Water | Barren | Forest | Agricult. | mIoU |
|---|---|---|---|---|---|---|---|---|
| CNN-based | ||||||||
| UNet [ 5 ] | 50.24 | 59.77 | 52.79 | 57.02 | 31.08 | 41.73 | 51.81 | 49.20 |
| PSPNet [ 6 ] | 52.50 | 59.46 | 51.15 | 61.10 | 27.46 | 37.48 | 50.32 | 48.49 |
| DenseASPP [ 60 ] | 50.77 | 55.67 | 53.08 | 55.49 | 23.99 | 35.24 | 43.54 | 45.40 |
| MANet [ 28 ] | 49.82 | 58.08 | 52.98 | 54.30 | 25.21 | 37.65 | 51.87 | 47.13 |
| HRNet [ 25 ] | 49.90 | 57.33 | 55.94 | 52.34 | 31.64 | 41.94 | 48.50 | 48.23 |
| Method | Bareland | Range. | Develop. | Road | Tree | Water | Agricult. | Building | mIoU |
|---|---|---|---|---|---|---|---|---|---|
| CNN-based | |||||||||
| UNet [ 5 ] | 33.6 | 49.2 | 49.8 | 56.1 | 64.7 | 71.4 | 70.9 | 72.3 | 58.50 |
| PSPNet [ 6 ] | 35.2 | 50.1 | 50.6 | 57.4 | 65.8 | 72.6 | 71.8 | 73.0 | 59.56 |
| DenseASPP [ 60 ] | 32.9 | 48.4 | 49.1 | 55.6 | 63.9 | 70.7 | 70.1 | 71.6 | 57.79 |
| MANet [ 28 ] | 36.0 | 50.8 | 51.2 | 58.0 | 66.5 | 73.4 | 72.5 | 73.8 | 60.27 |
| HRNet [ 25 ] | 35.6 | 50.5 | 50.9 | 57.7 | 66.1 | 73.0 | 72.1 | 73.4 | 59.91 |
| Setting | Pixel-wise | Extra cue | mIoU | ||
|---|---|---|---|---|---|
| Panel A: fusion-gate controls | |||||
| H1 | Mean fusion | – | – | 84.28 | 0.20 |
| H2 | Global-pool gate | – | GAP | 84.32 | 0.16 |
| H3 | Pixel gate, only | ✓ | – | 84.34 | 0.14 |
| H4 | Pixel gate input gradient | ✓ | 84.40 | 0.08 | |
| H5 | Pixel gate entropy | ✓ | 84.37 | 0.11 | |
| Variant | Par. | mIoU | BIoU | BF | Thin | Small |
|---|---|---|---|---|---|---|
| None (spatial only) | – | 84.31 | 71.42 | 74.86 | 68.26 | 76.61 |
| Spatial residual (matched) | 1.1 | 84.22 | 71.78 | 75.14 | 68.58 | 76.84 |
| Standard FFT filter | 1.2 | 84.19 | 72.06 | 75.41 | 68.94 | 76.72 |
| FFC-like branch [ 17 ] | 3.4 | 84.08 | 71.64 | 75.02 | 68.41 | 76.38 |
| GFNet mixer [ 16 ] | 4.8 | 83.96 | 71.28 | 74.71 | 68.05 | 76.12 |
| GFNet zero-init res. | 4.8 | 84.28 | 72.54 | 75.88 | 69.42 | 77.05 |
| Variant | mIoU | ||
|---|---|---|---|
| Panel A: matched auxiliary-loss controls (Vaihingen) | |||
| S1 | CE FocalDice only | 84.26 | 0.22 |
| S2 | boundary loss [ 22 ] | 84.28 | 0.20 |
| S3 | objectness loss | 84.29 | 0.19 |
| S4 | pixel contrast [ 23 ] | 84.31 | 0.17 |
| S5 | all three conventional losses | 84.32 | 0.16 |
| HG-SAF | FRA | CATP | Vaih. | Pots. | LoveDA | OEM | Avg. |
|---|---|---|---|---|---|---|---|
| Panel A: factorial combinations | |||||||
| – | – | – | 83.81 | 87.30 | 53.91 | 66.32 | 72.84 |
| ✓ | – | – | 84.04 | 87.65 | 54.46 | 66.85 | 73.25 |
| – | ✓ | – | 83.99 | 87.52 | 54.18 | 66.61 | 73.08 |
| – | – | ✓ | 84.19 | 87.78 | 54.58 | 66.94 | 73.37 |
| ✓ | ✓ | – | 84.26 | 87.93 | 54.81 | 67.27 | 73.57 |
| Preparation | mIoU | ||
|---|---|---|---|
| T1 | none (reference decoder) | 83.57 | – |
| T2 | SSDB only | 83.66 | |
| T3 | BCS-Mamba only | 83.71 | |
| T4 | SOE only | 83.64 | |
| T5 | SSDB BCS-Mamba SOE | 83.79 |
| Method | Par. | GFLOPs | Mem. | Lat. | FPS | mIoU |
|---|---|---|---|---|---|---|
| Swin-B FPN | 90.1 | 102.6 | 1.72 | 18.9 | 52.9 | 72.09 |
| Swin-B UPerNet | 89.3 | 118.4 | 1.84 | 21.4 | 46.7 | 72.49 |
| Swin-B UNetFormer dec. | 92.6 | 124.1 | 1.91 | 22.6 | 44.2 | 72.78 |
| Swin-B FFT mixer | 93.8 | 129.7 | 1.98 | 23.5 | 42.6 | 72.56 |
| HAFR-Net | 97.8 | 131.2 | 2.12 | 24.8 | 40.3 | 73.71 |
Remote Sensing Image Segmentation
Remote sensing image segmentation labels land cover and objects in satellite, aerial, radar, and multispectral imagery.
See all 220 papers