Explainable Deep Neural Network for Identifying Cardiac Abnormalities Using Class Activation Map
| Autor: | Yu-Cheng Lin, Wen-Chiao Tsai, An-Yeu Andy Wu, Yun-Chieh Lee, Win-Ken Beh |
|---|---|
| Rok vydání: | 2020 |
| Předmět: |
Normalization (statistics)
Artificial neural network Computer science business.industry 0206 medical engineering Pattern recognition 02 engineering and technology Sigmoid function Overfitting 020601 biomedical engineering Convolutional neural network 03 medical and health sciences 0302 clinical medicine Cross entropy Test set Artificial intelligence Entropy (energy dispersal) business 030217 neurology & neurosurgery |
| Zdroj: | CinC |
| ISSN: | 2325-887X |
| DOI: | 10.22489/cinc.2020.072 |
| Popis: | In this study, our team “NTU-Accesslab” present a deep convolutional neural network (CNN) approach, called CNN-GAP, for classifying 12-lead ECGs with multilabel cardiac abnormalities. Additionally, Class Activation Mapping (CAM) is employed for further understanding the decision-making process of this black-box model, making the model more explainable. The CNN-GAP model consists of 12 layer Conv Blocks along with Batch Normalization layer, Global Average Pooling and Fully Connected layer with sigmoid activation. To deal with the data imbalance problem, we oversample the minor datas. In the training stage, we applied Macro observed score loss (Macro-Obs) instead of the conventional Weighted Cross entropy loss (WCE), and we have shown that this results in higher challenge scores. Additionally, we augmented datas by randomly scaling datas to get better scores and prevent model overfitting. Our method achieved a challenge score of 0.58 on the validation set, but was unable to score and rank on the test set, due to a failure of the algorithm on the fully hidden dataset. |
| Databáze: | OpenAIRE |
| Externí odkaz: |
|