SegTTA: Training-Free Test-Time Augmentation for Zero-Shot Medical Imaging Segmentation

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Yao, Yihong, Li, Chunlei, Gang, Canxuan, Hu, Wenzhi, Zhang, Zeyu, Zhang, Hao, Li, Xiaoyan
Format: Preprint
Publié: 2026
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866914488788189184
author Yao, Yihong
Li, Chunlei
Gang, Canxuan
Hu, Wenzhi
Zhang, Zeyu
Zhang, Hao
Li, Xiaoyan
author_facet Yao, Yihong
Li, Chunlei
Gang, Canxuan
Hu, Wenzhi
Zhang, Zeyu
Zhang, Hao
Li, Xiaoyan
contents Increasingly advanced data augmentation techniques have greatly aided clinical medical research, increasing data diversity and improving model generalization capabilities. Although most current basic models exhibit strong generalization abilities, image quality varies due to differences in equipment and operators. To address these challenges, we present SegTTA, a framework that improves medical image segmentation without model retraining by combining four augmentations (Gamma correction, Contrast enhancement, Gaussian blur, Gaussian noise) with weighted voting across multiple MedSAM2 checkpoints. Experiments demonstrate consistent improvements across three diverse datasets: healthy uterus segmentation, uterine myoma detection, and multi class hepatic structure segmentation. Ablation studies reveal that large organs benefit from intensity augmentations while small lesions require noise augmentations. The voting threshold controls the coverage precision trade off, enabling task specific optimization for different clinical requirements. Ultimately, on a multiclass hepatic vessel dataset, compared to MedSAM2 baselines, our method achieves an increase of 1.6 in mIoU and 1.9 in aIoU, along with a reduction of approximately 2.0 in HD95. Code will be available at https://github.com/AIGeeksGroup/SegTTA.
format Preprint
id arxiv_https___arxiv_org_abs_2604_17451
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle SegTTA: Training-Free Test-Time Augmentation for Zero-Shot Medical Imaging Segmentation
Yao, Yihong
Li, Chunlei
Gang, Canxuan
Hu, Wenzhi
Zhang, Zeyu
Zhang, Hao
Li, Xiaoyan
Computer Vision and Pattern Recognition
Increasingly advanced data augmentation techniques have greatly aided clinical medical research, increasing data diversity and improving model generalization capabilities. Although most current basic models exhibit strong generalization abilities, image quality varies due to differences in equipment and operators. To address these challenges, we present SegTTA, a framework that improves medical image segmentation without model retraining by combining four augmentations (Gamma correction, Contrast enhancement, Gaussian blur, Gaussian noise) with weighted voting across multiple MedSAM2 checkpoints. Experiments demonstrate consistent improvements across three diverse datasets: healthy uterus segmentation, uterine myoma detection, and multi class hepatic structure segmentation. Ablation studies reveal that large organs benefit from intensity augmentations while small lesions require noise augmentations. The voting threshold controls the coverage precision trade off, enabling task specific optimization for different clinical requirements. Ultimately, on a multiclass hepatic vessel dataset, compared to MedSAM2 baselines, our method achieves an increase of 1.6 in mIoU and 1.9 in aIoU, along with a reduction of approximately 2.0 in HD95. Code will be available at https://github.com/AIGeeksGroup/SegTTA.
title SegTTA: Training-Free Test-Time Augmentation for Zero-Shot Medical Imaging Segmentation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2604.17451