Automatically segmenting bleeding regions is an essential task for computer-assisted surgery systems. Recent development in deep learning has boosted the performance of medical image segmentation. However, training deep neural networks, in general, requires high-quality pixel-wise annotations, in which such an annotating process is expensive and easy to introduce annotation noise. To address this issue, motivated by the observation that the noise should be data-dependent, we propose an uncertainty-guided label smoothing method instead of using a fixed label smoothing strategy. Aleatoric uncertainty, which accounts for inherent noise such as annotation error, is estimated via an additional branch of deep neural networks. With the help of estimated aleatoric uncertainty, we could guide the spatial label smoothing in a self-adaptive manner. We demonstrated the effectiveness of our proposal by evaluating the Dice coefficient in a private bleeding segmentation dataset. An improvement over the baseline was observed for our proposal.
|