DAM Lab Research Intelligence

Curated AI research papers in Dental and Medical imaging.

CLINICAL

Attention-Based Prototype Calibration for Multi-Rater Few-Shot Medical Image Segmentation

Source: ArXiv Computer Vision Date: 2026-06-15 Score: 9.3/10

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.

Keywords

attentionmedical imagingsegmentationdataset