English

Bayesian adaptively-weighted ensembles for few-shot abdominal segmentation

Computer Vision and Pattern Recognition 2026-08-06 v1

Abstract

Few-shot learning has emerged as a promising approach for anatomical segmentation when labelled data are scarce. However, different few-shot learning algorithms exhibit complementary strengths and weaknesses, with performance varying across anatomical targets and institutions. Existing few-shot segmentation ensembles, that combine predictions from multiple algorithms, typically employ fixed weighting schemes and therefore cannot adjust model contributions according to the target domain. In this work, we propose a Bayesian adaptively-weighted ensemble framework for segmentation under label scarcity and domain shift. Multiple few-shot segmentation algorithms are first adapted using a small labelled support set. Bayesian optimisation is then used to automatically identify ensemble weights that maximise segmentation performance on a target-domain validation set. The learned weights are subsequently fixed and applied to combine predictions on previously unseen query images from the target domain. The proposed framework is evaluated on the Cross-institution Male Pelvic Structures dataset using held-out anatomical structures and institutions to simulate simultaneous label scarcity and institutional domain shift. Results demonstrate statistically significant improvements over individual few-shot learners, fixed-weight ensembles, training-from-scratch baselines and recent state-of-the-art ensembling approaches. By adapting model contributions to the target anatomy and institutional domain, the proposed framework provides a practical mechanism for deploying segmentation systems to new clinical sites under severe annotation constraints.

Cite

@article{arxiv.2608.05815,
  title  = {Bayesian adaptively-weighted ensembles for few-shot abdominal segmentation},
  author = {Abbas Al-Sabbagh and Shalom F. Mushtaq and Tomás M. da Silva and Kushagra Soni and Binawei Gbamila and Sri Atluri and Qianye Yang and Yipeng Hu and Claire C. Villette and Shaheer U. Saeed},
  journal= {arXiv preprint arXiv:2608.05815},
  year   = {2026}
}

Comments

Accepted at DEMI at MICCAI 2026 - The 4th MICCAI Workshop in Data Engineering in Medical Imaging