Multi-Spectral Remote Sensing Image Retrieval Using Geospatial Foundation Models (2403.02059v2)
Abstract: Image retrieval enables an efficient search through vast amounts of satellite imagery and returns similar images to a query. Deep learning models can identify images across various semantic concepts without the need for annotations. This work proposes to use Geospatial Foundation Models, like Prithvi, for remote sensing image retrieval with multiple benefits: i) the models encode multi-spectral satellite data and ii) generalize without further fine-tuning. We introduce two datasets to the retrieval task and observe a strong performance: Prithvi processes six bands and achieves a mean Average Precision of 97.62% on BigEarthNet-43 and 44.51% on ForestNet-12, outperforming other RGB-based models. Further, we evaluate three compression methods with binarized embeddings balancing retrieval speed and accuracy. They match the retrieval speed of much shorter hash codes while maintaining the same accuracy as floating-point embeddings but with a 32-fold compression. The code is available at https://github.com/IBM/remote-sensing-image-retrieval.
- “Query-by-shape in meteorological image archives using the point diffusion technique,” IEEE transactions on geoscience and remote sensing, vol. 39, no. 9, pp. 1834–1843, 2001.
- “The potential and uptake of remote sensing in insurance: A review,” Remote Sensing, vol. 6, no. 11, pp. 10888–10912, 2014.
- “Challenges and opportunities of open data in ecology,” Science, vol. 331, no. 6018, pp. 703–705, 2011.
- “Image retrieval from remote sensing big data: A survey,” Information Fusion, vol. 67, pp. 94–115, 2021.
- “Asymmetric hash code learning for remote sensing image retrieval,” IEEE Transactions on Geoscience and Remote Sensing, vol. 60, pp. 1–14, 2022.
- “Metric-learning-based deep hashing network for content-based retrieval of remote sensing images,” IEEE Geoscience and Remote Sensing Letters, vol. 18, no. 2, pp. 226–230, 2020.
- “A novel self-supervised cross-modal image retrieval method in remote sensing,” in 2022 IEEE International Conference on Image Processing (ICIP). IEEE, 2022, pp. 2426–2430.
- “Meta-hashing for remote sensing image retrieval,” IEEE Transactions on Geoscience and Remote Sensing, vol. 60, pp. 1–19, 2022.
- “Foundation models for generalist geospatial artificial intelligence,” arXiv preprint arXiv:2310.18660, 2023.
- “Multi-scale context deep hashing for remote sensing image retrieval,” IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, 2023.
- “Deep hash learning for remote sensing image retrieval,” IEEE Transactions on Geoscience and Remote Sensing, vol. 59, no. 4, pp. 3420–3443, 2020.
- “Deep metric and hash-code learning for content-based retrieval of remote sensing images,” in IGARSS 2018-2018 IEEE International Geoscience and Remote Sensing Symposium. IEEE, 2018, pp. 4539–4542.
- Yi Yang and Shawn Newsam, “Bag-of-visual-words and spatial extensions for land-use classification,” in Proceedings of the 18th SIGSPATIAL international conference on advances in geographic information systems, 2010, pp. 270–279.
- “Aid: A benchmark data set for performance evaluation of aerial scene classification,” IEEE Transactions on Geoscience and Remote Sensing, vol. 55, no. 7, pp. 3965–3981, 2017.
- “Supervised contrastive learning based on fusion of global and local features for remote sensing image retrieval,” IEEE Transactions on Geoscience and Remote Sensing, 2023.
- “Large-scale remote sensing image retrieval by deep hashing neural networks,” IEEE Transactions on Geoscience and Remote Sensing, vol. 56, no. 2, pp. 950–965, 2017.
- “Bigearthnet: A large-scale benchmark archive for remote sensing image understanding,” in IGARSS 2019-2019 IEEE International Geoscience and Remote Sensing Symposium. IEEE, 2019, pp. 5901–5904.
- “Lightweight, pre-trained transformers for remote sensing timeseries,” NeurIPS 2023 Workshop on Tackling Climate Change with Machine Learning: Blending New and Existing Knowledge Systems, 2023.
- “Transductive zero-shot hashing for multilabel image retrieval,” IEEE Transactions on Neural Networks and Learning Systems, vol. 33, no. 4, pp. 1673–1687, 2020.
- “Milvus: A purpose-built vector data management system,” in Proceedings of the 2021 International Conference on Management of Data, 2021, pp. 2614–2627.
- “An image is worth 16x16 words: Transformers for image recognition at scale,” in International Conference on Learning Representations, 2020.
- “Locality-sensitive hashing for finding nearest neighbors in probability distributions,” in Social Media Processing: 6th National Conference, SMP 2017, Beijing, China, September 14-17, 2017, Proceedings. Springer, 2017, pp. 3–15.
- “Earthnets: Empowering ai in earth observation,” arXiv preprint arXiv:2210.04936, 2022.
- “Geo-bench: Toward foundation models for earth monitoring,” Advances in Neural Information Processing Systems, 2023.
- “Forestnet: Classifying drivers of deforestation in indonesia using deep learning on satellite imagery,” NeurIPS 2020 workshop on Tackling Climate Change with Machine Learning, 2020.