Improving medical image segmentation in pre-trained U-Nets using Shapley-guided pruning of adaptive skip-connection modules.

Authors

Emil Benedykciuk, Marcin Denkowski, Grzegorz Marcin Wójcik

Published in

Scientific reports. Jun 27, 2026. Epub Jun 27, 2026.

Abstract

Neural architecture search (NAS) can improve medical image segmentation, but its practical use is limited by computational cost and instability of the discovered architectures, particularly when applied to pre-trained models and limited data. We present Shapley-guided pruning as a practical validation-guided extension of retrofit NAS for pre-trained U-Nets. Rather than defining a new NAS family, the method keeps the IAC search space and PC-DARTS-style supernet optimization, while adding iterative pruning driven by Shapley value estimates computed on held-out validation data. By progressively removing low-impact components while preserving learned architecture parameters, the approach reduces search space complexity and improves the reliability of the final discrete architecture. We evaluate the method on four public benchmarks (ACDC, BraTS, KiTS, and AMOS) in a controlled 2D slice-based, [Formula: see text], single-GPU setting. Within this protocol, the proposed approach improves or matches strong baselines in most comparisons, accelerates search by up to four times, and yields more stable operation choices across runs. The findings support Shapley-guided pruning as a practical retrofit-NAS mechanism under resource constraints, without implying direct clinical competitiveness with high-resolution or fully 3D segmentation pipelines.

PMID:
42365103
Bibliographic data and abstract were imported from PubMed on 28 Jun 2026.

Read full publication at:
Please sign in to see all details.

Stats

1-terrible, 9-excellent. How would you rate this publication? Sign in in to submit your rating.

Post a comment

You need to be signed in to post comments. You can sign in here.

Comments

There are no comments yet.