arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2501.13643eess.IVcs.CV

通过几何和光度变换增强医学图像分析

Enhancing Medical Image Analysis through Geometric and Photometric transformations

Echahid Cheikh Larbi Tebessi University(谢赫拉尔比特贝西大学)

Khadija Rais, Mohamed Amroune, Mohamed Yassine Haouam, Abdelmadjid Benmachiche

AI总结:

本文通过几何和光度变换增强技术提升医学图像分类和分割性能,实验表明数据增强有效提高模型准确率和Dice系数。

摘要:

医学图像分析由于患者隐私和专家缺乏等原因导致标注数据不足。尽管一些AI模型在大量数据下表现良好,我们转向数据增强,通过传统或先进技术提高模型性能并扩大数据集规模。本文评估了数据增强技术在两个不同医学图像数据集上的有效性。第一步,我们对包含良性与恶性类别的皮肤癌数据集应用了一些变换技术。然后,我们在增强前后的数据集上训练卷积神经网络(CNN),显著提高了测试准确率从90.74%提升到96.88%,并将测试损失从0.7921降低到0.1468。第二步,我们通过混合两个随机图像及其对应的掩码(使用视网膜和血管数据集)应用Mixup技术,然后训练U-Net模型,得到Dice系数从增强前的0提升到增强后的0.4163。结果表明,使用数据增强扩大数据集规模对分类和分割性能有积极影响。

英文摘要:

Medical image analysis suffers from a lack of labeled data due to several challenges including patient privacy and lack of experts. Although some AI models only perform well with large amounts of data, we will move to data augmentation where there is a solution to improve the performance of our models and increase the dataset size through traditional or advanced techniques. In this paper, we evaluate the effectiveness of data augmentation techniques on two different medical image datasets. In the first step, we applied some transformation techniques to the skin cancer dataset containing benign and malignant classes. Then, we trained the convolutional neural network (CNN) on the dataset before and after augmentation, which significantly improved test accuracy from 90.74% to 96.88% and decreased test loss from 0.7921 to 0.1468 after augmentation. In the second step, we used the Mixup technique by mixing two random images and their corresponding masks using the retina and blood vessels dataset, then we trained the U-net model and obtained the Dice coefficient which increased from 0 before augmentation to 0.4163 after augmentation. The result shows the effect of using data augmentation to increase the dataset size on the classification and segmentation performance.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑