LLM2D
一种用于增强糖尿病视网膜病变检测的新型自适应混合焦散-熵损失函数
A Novel Adaptive Hybrid Focal-Entropy Loss for Enhancing Diabetic Retinopathy Detection Using Convolutional Neural Networks
作者: Santhosh Malarvannan, Pandiyaraju V, Shravan Venkatraman, Abeshek A, Priyadarshini B, Kannan A
发布日期: 4/24/2025
arXiv ID: oai:arXiv.org:2411.10843v2

摘要

arXiv:2411.10843v2 更新类型: replace-cross 摘要:糖尿病视网膜病变是全球-leading致盲原因,并且需要精确的AI诊断工具。传统的多类分类中的损失函数,如类别交叉熵(CCE),在类不平衡的情况下会失效,尤其是在具有固有挑战性或重叠类别的情况下,会导致有偏且敏感度较低的模型。由于高严重程度的4级糖尿病视网膜病变等类别的例子数量远少于非常早期阶段如0类的类别数量,因此实现类平衡是关键。为此,我们提出了自适应混合焦点-熵损失,该损失结合了焦点损失和熵损失的思想,并通过自适应加权来关注少数类并突出挑战性样本。应用于糖尿病视网膜病变检测的先进技术AHFE模型显示出了良好的性能提升,表明ResNet50在准确率为99.79%、DenseNet121在98.86%、Xception在98.92%、MobileNetV2在97.84%、InceptionV3在93.62%的情况下表现最佳。这表明AHFE如何促进复杂且不平衡的医学数据集中的AI驱动诊断性能提升。