Search papers, labs, and topics across Lattice.
This paper introduces an attention-based prototype calibration framework that addresses the limitations of few-shot medical image segmentation by accounting for variability in expert annotations. By refining rater-specific prototypes without altering the backbone feature extractor, the method maintains semantic consistency while enabling personalized segmentation outputs. Experiments on multi-rater datasets show significant improvements over traditional prototype approaches, underscoring the importance of modeling annotation variability in clinical settings.
Systematic variability in expert annotations can be effectively modeled, leading to personalized segmentation outputs that outperform traditional methods.
Few-shot medical image segmentation methods typically assume a single ground-truth annotation, overlooking systematic variability across expert raters commonly observed in clinical datasets. We propose an attention-based prototype calibration framework for few-shot multi-rater segmentation that models rater-specific deviations from a consensus representation in prototype space. A lightweight yet principled attention operator directly refines rater prototypes without modifying the backbone feature extractor, making the approach fully compatible with existing prototype-based few-shot segmentation methods. This design preserves semantic consistency while enabling personalized segmentation outputs with minimal computational overhead. Experiments on multi-rater medical imaging datasets demonstrate consistent improvements over baseline prototype approaches, highlighting the effectiveness of structured prototype calibration for modeling annotation variability. Our code is available at https://github.com/truong2710-cyber/JAPC.
