Current browse context:
eess.IV
Change to browse by:
References & Citations
Electrical Engineering and Systems Science > Image and Video Processing
Title: COVID-19 Pneumonia Severity Prediction using Hybrid Convolution-Attention Neural Architectures
(Submitted on 6 Jul 2021 (v1), last revised 7 Jul 2021 (this version, v2))
Abstract: This study proposed a novel framework for COVID-19 severity prediction, which is a combination of data-centric and model-centric approaches. First, we propose a data-centric pre-training for extremely scare data scenarios of the investigating dataset. Second, we propose two hybrid convolution-attention neural architectures that leverage the self-attention from the Transformer and the Dense Associative Memory (Modern Hopfield networks). Our proposed approach achieves significant improvement from the conventional baseline approach. The best model from our proposed approach achieves $R^2 = 0.85 \pm 0.05$ and Pearson correlation coefficient $\rho = 0.92 \pm 0.02$ in geographic extend and $R^2 = 0.72 \pm 0.09, \rho = 0.85\pm 0.06$ in opacity prediction.
Submission history
From: Nam Nguyen [view email][v1] Tue, 6 Jul 2021 15:26:07 GMT (802kb,D)
[v2] Wed, 7 Jul 2021 17:59:00 GMT (803kb,D)
Link back to: arXiv, form interface, contact.