Explainable fetal ultrasound quality assessment with progressive concept bottleneck models
This addresses the challenge of operator-dependent, labor-intensive ultrasound quality assessment for medical diagnostics, offering an explainable and generalizable solution.
The paper tackles the problem of assessing fetal ultrasound scan quality by introducing a hierarchical concept bottleneck model that mimics expert decision-making, resulting in improved performance over concept-free models and better generalization across diverse datasets without fine-tuning.
The quality of fetal ultrasound screening scans directly influences the precision of biometric measurements. However, acquiring high-quality scans is labor-intensive and highly relies on the operator's skills. Considering the low contrastiveness and imaging artifacts that widely exist in ultrasound, even a dedicated deep-learning model can be vulnerable to learning from confounding information in the image. In this paper, we propose a holistic and explainable method for fetal ultrasound quality assessment, where we design a hierarchical concept bottleneck model by introducing human-readable ``concepts" into the task and imitating the sequential expert decision-making process. This hierarchical information flow forces the model to learn concepts from semantically meaningful areas: The model first passes through a layer of visual, segmentation-based concepts, and next a second layer of property concepts directly associated with the decision-making task. We consider the quality assessment to be in a more challenging but more realistic setting, with fine-grained image recognition. Experiments show that our model outperforms equivalent concept-free models on an in-house dataset, and shows better generalizability on two public benchmarks, one from Spain and one from Africa, without any fine-tuning.