REVIEW 5 minor 34 references
Align and Segment: Unsupervised Learning for Building Segmentation From Misaligned Labels
T0 review · 0 major / 5 minor · reviewed 2026-07-14 · grok-4.5
Pith's one-line read Jointly learning an affine alignment of misaligned labels and a building segmenter yields high-quality footprints without any clean ground-truth labels.
desk verdict Clean joint unsupervised alignment+segmentation for OSM-style building labels; works without golden labels and beats the unsupervised baselines. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
The TNet–SNet pair: the TNet predicts an affine matrix that realigns the given noisy mask to the SNet’s prediction; a consistency loss that applies a known random affine warp to the noisy mask and requires the TNet to recover its inverse supplies an absolute learning signal independent of the SNet.
What would settle it
Train the identical architecture on a dataset whose misalignments contain large non-affine local warps or systematic missing-label noise; if IoU_seg and IoU_align both collapse relative to the affine-only case, the central claim fails.
Extended reading notes
Core claim
An end-to-end “align and segment” architecture that couples a spatial-transformer network to any semantic segmentation backbone can learn both a corrective affine transform and a clean building mask from misaligned labels alone, provided a self-supervised consistency loss and strong geometric augmentations are used to block shortcut learning of the original misalignment.
Load-bearing premise
All misalignment inside each image patch can be captured by a single global rotation-plus-translation; local deformations, missing buildings, or non-affine distortions are treated as residual noise that the method does not model.
Signed reviews
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper proposes Align and Segment (AnS), a model-agnostic framework that jointly learns an affine label alignment (via a lightweight TNet operating on pairs of segmentation maps) and a semantic segmentation network (SNet) from misaligned image–label pairs, without any golden labels. It combines a cross-entropy loss on the realigned masks with a self-supervised consistency loss L_Con (MSE on transformation matrices plus IoU on re-warped masks under known random affines) and geometric data augmentations (flips and 90° rotations) to prevent the SNet from simply memorizing the systematic bias. Experiments on synthetic D_uni/D_bias variants of SpaceNet-2 (Las Vegas, Paris, Khartoum), the multi-city ReBO dataset, and real OpenStreetMap labels over San Juan show that AnS recovers both high IoU_seg and IoU_align, outperforming unsupervised baselines (SC, MA, AC) while remaining competitive with supervised ones (MR, ACN).
Significance. If the claims hold, AnS removes a major practical bottleneck in remote-sensing building segmentation: the inability to train directly on abundant but misaligned open sources such as OSM or Google Open Buildings. The method is plug-and-play with any SNet, requires no golden labels at training time, ships code and derived datasets, and is stress-tested across noise magnitudes (Fig. 4), cities, and real misalignments (Table 3). These properties make the contribution immediately usable for large-scale mapping in data-scarce regions and for other multi-modal registration-plus-segmentation tasks that share the same affine-bias pathology.
minor comments (5)
- Table 2 and Table 3 report point estimates only; adding standard deviations over a few random seeds (or at least noting that the ranking is stable) would strengthen the robustness claims already supported by Fig. 4.
- Section 3.1 / Eq. (1): the three-parameter affine is clearly scoped, yet a short sentence quantifying residual non-affine error on ReBO (e.g., mean residual after optimal affine) would help readers judge how much of the remaining IoU gap is attributable to the modeling choice versus label noise.
- Figure 3 and the qualitative OSM examples: the predicted Îy masks are visibly blurry; a brief remark on whether a simple morphological post-process or a boundary-aware loss would tighten the edges without harming alignment would be useful.
- Appendix A.1 Table 4: the freezing-vs-finetuning comparison is informative; moving a one-sentence summary of the main finding (frozen DINOv3-ConvNeXt best) into the main text would improve self-containment.
- Minor notation: the homogeneous-coordinate convention and the exclusion of non-overlapping border pixels from the losses are stated clearly, but a single clarifying sentence near Eq. (2) would prevent any ambiguity for readers less familiar with STNs.
Circularity Check
No significant circularity; empirical method paper whose claims are validated against held-out golden labels never used in training or selection.
full rationale
This is a standard empirical computer-vision methods paper proposing AnS (TNet + SNet) for joint affine alignment and building segmentation from misaligned labels. The training objective is the ordinary cross-entropy L_CE between the TNet-realigned mask and the SNet prediction, regularized by an absolute-consistency loss L_Con (Eq. 3) that applies a known random affine transform to the misaligned label and measures both matrix MSE and IoU against the inverse; geometric flips/rotations further debias systematic shifts. Model selection uses only the internal IoUlearn (overlap of realigned label with SNet output). All reported claims of high-quality segmentation and alignment (IoUseg, IoUalign in Tables 2–3 and Fig. 4) are computed against held-out golden labels that are never seen during training or selection (explicitly stated in Sec. 5). No equation reduces a claimed prediction to a fitted free parameter by construction; there is no uniqueness theorem, no load-bearing self-citation chain, and no renaming of a known empirical pattern. The single-global-affine-per-patch modeling choice (Eq. 1) is an explicit limitation (Sec. 7), not a circular derivation. The paper is therefore self-contained against its external benchmarks.
Assumptions & free parameters
free parameters (4)
- c (tanh scaling) =
0.35
- λ (MSE weight in LCon) =
100
- learning rate / epochs / batch size =
1e-5 / 300 / 48
- augmentation probability =
0.5
assumptions (3)
- domain assumption Image–label misalignment is exactly an unknown affine transform A_θ* (rotation + translation) per patch.
- domain assumption A frozen DINOv3-pretrained ConvNeXt-Tiny encoder supplies sufficiently general features for building segmentation.
- ad hoc to paper Random geometric flips and 90° rotations of image–mask pairs remove systematic translation bias in expectation.
invented entities (2)
-
TNet (transformation network operating on two segmentation maps)
-
LCon consistency loss (MSE on matrices + IoU on re-warped masks)
Cite this review
Pith. "Pith review of Align and Segment: Unsupervised Learning for Building Segmentation From Misaligned Labels." pith.science (2026). https://pith.science/paper/Q3UR6CD3
@misc{pith2026260710841,
author = {Pith},
title = {Pith review of: Align and Segment: Unsupervised Learning for Building Segmentation From Misaligned Labels},
year = {2026},
howpublished = {\url{https://pith.science/paper/Q3UR6CD3}},
note = {Machine review of arXiv:2607.10841}
}
read the original abstract
Supervised learning for image segmentation typically requires spatially aligned image and label sets. When images and labels originate from different sources, the pairing may be misaligned, which can significantly deteriorate the performance of the learned models. This is especially common in remote sensing, when aerial or satellite images are co-registered with labels from another source (e.g., OpenStreetMap). In this work, we propose a novel approach for training on misaligned labels, where we simultaneously learn the label alignment. Our align and segment (AnS) approach builds on the spatial transformer module to transform the misaligned labels using an affine transformation to provide a better learning target for a canonical semantic segmentation network. We prevent shortcut learning of misaligned labels in these semantic segmentation networks through a self-supervised regularization loss and show that it is complementary to data augmentation, especially for systematically misaligned training data. A decisive characteristic of our AnS approach is that it learns without requiring any golden labels. We experimentally show on both synthetic and real-world data from different cities that our approach enables high-quality building segmentation and precise label-image alignment at the same time. Code and derived datasets are available at https://github.com/venkanna37/align-and-segment
Figures
Figures from the paper (12 more)
Reference graph
Works this paper leans on
-
[1]
Nature645(8080), 399–406 (2025)
Bettencourt, L.M.A., Marchio, N.: Infrastructure deficits and informal settlements in sub-Saharan Africa. Nature645(8080), 399–406 (2025)
2025
-
[2]
figshare (2016)
Bradbury, K., Brigman, B., Collins, L., Johnson, T., Lin, S., Newell, R., Park, S., Suresh, S., Wiesner, H., Xi, Y.: Aerial imagery object identification dataset for building and road detection, and building height estimation. figshare (2016)
2016
-
[3]
In: British Machine Vision Conference (2019)
Chen, H., Xie, W., Vedaldi, A., Zisserman, A.: AutoCorrect: Deep inductive align- ment of noisy geometric annotations. In: British Machine Vision Conference (2019)
2019
-
[4]
Corporation, M.: Microsoft global ML building footprints.https : //planetarycomputer.microsoft.com/dataset/ms-buildings(2023), accessed: 2025-11-15
2023
-
[5]
In: International Conference on Learning Representations (ICLR) (2021)
Dosovitskiy, A., Beyer, L., Kolesnikov, A., Weissenborn, D., Zhai, X., Unterthiner, T., Dehghani, M., Minderer, M., Heigold, G., Gelly, S., Uszkoreit, J., Houlsby, N.: An image is worth 16x16 words: Transformers for image recognition at scale. In: International Conference on Learning Representations (ICLR) (2021)
2021
-
[6]
Etten,A.V.,Lindenbaum,D.,Bacastow,T.M.:SpaceNet:Aremotesensingdataset and challenge series (2019)
2019
-
[7]
In: ACM Conference on Computing and Sustainable Societies (SIGCAS)
Fobi, S., Conlon, T., Taneja, J., Modi, V.: Learning to segment from misaligned and partial labels. In: ACM Conference on Computing and Sustainable Societies (SIGCAS). p. 286–290 (2020)
2020
-
[8]
In: Asian Conference on Computer Vision (ACCV)
Girard, N., Charpiat, G., Tarabalka, Y.: Aligning and updating cadaster maps with aerial images by multi-task, multi-resolution deep learning. In: Asian Conference on Computer Vision (ACCV). pp. 675–690 (2019)
2019
Show all 34 references
-
[9]
In: International Geoscience and Remote Sensing Symposium (IGARSS)
Girard,N.,Charpiat,G.,Tarabalka,Y.:Noisysupervisionforcorrectingmisaligned cadaster maps without perfect ground truth data. In: International Geoscience and Remote Sensing Symposium (IGARSS). pp. 10103–10106 (2019)
2019
-
[10]
In: Computer Vision and Pattern Recognition (CVPR)
Girard, N., Smirnov, D., Solomon, J., Tarabalka, Y.: Polygonal building extraction by frame field learning. In: Computer Vision and Pattern Recognition (CVPR). pp. 5891–5900 (2021)
2021
-
[11]
Science of Remote Sensing p
Guthula, V.B., Oehmcke, S., Chilaule, R., Zhang, H., Lang, N., Kariryaa, A., Mot- telson, J., Igel, C.: Drone imagery for roof detection, classification, and segmenta- tion to support mosquito-borne disease risk assessment: The Nacala-roof-material dataset. Science of Remote S...
2025
-
[12]
In: Computer Vision and Pattern Recognition (CVPR)
He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. In: Computer Vision and Pattern Recognition (CVPR). pp. 770–778 (2016)
2016
-
[13]
Hendrycks, D., Gimpel, K.: Gaussian error linear units (GELUs) (2016)
2016
-
[14]
In: Advances in Neural Information Processing Systems (NeurIPS)
Jaderberg, M., Simonyan, K., Zisserman, A., kavukcuoglu, k.: Spatial transformer networks. In: Advances in Neural Information Processing Systems (NeurIPS). vol. 28 (2015)
2015
-
[15]
In: Computer Vision and Pattern Recognition (CVPR) (2019)
Kun, Y., Jianxin, W.: Probabilistic end-to-end noise correction for learning with noisy labels. In: Computer Vision and Pattern Recognition (CVPR) (2019)
2019
-
[16]
Li, K., Weng, X., Deng, Y., Meng, Y., Pang, C., Xia, G.S., Zhao, X.: DragOSM: Extractbuildingroofsandfootprintsfromaerialimagesbyaligninghistoricallabels (2025)
2025
-
[17]
In: Computer Vision and Pat- tern Recognition (CVPR)
Liu, S., Liu, K., Zhu, W., Shen, Y., Fernandez-Granda, C.: Adaptive early-learning correction for segmentation from noisy annotations. In: Computer Vision and Pat- tern Recognition (CVPR). pp. 2606–2616 (2022)
2022
-
[18]
In: Advances in Neural Informa- tion Processing Systems (NeurIPS)
Liu, S., Niles-Weed, J., Razavian, N., Fernandez-Granda, C.: Early-learning regu- larization prevents memorization of noisy labels. In: Advances in Neural Informa- tion Processing Systems (NeurIPS). vol. 33 (2020)
2020
-
[19]
Liu, Z., Mao, H., Wu, C.Y., Feichtenhofer, C., Darrell, T., Xie, S.: A ConvNet for the2020s.In:ComputerVisionandPatternRecognition(CVPR).pp.11976–11986 (2022)
2022
-
[20]
In: International Conference on Learning Representations (ICLR) (2019)
Loshchilov, I., Hutter, F.: Decoupled weight decay regularization. In: International Conference on Learning Representations (ICLR) (2019)
2019
-
[21]
In: Inter- national Geoscience and Remote Sensing Symposium (IGARSS)
Maggiori, E., Tarabalka, Y., Charpiat, G., Alliez, P.: Can semantic labeling meth- ods generalize to any city? the inria aerial image labeling benchmark. In: Inter- national Geoscience and Remote Sensing Symposium (IGARSS). pp. 3226–3229 (2017)
2017
-
[22]
ISPRS Annals of the Photogrammetry, Remote Sensing and Spatial Information Sciences2, 275–282 (2022)
Maiti, A., Oude Elberink, S., Vosselman, G.: Effect of label noise in semantic segmentation of high resolution aerial images and height data. ISPRS Annals of the Photogrammetry, Remote Sensing and Spatial Information Sciences2, 275–282 (2022)
2022
-
[23]
org.https://www.openstreetmap.org(2017)
OpenStreetMap contributors: Planet dump retrieved fromhttps://planet.osm. org.https://www.openstreetmap.org(2017)
2017
-
[24]
In: Medical Image Computing and Computer Assisted Intervention (MICCAI)
Ronneberger, O., Fischer, P., Brox, T.: U-Net: Convolutional networks for biomed- ical image segmentation. In: Medical Image Computing and Computer Assisted Intervention (MICCAI). pp. 234–241. Springer (2015)
2015
-
[25]
International Journal of Computer Vision (IJCV)115(3), 211–252 (2015)
Russakovsky, O., Deng, J., Su, H., Krause, J., Satheesh, S., Ma, S., Huang, Z., Karpathy, A., Khosla, A., Bernstein, M., Berg, A.C., Fei-Fei, L.: ImageNet Large Scale Visual Recognition Challenge. International Journal of Computer Vision (IJCV)115(3), 211–252 (2015)
2015
-
[26]
Siméoni, O., Vo, H.V., Seitzer, M., Baldassarre, F., Oquab, M., Jose, C., Khalidov, V., Szafraniec, M., Yi, S., Ramamonjisoa, M., Massa, F., Haziza, D., Wehrstedt, L., Wang, J., Darcet, T., Moutakanni, T., Sentana, L., Roberts, C., Vedaldi, A., Tolan, J., Brandt, J., Couprie, ...
2025
-
[27]
Sirko, W., Kashubin, S., Ritter, M., Annkah, A., Bouchareb, Y.S.E., Dauphin, Y., Keysers, D., Neumann, M., Cisse, M., Quinn, J.: Continental-scale building detection from high resolution satellite imagery (2021)
2021
-
[28]
In: Advances in Neural Processing Systems (NeurIPS)
Vaswani,A.,Shazeer,N.,Parmar,N.,Uszkoreit,J.,Jones,L.,Gomez,A.N.,Kaiser, Ł., Polosukhin, I.: Attention is all you need. In: Advances in Neural Processing Systems (NeurIPS). vol. 30 (2017) Align and Segment 17
2017
-
[30]
In: International Conference on Learning Representations (ICLR) (2023)
Yao, J., Zhang, Y., Zheng, S., Goswami, M., Prasanna, P., Chen, C.: Learning to segment from noisy annotations: A spatial correction approach. In: International Conference on Learning Representations (ICLR) (2023)
2023
-
[31]
In: European Conference on Computer Vision (ECCV) (2018)
Zampieri,A.,Charpiat,G.,Girard,N.,Tarabalka,Y.:Multimodalimagealignment through a multiscale chain of neural networks with application to remote sensing. In: European Conference on Computer Vision (ECCV) (2018)
2018
-
[32]
In: Medical Image Computing and Computer Assisted Intervention (MICCAI)
Zhang, M., Gao, J., Lyu, Z., Zhao, W., Wang, Q., Ding, W., Wang, S., Li, Z., Cui, S.: Characterizing label errors: Confident learning for noisy-labeled image segmentation. In: Medical Image Computing and Computer Assisted Intervention (MICCAI). pp. 721–730 (2020)
2020
-
[33]
In: Advances in Neural Information Processing Systems (NeurIPS)
Zhang, Z., Sabuncu, M.R.: Generalized cross entropy loss for training deep neural networks with noisy labels. In: Advances in Neural Information Processing Systems (NeurIPS). pp. 8792–8802 (2018)
2018
-
[34]
In: Medical Image Computing and Computer Assisted Intervention (MICCAI)
Zhu, H., Shi, J., Wu, J.: Pick-and-Learn: Automatic quality evaluation for noisy- labeled image segmentation. In: Medical Image Computing and Computer Assisted Intervention (MICCAI). pp. 576–584 (2019)
2019
-
[35]
In: International Geoscience and Remote Sensing Symposium (IGARSS)
Zorzi, S., Bittner, K., Fraundorfer, F.: MapRepair: Deep cadastre maps alignment and temporal inconsistencies fix in satellite images. In: International Geoscience and Remote Sensing Symposium (IGARSS). pp. 1829–1832 (2020) 18 Guthula et al. A Appendix In the following, we pro...
2020
Reviewed July 14, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.