Skeleton-Guided Progressive Test-Time Adaptation for Thin Curvilinear Structures
Organizations: Seoul National University · Seoul National University Hospital · Yonsei University · Korea University
Abstract
Accurate segmentation of thin curvilinear structures is vital for various real-world applications, from vessel analysis to road extraction. Yet their intricate geometry makes even minor pixel-wise errors enough to break the global topology, and this structural fragility turns severe domain shifts into catastrophic failures. The difficulty is most acute under cross-modality gaps, where the imaging process itself differs fundamentally between source and target. While test-time adaptation (TTA) offers a practical source-free remedy, existing methods adapt feature statistics and confidence, neither of which constrains connectivity, and thus degrade under such extreme gaps. To address this, we propose Skeleton-Guided Progressive Test-Time Adaptation (SGP-TTA). Progressive Batch Normalization (ProgBN) shifts normalization from frozen source statistics toward current target estimates under a sample-count schedule, so that the source-target balance follows the stage of adaptation rather than a fixed coefficient. Consensus Skeleton Recall (CSR) then derives a structural target from geometrically aligned multi-view predictions and updates only the BN affine parameters to preserve connected structures. Extensive experiments show that SGP-TTA consistently outperforms existing TTA methods in topological connectivity, with the largest margins under cross-modality shift. The project page is available at https://boa-jang.github.io/SGP-TTA.
Figures & tables
| Method | DRIVE STARE | DRIVE CHASEDB | OCTA3mm ROSE1 | OCTA6mm ROSE1 | ROSE1 OCTA3mm | ROSE1 OCTA6mm | ||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Dice | clDice | Dice | clDice | Dice | clDice | Dice | clDice | Dice | clDice | Dice | clDice | |
| Source | 51.32 | 48.55 | 37.08 | 38.95 | 47.47 | 41.75 | 56.49 | 49.72 | 59.76 | 64.22 | 68.92 | 80.51 |
| TENT ( Wang et al. 2020 ) | 70.96 | 68.66 | 71.60 | 73.51 | 48.47 | 41.65 | 53.77 | 48.79 | 63.17 | 66.07 | 71.71 | 80.89 |
| CoTTA ( Wang et al. 2022b ) | 71.67 | 69.66 | 71.27 | 73.43 | 47.54 | 42.75 | 53.94 | 48.95 | 62.26 | 64.52 | 71.41 | 80.30 |
| SAR ( Niu et al. 2023 ) | 70.96 | 68.69 | 71.37 | 73.22 | 48.55 | 41.73 | 53.84 | 48.87 | 62.17 | 64.44 | 71.36 | 80.26 |
| EATA ( Niu et al. 2022 ) | 71.06 | 68.78 | 71.61 | 73.56 | 48.57 | 43.73 | 53.87 | 49.79 | 62.44 | 65.05 | 71.42 | 80.60 |