Fig. 4: Influence of the Inter-epoch ViT.
From: Explainable vision transformer for automatic visual sleep staging on multimodal PSG signals
Visualizations demonstrate the influence and role of the Inter-epoch ViT. a Illustration of the inference process and explainability analysis for the Inter-epoch ViT. The color scale bar represents the relevance to the target epoch, with higher values indicating greater relevance. b For the target epoch (the 321st), this figure shows the ground truth and the prediction results of the Intra-epoch ViT, alongside corresponding input images with overlaid heatmaps from epochs 312th to 330th of a patient. c Correction of predictions by the Inter-epoch ViT, which accurately classifies the 321st epoch as REM by aggregating influences from adjacent REM stages, specifically referencing two subsequent and one preceding REM epoch.
