Bayesian adaptively-weighted ensembles for few-shot abdominal segmentation

• URL: http://arxiv.org/abs/2608.05815v1
• Date: Thu, 06 Aug 2026 09:46:57 GMT
• Title: Bayesian adaptively-weighted ensembles for few-shot abdominal segmentation
• Abstract summary: Few-shot learning has emerged as a promising approach for anatomical segmentation when labelled data are scarce.<n>We propose a Bayesian adaptively-weighted ensemble framework for segmentation under label scarcity and domain shift.<n>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.
• Score: 1.7804186825965627
• License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
• 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.

Related papers

• Active few-shot segmentation by reinforcing data selection [1.7289593940332797]
  Few-shot learning enables medical image segmentation models to adapt to new tasks using only a small number of labelled examples.<n>Support sets should capture relevant variation within the target domain and be informative for adaptation.<n>We propose a reinforcement learning framework for support-set selection in few-shot medical image segmentation.
  arXiv  Detail & Related papers  (2026-07-24T14:58:35Z)
• Leveraging Causal Reasoning Method for Explaining Medical Image Segmentation Models [15.976622378615714]
  Medical image segmentation plays a vital role in clinical decision-making, enabling precise localization of lesions and guiding interventions.<n>Current explanation techniques have primarily focused on classification tasks, leaving the segmentation domain relatively underexplored.<n>We introduce an explanation model for segmentation task which employs the causal inference framework and backpropagates the average treatment effect (ATE) into a metric to determine the influence of input regions, as well as network components, on target segmentation areas.
  arXiv  Detail & Related papers  (2026-02-24T03:26:27Z)
• Adapting In-Domain Few-Shot Segmentation to New Domains without Retraining [53.963279865355105]
  Cross-domain few-shot segmentation (CD-FSS) aims to segment objects of novel classes in new domains.<n>Most CD-FSS methods redesign and retrain in-domain FSS models using various domain-generalization techniques.<n>We propose adapting informative model structures of the well-trained FSS model for target domains by learning domain characteristics from few-shot labeled support samples.
  arXiv  Detail & Related papers  (2025-04-30T08:16:33Z)
• Optimized Vessel Segmentation: A Structure-Agnostic Approach with Small Vessel Enhancement and Morphological Correction [7.882674026364302]
  We propose a structure-agnostic approach incorporating small vessel enhancement and morphological correction for multi-modality vessel segmentation. Our approach achieves superior segmentation accuracy, generalization, and a 34.6% improvement in connectivity, underscoring its clinical potential.
  arXiv  Detail & Related papers  (2024-11-22T08:38:30Z)
• Taxonomy Adaptive Cross-Domain Adaptation in Medical Imaging via Optimization Trajectory Distillation [73.83178465971552]
  The success of automated medical image analysis depends on large-scale and expert-annotated training sets. Unsupervised domain adaptation (UDA) has been raised as a promising approach to alleviate the burden of labeled data collection. We propose optimization trajectory distillation, a unified approach to address the two technical challenges from a new perspective.
  arXiv  Detail & Related papers  (2023-07-27T08:58:05Z)
• Towards Better Certified Segmentation via Diffusion Models [62.21617614504225]
  segmentation models can be vulnerable to adversarial perturbations, which hinders their use in critical-decision systems like healthcare or autonomous driving. Recently, randomized smoothing has been proposed to certify segmentation predictions by adding Gaussian noise to the input to obtain theoretical guarantees. In this paper, we address the problem of certifying segmentation prediction using a combination of randomized smoothing and diffusion models.
  arXiv  Detail & Related papers  (2023-06-16T16:30:39Z)
• Prototypical few-shot segmentation for cross-institution male pelvic structures with spatial registration [24.089382725904304]
  This work describes a fully 3D few-shot segmentation algorithm. The trained networks can be effectively adapted to clinically interesting structures that are absent in training. Experiments are presented in an application of segmenting eight anatomical structures important for interventional planning.
  arXiv  Detail & Related papers  (2022-09-12T11:34:57Z)
• f-Domain-Adversarial Learning: Theory and Algorithms [82.97698406515667]
  Unsupervised domain adaptation is used in many machine learning applications where, during training, a model has access to unlabeled data in the target domain. We derive a novel generalization bound for domain adaptation that exploits a new measure of discrepancy between distributions based on a variational characterization of f-divergences.
  arXiv  Detail & Related papers  (2021-06-21T18:21:09Z)
• Multi-Task, Multi-Domain Deep Segmentation with Shared Representations and Contrastive Regularization for Sparse Pediatric Datasets [0.5249805590164902]
  We propose to train a segmentation model on multiple datasets, arising from different parts of the anatomy, in a multi-task and multi-domain learning framework. The proposed segmentation network comprises shared convolutional filters, domain-specific batch normalization parameters that compute the respective dataset statistics. We evaluate our contributions on two pediatric imaging datasets of the ankle and shoulder joints for bone segmentation.
  arXiv  Detail & Related papers  (2021-05-21T12:26:05Z)
• Self-supervised Augmentation Consistency for Adapting Semantic Segmentation [56.91850268635183]
  We propose an approach to domain adaptation for semantic segmentation that is both practical and highly accurate. We employ standard data augmentation techniques $-$ photometric noise, flipping and scaling $-$ and ensure consistency of the semantic predictions. We achieve significant improvements of the state-of-the-art segmentation accuracy after adaptation, consistent both across different choices of the backbone architecture and adaptation scenarios.
  arXiv  Detail & Related papers  (2021-04-30T21:32:40Z)
• Shape-aware Meta-learning for Generalizing Prostate MRI Segmentation to Unseen Domains [68.73614619875814]
  We present a novel shape-aware meta-learning scheme to improve the model generalization in prostate MRI segmentation. Experimental results show that our approach outperforms many state-of-the-art generalization methods consistently across all six settings of unseen domains.
  arXiv  Detail & Related papers  (2020-07-04T07:56:02Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.