Interactive medical image segmentation with self-adaptive confidence calibration
Chuyun Shen,Wenhao Li,Qisen Xu,Bin Hu,Bo Jin,Haibin Cai,Fengping Zhu,Yuxin Li,Xiangfeng Wang
DOI: https://doi.org/10.1631/fitee.2200299
IF: 2.526
2023-09-23
Frontiers of Information Technology & Electronic Engineering
Abstract:Interactive medical image segmentation based on human-in-the-loop machine learning is a novel paradigm that draws on human expert knowledge to assist medical image segmentation. However, existing methods often fall into what we call interactive misunderstanding, the essence of which is the dilemma in trading off short- and long-term interaction information. To better use the interaction information at various timescales, we propose an interactive segmentation framework, called interactive MEdical image segmentation with self-adaptive Confidence CAlibration (MECCA), which combines action-based confidence learning and multi-agent reinforcement learning. A novel confidence network is learned by predicting the alignment level of the action with short-term interaction information. A confidence-based reward-shaping mechanism is then proposed to explicitly incorporate confidence in the policy gradient calculation, thus directly correcting the model's interactive misunderstanding. MECCA also enables user-friendly interactions by reducing the interaction intensity and difficulty via label generation and interaction guidance, respectively. Numerical experiments on different segmentation tasks show that MECCA can significantly improve short- and long-term interaction information utilization efficiency with remarkably fewer labeled samples. The demo video is available at https://bit.ly/mecca-demo-video.
engineering, electrical & electronic,computer science, information systems, software engineering