对比两种数据增强方法,提升罕见病血关节积血检测准确率
Comparison of Two Augmentation Methods in Improving Detection Accuracy of Hemarthrosis
- 用VGG-16提取超声图像特征,比较真实、合成与传统增强图像的相似性
- 传统增强使模型准确率提升,优于数据合成方法(具体数值未给出)
- 通过Grad-CAM发现性能下降源于领域偏移,适合医学影像算法优化研究者
随着计算能力提升,机器学习模型被广泛应用于医疗影像辅助诊断,如血友病这类罕见病。其诊断瓶颈常因训练数据不足导致模型精度受限。本研究探究了数据合成与传统增强技术在提升模型检测准确率方面的效果。采用预训练VGG-16提取超声图像特征,通过余弦相似度比较真实图像、合成图像及增强图像间的分布差异(真实对真实、合成对合成、真实对不同批次合成、真实对增强图像)。使用EfficientNet-B4模型识别‘血液’图像,并评估两种增强方法的效果。结果表明,合成图像与真实图像平均相似度仅为0.4737,而水平翻转生成的图像与原图更相似。传统增强技术相比数据合成能带来更高模型准确率。结合Grad-CAM可视化分析,发现准确率下降主要源于域偏移问题。总体而言,两种增强方法均能在一定程度上提升模型表现,有助于罕见病的自动化诊断。
原文摘要 · Abstract (English)
With the increase of computing power, machine learning models in medical imaging have been introduced to help in rending medical diagnosis and inspection, like hemophilia, a rare disorder in which blood cannot clot normally. Often, one of the bottlenecks of detecting hemophilia is the lack of data available to train the algorithm to increase the accuracy. As a possible solution, this research investigated whether introducing augmented data by data synthesis or traditional augmentation techniques can improve model accuracy, helping to diagnose the diseases. To tackle this research, features of ultrasound images were extracted by the pre-trained VGG-16, and similarities were compared by cosine similarity measure based on extracted features in different distributions among real images, synthetic images, and augmentation images (Real vs. Real, Syn vs. Syn, Real vs. Different Batches of Syn, Real vs. Augmentation Techniques). Model testing performance was investigated using EffientNet-B4 to recognize "blood" images with two augmentation methods. In addition, a gradient-weighted class activation mapping (Grad-CAM) visualization was used to interpret the unexpected results like loss of accuracy. Synthetic and real images do not show high similarity, with a mean similarity score of 0.4737. Synthetic batch 1 dataset and images by horizontal flip are more similar to the original images. Classic augmentation techniques and data synthesis can improve model accuracy, and data by traditional augmentation techniques have a better performance than synthetic data. In addition, the Grad-CAM heatmap figured out the loss of accuracy is due to a shift in the domain. Overall, this research found that two augmentation methods, data synthesis and traditional augmentation techniques, both can improve accuracy to a certain extent to help to diagnose rare diseases.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。