Yao Siqiong, Yan Junchi, Wu Mingyu, Yang Xue, Zhang Weituo, Lu Hui, Qian Biyun
School of Life Sciences and Biotechnology, Shanghai Jiao Tong University, Shanghai, China.
School of Electronic Information and Electrical Engineering, Shanghai Jiao Tong University, Shanghai, China.
Front Bioeng Biotechnol. 2020 Jun 17;8:599. doi: 10.3389/fbioe.2020.00599. eCollection 2020.
Deep learning method have been offering promising solutions for medical image processing, but failing to understand what features in the input image are captured and whether certain artifacts are mistakenly included in the model, thus create crucial problems in generalizability of the model. We targeted a common issue of this kind caused by manual annotations appeared in medical image. These annotations are usually made by the doctors at the spot of medical interest and have adversarial effect on many computer vision AI tasks. We developed an inpainting algorithm to remove the annotations and recover the original images. Besides we applied variational information bottleneck method in order to filter out the unwanted features and enhance the robustness of the model. Our impaiting algorithm is extensively tested in object detection in thyroid ultrasound image data. The mAP (mean average precision, with IoU = 0.3) is 27% without the annotation removal. The mAP is 83% if manually removed the annotations using Photoshop and is enhanced to 90% using our inpainting algorithm. Our work can be utilized in future development and evaluation of artificial intelligence models based on medical images with defects.
深度学习方法一直在为医学图像处理提供有前景的解决方案,但无法理解输入图像中捕获了哪些特征以及某些伪影是否被错误地纳入模型,从而在模型的通用性方面产生了关键问题。我们针对医学图像中出现的由手动标注引起的此类常见问题。这些标注通常由医生在医学关注部位现场进行,并且对许多计算机视觉人工智能任务具有对抗性影响。我们开发了一种修复算法来去除标注并恢复原始图像。此外,我们应用了变分信息瓶颈方法,以过滤掉不需要的特征并增强模型的鲁棒性。我们的修复算法在甲状腺超声图像数据的目标检测中进行了广泛测试。在不进行标注去除的情况下,平均精度均值(mAP,交并比IoU = 0.3)为27%。如果使用Photoshop手动去除标注,mAP为83%,而使用我们的修复算法则提高到90%。我们的工作可用于未来基于有缺陷的医学图像的人工智能模型的开发和评估。