VistaFormer: Scalable Vision Transformers for Satellite Image Time Series Segmentation
This work addresses crop-type segmentation in remote sensing, offering incremental improvements in efficiency and accuracy for domain-specific applications.
The paper tackles semantic segmentation of satellite image time series by introducing VistaFormer, a lightweight Transformer-based model that achieves better performance than comparable models with significantly fewer floating point operations and parameters, improving mIoU scores by up to 3.7% on benchmarks.
We introduce VistaFormer, a lightweight Transformer-based model architecture for the semantic segmentation of remote-sensing images. This model uses a multi-scale Transformer-based encoder with a lightweight decoder that aggregates global and local attention captured in the encoder blocks. VistaFormer uses position-free self-attention layers which simplifies the model architecture and removes the need to interpolate temporal and spatial codes, which can reduce model performance when training and testing image resolutions differ. We investigate simple techniques for filtering noisy input signals like clouds and demonstrate that improved model scalability can be achieved by substituting Multi-Head Self-Attention (MHSA) with Neighbourhood Attention (NA). Experiments on the PASTIS and MTLCC crop-type segmentation benchmarks show that VistaFormer achieves better performance than comparable models and requires only 8% of the floating point operations using MHSA and 11% using NA while also using fewer trainable parameters. VistaFormer with MHSA improves on state-of-the-art mIoU scores by 0.1% on the PASTIS benchmark and 3% on the MTLCC benchmark while VistaFormer with NA improves on the MTLCC benchmark by 3.7%.