GeoFormer: A Multi-Polygon Segmentation Transformer


Maxim Khomiakov (Technical University of Denmark), Michael Riis Andersen (Technical University of Denmark), Jes Frellsen (Technical University of Denmark)
The 35th British Machine Vision Conference
PDFPosterVideo (Right click to download)Supplementary

Abstract

In remote sensing, there exists a common need for learning scale invariant shapes of objects like buildings. Prior works rely on tweaking multiple loss functions to convert segmentation maps into the final vectorised representation, necessitating arduous design and optimisation. For this purpose, we introduce the GeoFormer, a novel architecture that presents a remedy to the said challenges, learning to generate multi-polygons end-to-end. By modelling keypoints as spatially dependent tokens in an auto-regressive manner, the GeoFormer outperforms existing works in delineating building objects from satellite imagery. We evaluate the robustness of the GeoFormer against former methods through a variety of parameter ablations and highlight the advantages of optimising a single likelihood function. Our study presents the first successful application of auto-regressive transformer models for multi-polygon predictions in remote sensing, suggesting a promising methodological alternative for building vectorisation.

Citation

@inproceedings{Khomiakov_2024_BMVC,
author    = {Maxim Khomiakov and Michael Riis Andersen and Jes Frellsen},
title     = {GeoFormer:  A Multi-Polygon Segmentation Transformer},
booktitle = {35th British Machine Vision Conference 2024, {BMVC} 2024, Glasgow, UK, November 25-28, 2024},
publisher = {BMVA},
year      = {2024},
url       = {https://papers.bmvc2024.org/0217.pdf}
}


Copyright © 2024 The British Machine Vision Association and Society for Pattern Recognition
The British Machine Vision Conference is organised by The British Machine Vision Association and Society for Pattern Recognition. The Association is a Company limited by guarantee, No.2543446, and a non-profit-making body, registered in England and Wales as Charity No.1002307 (Registered Office: Dept. of Computer Science, Durham University, South Road, Durham, DH1 3LE, UK).

Imprint | Data Protection