SSS-TT: Self-supervised sequential spatio-temporal transformers with adaptive multimodal fusion for automated neonatal pain assessment
摘要
Accurate neonatal pain assessment is critical for clinical care but remains challenging due to subtle non-verbal expressions and limited labeled data. We introduce SSS-TT (Self-Supervised Sequential Spatio-Temporal Transformers with Cross-Attention Fusion), a novel framework addressing three key limitations: (1) data scarcity through masked autoencoder (MAE) pretraining that reduces annotation requirements by 75% while improving accuracy by 4.9%, (2) inadequate temporal modeling through a cascaded Vision Transformer and Temporal Convolutional Network architecture capturing pain dynamics over 1-second windows, and (3) incomplete multimodal data via cross-attention fusion that maintains 82.9% accuracy with RGB-only input versus 74.2% for robustly-trained static concatenation baselines (trained with identical modality-dropout augmentation). Using ordinal regression with CORAL loss, SSS-TT predicts pain intensity levels (0–3) aligned with clinical NIPS scoring. Evaluated on the iCOPE dataset (1,000 infants, 5-fold cross-validation with strict subject-level splitting to prevent data leakage), our method achieves 84.6% ± 0.7% accuracy and quadratic weighted kappa of 0.82—comparable to the inter-clinician agreement range reported in the literature (