期刊
DIAGNOSTICS
卷 13, 期 7, 页码 -出版社
MDPI
DOI: 10.3390/diagnostics13071329
关键词
COVID-19; deep learning; attention mechanism; mixed loss; NVIDIA Jetson devices
This paper proposes a deep learning-based COVID-19 detection method and evaluates its performance on embedded edge-computing devices. By adding attention module and mixed loss into the original VGG19 model, the method achieves parameter reduction and increased classification accuracy. The experimental results demonstrate that the proposed method is practicable and convenient on a low-cost medical edge-computing terminal.
The rapid spread of coronavirus disease 2019 (COVID-19) has posed enormous challenges to the global public health system. To deal with the COVID-19 pandemic crisis, the more accurate and convenient diagnosis of patients needs to be developed. This paper proposes a deep-learning-based COVID-19 detection method and evaluates its performance on embedded edge-computing devices. By adding an attention module and mixed loss into the original VGG19 model, the method can effectively reduce the parameters of the model and increase the classification accuracy. The improved model was first trained and tested on the PC X86 GPU platform using a large dataset (COVIDx CT-2A) and a medium dataset (integrated CT scan); the weight parameters of the model were reduced by around six times compared to the original model, but it still approximately achieved 98.80%and 97.84% accuracy, outperforming most existing methods. The trained model was subsequently transferred to embedded NVIDIA Jetson devices (TX2, Nano), where it achieved 97% accuracy at a 0.6-1 FPS inference speed using the NVIDIA TensorRT engine. The experimental results demonstrate that the proposed method is practicable and convenient; it can be used on a low-cost medical edge-computing terminal. The source code is available on GitHub for researchers.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据