Toward an Interpretable CNN Model for the Classification of Lightning‐Produced VLF/LF Signals

Abstract

AbstractAn interpretable convolutional neural network model is proposed for the classification of very low frequency and low frequency lightning electric field waveforms. This model adopts multi‐scale convolutional kernels and shortcut connections to enhance the ability of lightning waveform classification. Based on the data recorded from five provinces in China, the proposed model achieves an accuracy of 98.56% for a four‐type classification task including return strokes, the intra‐cloud lightning, preliminary breakdown, and narrow bipolar events. The proposed model is validated with another open‐source data set from Argentina with an accuracy of 98.45%, which shows good robustness. To ensure the classification, the features learned by the model are visualized. The class activation mapping (CAM) method is adopted to visualize the class‐specific contribution of different waveform parts by using the feature maps of the final convolutional layer. It is highlighted by the CAM method that the proposed model focuses on waveform parts that align with those areas of interests identified by human experts. The high‐contribution waveform parts are furtherly analyzed, which indicate that the proposed model possesses the capability to associate waveform features with the corresponding lightning discharge processes.