4.7 Article

Class-Aware Attention Network for infectious keratitis diagnosis using corneal photographs

期刊

COMPUTERS IN BIOLOGY AND MEDICINE
卷 151, 期 -, 页码 -

出版社

PERGAMON-ELSEVIER SCIENCE LTD
DOI: 10.1016/j.compbiomed.2022.106301

关键词

Multi-class classification; Corneal photographs; Infectious keratitis; Attention mechanism; Convolutional neural network

资金

  1. Natural Science Foundation of Shandong Province
  2. Natural Science Foundation of Jiangsu Province
  3. National Natural Science Founda-tion of China
  4. National Natural Science Foundation Regional Innovation and Development Joint Fund
  5. [2022HWYQ-041]
  6. [BK20220266]
  7. [62201323]
  8. [U20A20386]

向作者/读者索取更多资源

This paper proposes a novel method, CAA-Net, for automatically diagnosing infectious keratitis using corneal photographs. The method combines class-awareness and attention strategies, and has been verified to be effective in diagnosing different types of infectious keratitis.
Infectious keratitis is one of the common ophthalmic diseases and also one of the main blinding eye diseases in China, hence rapid and accurate diagnosis and treatment for infectious keratitis are urgent to prevent the progression of the disease and limit the degree of corneal injury. Unfortunately, the traditional manual diagnosis accuracy is usually unsatisfactory due to the indistinguishable visual features. In this paper, we propose a novel end-to-end fully convolutional network, named Class-Aware Attention Network (CAA-Net), for automatically diagnosing infectious keratitis (normal, viral keratitis, fungal keratitis, and bacterial keratitis) using corneal photographs. In CAA-Net, a class-aware classification module is first trained to learn class-related discriminative features using separate branches for each class. Then, the learned class-aware discriminative features are fed into the main branch and fused with other feature maps using two attention strategies to assist the final multi-class classification performance. For the experiments, we have built a new corneal photograph dataset with 1886 images from 519 patients and conducted comprehensive experiments to verify the effectiveness of our proposed method. The code is available at https://github.com/SWF-hao/CAA-Net_ Pytorch.

作者

我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。

评论

主要评分

4.7
评分不足

次要评分

新颖性
-
重要性
-
科学严谨性
-
评价这篇论文

推荐

暂无数据
暂无数据