Seeing the roads through the trees: A benchmark for modeling spatial dependencies with aerial imagery (2401.06762v1)
Abstract: Fully understanding a complex high-resolution satellite or aerial imagery scene often requires spatial reasoning over a broad relevant context. The human object recognition system is able to understand object in a scene over a long-range relevant context. For example, if a human observes an aerial scene that shows sections of road broken up by tree canopy, then they will be unlikely to conclude that the road has actually been broken up into disjoint pieces by trees and instead think that the canopy of nearby trees is occluding the road. However, there is limited research being conducted to understand long-range context understanding of modern machine learning models. In this work we propose a road segmentation benchmark dataset, Chesapeake Roads Spatial Context (RSC), for evaluating the spatial long-range context understanding of geospatial machine learning models and show how commonly used semantic segmentation models can fail at this task. For example, we show that a U-Net trained to segment roads from background in aerial imagery achieves an 84% recall on unoccluded roads, but just 63.5% recall on roads covered by tree canopy despite being trained to model both the same way. We further analyze how the performance of models changes as the relevant context for a decision (unoccluded roads in our case) varies in distance. We release the code to reproduce our experiments and dataset of imagery and masks to encourage future research in this direction -- https://github.com/isaaccorley/ChesapeakeRSC.
- “Hybrid multiple attention network for semantic segmentation in aerial images,” IEEE Transactions on Geoscience and Remote Sensing, vol. 60, pp. 1–18, 2021.
- “Fast building segmentation from satellite imagery and few local labels,” in Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2022, pp. 1463–1471.
- “Segment anything,” arXiv preprint arXiv:2304.02643, 2023.
- “Mining self-similarity: Label super-resolution with epitomic representations,” in Computer Vision–ECCV 2020: 16th European Conference, Glasgow, UK, August 23–28, 2020, Proceedings, Part XXVI 16. Springer, 2020, pp. 531–547.
- “Imagenet-trained cnns are biased towards texture; increasing shape bias improves accuracy and robustness,” arXiv preprint arXiv:1811.12231, 2018.
- “Local features and global shape information in object classification by deep convolutional neural networks,” Vision research, vol. 172, pp. 46–61, 2020.
- “Approximating CNNs with bag-of-local-features models works surprisingly well on imagenet,” in International Conference on Learning Representations, 2019.
- “Show why the answer is correct! towards explainable ai using compositional temporal attention,” in 2021 IEEE International Conference on Systems, Man, and Cybernetics (SMC). IEEE, 2021, pp. 3006–3012.
- “Mega: moving average equipped gated attention,” arXiv preprint arXiv:2209.10655, 2022.
- “Megabyte: Predicting million-byte sequences with multiscale transformers,” arXiv preprint arXiv:2305.07185, 2023.
- “Resurrecting recurrent neural networks for long sequences,” arXiv preprint arXiv:2303.06349, 2023.
- “Pretraining without attention,” arXiv preprint arXiv:2212.10544, 2022.
- “Efficiently modeling long sequences with structured state spaces,” arXiv preprint arXiv:2111.00396, 2021.
- “On the parameterization and initialization of diagonal state space models,” Advances in Neural Information Processing Systems, vol. 35, pp. 35971–35983, 2022.
- “What makes convolutional models great on long sequence modeling?,” arXiv preprint arXiv:2210.09298, 2022.
- “Simplified state space layers for sequence modeling,” arXiv preprint arXiv:2208.04933, 2022.
- “Long range language modeling via gated state spaces,” arXiv preprint arXiv:2206.13947, 2022.
- “Hungry hungry hippos: Towards language modeling with state space models,” arXiv preprint arXiv:2212.14052, 2022.
- “S4nd: Modeling images and videos as multidimensional signals with state spaces,” Advances in neural information processing systems, vol. 35, pp. 2846–2861, 2022.
- “Scalable diffusion models with transformers,” in Proceedings of the IEEE/CVF International Conference on Computer Vision, 2023, pp. 4195–4205.
- “Diffusion models without attention,” 2023.
- “A technique for detecting burn scars using modis data,” IEEE Transactions on Geoscience and Remote Sensing, vol. 42, no. 6, pp. 1300–1308, 2004.
- “Roadtracer: Automatic extraction of road networks from aerial images,” in Proceedings of the IEEE conference on computer vision and pattern recognition, 2018, pp. 4720–4728.
- “Beyond road extraction: A dataset for map update using aerial images,” in Proceedings of the IEEE/CVF International Conference on Computer Vision, 2021, pp. 11905–11914.
- “Single-shot end-to-end road graph extraction,” in Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2022, pp. 1403–1412.
- “Torchgeo: deep learning with geospatial data,” in Proceedings of the 30th international conference on advances in geographic information systems, 2022, pp. 1–12.
- Chesapeake Bay Program, “Chesapeake Bay Land Use and Land Cover Database 2022 Edition: Land Use/Land Cover,” U.S. Geological Survey data release, 2023, Developed by the Chesapeake Conservancy, U.S. Geological Survey and University of Vermont Spatial Analysis Lab.
- “Understanding the effective receptive field in deep convolutional neural networks,” Advances in neural information processing systems, vol. 29, 2016.
- “U-net: Convolutional networks for biomedical image segmentation,” in Medical Image Computing and Computer-Assisted Intervention–MICCAI 2015: 18th International Conference, Munich, Germany, October 5-9, 2015, Proceedings, Part III 18. Springer, 2015, pp. 234–241.
- Pavel Iakubovskii, “Segmentation models pytorch,” https://github.com/qubvel/segmentation_models.pytorch, 2019.
- “Encoder-decoder with atrous separable convolution for semantic image segmentation,” in Proceedings of the European conference on computer vision (ECCV), 2018, pp. 801–818.
- “Decoupled weight decay regularization,” arXiv preprint arXiv:1711.05101, 2017.