† Equal Contribution
Corresponding Author: Ali Zia (A.Zia@latrobe.edu.au)
Acknowledgment: This work was supported by La Trobe University, Melbourne, Australia.
Test-time adaptation (TTA) has emerged as a promising paradigm for mitigating distribution shifts in deep models. However, existing TTA approaches for anomaly segmentation remain limited by their reliance on pixel-level heuristics, such as confidence thresholding or entropy minimisation, which fail to preserve structural consistency under noise and texture variation. Moreover, they typically treat anomaly maps as flat intensity fields, ignoring the higher-order spatial relationships that characterise complex defect geometries. We introduce TopoTTA (Topological Test-Time Adaptation), a novel framework that integrates persistent homology, a tool from topological data analysis, into the TTA pipeline to enforce geometric and structural coherence during adaptation. By applying multi-level cubical complex filtration to anomaly score maps, TopoTTA derives robust topological pseudo-labels that guide a lightweight test-time classifier, enhancing segmentation quality without retraining the backbone model. The approach eliminates heuristic thresholding, preserves connectivity, and generalises across both 2D and 3D modalities. Extensive experiments across five standard benchmarks (MVTec AD, VisA, Real-IAD, MVTec 3D-AD, and AnomalyShapeNet) demonstrate an average 15% F1 improvement over state-of-the-art unsupervised anomaly detection and segmentation methods, with the largest gains on anomalies exhibiting complex geometric or structural variations. These findings suggest that integrating topological reasoning into test-time adaptation provides a principled route to structure-aware generalisation, bridging the gap between geometric learning and robust adaptation.
Overview of the TopoTTA architecture. Given a test image I, an AD&S method produces an anomaly score map Ψ. A pre-trained feature extractor F generates dense feature maps from I. Topological pseudo-labels are extracted by applying multi-level cubical complex filtrations (both sublevel and superlevel) to Ψ, producing structurally meaningful binary masks via persistent homology. These masks are fused using IoU to generate sparse pseudo-labels. A lightweight classifier is then trained on selected feature points from F(I) using these labels and applied across the full feature map to produce a refined binary anomaly segmentation. This test-time adaptation pipeline exploits both intensity-based cues and topological structure to improve segmentation robustness and generalisation.
Comparison of binary segmentation results. Best results in bold; second-best in blue.
Qualitative comparison across MVTec AD, VisA, and Real-IAD. Columns: RGB, Ground Truth, anomaly heat map, simple thresholding (THR), TTT4AS, and our TopoTTA. TopoTTA produces sharper, topologically consistent anomaly segmentations across diverse categories and datasets.
Qualitative comparison of AD\&S methods for different objects using on 3D MvTec AD Dataset.