cs.CVAug 1, 2026

Test-time Adaptation of Pelvic Bone Segmentation Models via Dynamic Reliability-Guided

Authors: Ling RenChao DengZiming WangYuecong XuKai Zheng

Organizations: College of Automation, Nanjing University of Posts and Telecommunications, Nanjing 210003, China · Department of Electrical and Computer Engineering, National University of Singapore, Singapore 117583 · State Key Laboratory Cultivation Base of Research, Prevention and Treatment for Oral Diseases, the Affiliated Stomatological Hospital of Nanjing Medical University, Nanjing 210029, China

Abstract

Reliable pelvic bone segmentation (PBS) from CT is essential for robot-assisted pelvic trauma surgery, yet deploying a source-trained model to a new hospital suffers from severe performance degradation due to cross-center domain shifts. While test-time adaptation (TTA) enables online model adaptation without accessing source data, existing methods show limited effectiveness for PBS, facing challenges including boundary degradation, anatomical inconsistency under domain shifts, and voxel-level class imbalance. To address these challenges, we propose a novel closed-loop dynamic Reliability-Guided TTA framework (ReGA) for PBS. Specifically, we introduce a pseudo-label reliability criterion termed Segmentation Inference Consistency Evaluation (SICE), which jointly measures region overlap and boundary deviation via dropout-based ensemble predictions. Based on SICE, a trust-weighted refinement module adaptively updates features to mitigate boundary errors in pseudo-labels. Furthermore, a confidence-weighted region-level contrastive learning strategy is proposed to enforce anatomical consistency. Finally, ReGA follows the teacher-student (TS) scheme to alleviate voxel-level class imbalance. Experiments on three heterogeneous 3D pelvic CT datasets demonstrate that ReGA consistently outperforms state-of-the-art TTA methods, enabling effective adaptation of the source-trained PBS model to unseen clinical domains. The code is available at https://github.com/Ren-ling/ReGA.

Explore similar work

CardsList