Метод классификации диабетической ретинопатии на основе регуляризации с учетом стоимости и EfficientNet

WANG Ming-zhi ,  

MA Zhi-qiang ,  

ZHAO Feng-feng ,  

WANG Yong-jie ,  

GUO Ji-feng ,  

摘要

Диабетическая ретинопатия (ДР) является распространенным осложнением диабета и одной из основных причин слепоты во всем мире. На ранних клинических стадиях ДР трудно обнаружить. В данной работе предлагается метод компьютерной поддержки диагностики на основе сверточных нейронных сетей для автоматической классификации степени тяжести ДР по изображениям глазного дна. Используются различные методы предварительной обработки для улучшения качества входных изображений, а также различные методы увеличения данных для повышения сбалансированности набора данных. Применяется регуляризация с учетом стоимости для расширения стандартной функции потерь классификации, которая налагает разные штрафы в зависимости от степени расхождения между прогнозируемой и истинной степенью. Предварительное обучение проведено на наборе данных ImageNet для внедрения переноса обучения, а полносвязный слой с функцией активации Softmax обеспечивает лучшую производительность модели. Эксперименты на двух наборах данных показали, что модель улучшает взвешенный коэффициент каппа примерно на 5%, а AUC — примерно на 3% по сравнению с последними исследованиями. Введение регуляризации с учетом стоимости в модель EfficientNet позволяет повысить точность классификации диабетической ретинопатии и достичь хороших показателей модели.

关键词

диабетическая ретинопатия;глубокое обучение;регуляризация с учетом стоимости;сверточные нейронные сети;классификация изображений

阅读全文

以上内容由讯飞翻译自动生成,翻译内容仅供参考。对于因使用本网站翻译内容产生的相关后果,本网站不承担任何商业和法律责任。

The above content is generated by Large Model Translation. The translated content is for reference only. We do not assume any commercial or legal responsibilty for any consequences arising from the use of our website