English

SatReg: Regression-based Neural Architecture Search for Lightweight Satellite Image Segmentation

Computer Vision and Pattern Recognition 2026-04-14 v1

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.

Keywords

Cite

@article{arxiv.2604.10306,
  title  = {SatReg: Regression-based Neural Architecture Search for Lightweight Satellite Image Segmentation},
  author = {Edward Humes and Tinoosh Mohsenin},
  journal= {arXiv preprint arXiv:2604.10306},
  year   = {2026}
}
R2 v1 2026-07-01T12:04:30.881Z