SatReg: Regression-based Neural Architecture Search for Lightweight Satellite Image Segmentation
Organizations: Johns Hopkins University, Baltimore, Maryland, USA
Abstract
As Earth-observation workloads move toward onboard and edge processing, remote-sensing segmentation models must operate under tight latency and energy constraints. We present SatReg, a regression-based hardware-aware tuning framework for lightweight remote-sensing segmentation on edge platforms. Using CM-UNet as the teacher architecture, we reduce the search space to two dominant width-related variables, profile a small set of student models on an NVIDIA Jetson Orin Nano, and fit low-order surrogate models for mIoU, latency, and power. Knowledge distillation is used to efficiently train the sampled students. The learned surrogates enable fast selection of near-optimal architecture settings for deployment targets without exhaustive search. Results show that the selected variables affect task accuracy and hardware cost differently, making reduced-space regression a practical strategy for adapting hybrid CNN-Mamba segmentation models to future space-edge systems.
Figures & tables
| Predicted | Deployed | Predicted | Deployed | |||
| Parameters | CM-UNet ( Liu et al., 2024a ) | SatReg (Ours) | CM-UNet | SatReg (Ours) | ||
| Dataset | LoveDA | Vaihingen | ||||
| b | 64 | 32 | 64 | 40 | ||
| h | 32 | 8 | 32 | 4 | ||
| Test mIoU (%) | 50.21 | 47.54 | 47.45 | 84.64 | 80.48 | 80.64 |
| Latency (ms) | 178.63 | 108.20 | 109.72 | 178.27 | 110.22 | 119.69 |
Hardware-Aware NAS
Hardware-aware neural architecture search selects and co-designs models for accuracy, latency, energy, memory, and target-device constraints.
See all 49 papers