Code for "Classification of Seismic Events in Mainland China Based on Spectrograms and Model Interpretability"
Description
This repository contains the complete code implementation for "Classification of Seismic Events in Mainland China Based on Spectrograms and Model Interpretability". The upload includes two main components:
1. ResWaveQuake_Classification.zip
Implementation of seismic event classification using ResWaveQuake encoder architecture. The framework performs multi-class classification on seismic waveform data using spectrogram-based analysis with PyTorch Lightning.
Key features:
- ResWaveQuake encoder for feature extraction
- Multi-class classification
- Event-level prediction with majority voting across components
- Automatic class balancing and comprehensive evaluation
- Support for multiple seismic data formats (.mseed, .seed, .SAC, etc.)
- Resume training capability from checkpoints
2. Code for Interpretability Analysis.zip
Comprehensive interpretability analysis tools for understanding model decisions and feature importance in seismic event classification.
Components include:
- Cumulative heatmap analysis experiments
- Occlusion-based interpretability tests
- Single-event interpretation analysis
Dataset Availability
The seismic waveform dataset used in this study has been uploaded to Science Data Bank at https://doi.org/10.57760/sciencedb.29192
Both packages include detailed README files with installation instructions and usage guidelines.