Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

PIMSM: Physics-Informed Multi-Scale Mamba for Stable Neural Representations under Distribution Shift

About

Scientific foundation models are expected to reuse representations under changes in dataset, acquisition protocol, and deployment domain, yet many sequence backbones treat scientific temporal structure as an unconstrained pattern to be fitted. We argue that this misses a central property of natural dynamical systems: neural and atmospheric time series are organized by interacting processes across multiple physical timescales, and failure to preserve this multiscale structure contributes to brittleness under distribution shift. We formalize this failure mode as temporal kernel mismatch, where a model fits in-distribution dynamics with an effective memory policy that is not anchored to the signal's physical timescales, leading to representation drift and degraded transfer. We propose Physics-Informed Multi-Scale Mamba (PIMSM), a state-space architecture that maps spectrum-estimated transition points between frequency regimes (knee frequencies) to scale-specific discretization parameters and anchors them to acquisition time units. On Human Connectome Project fMRI, PIMSM improves robustness and representation stability under severe temporal-context truncation, extreme low-resource transfer, and resting-state-to-task-state generalization. Without modality-specific adaptation, the same architecture also attains the lowest variable-wise MAE across all reported horizons and variables on Weather-5K held-out-station spatial out-of-distribution forecasting. These results support temporal-scale alignment as a practical inductive bias for scientific foundation models that must preserve structure, not only fit correlations, under deployment shift.

Sangyoon Bae, Shinjae Yoo, Jiook Cha• 2026

Related benchmarks

TaskDatasetResultRank
Motor DecodingHCP motor decoding
Accuracy98.7
16
Motor DecodingHCP Full block
Accuracy98.7
12
Motor DecodingHCP 2 TRs truncation
Accuracy93.3
6
Motor DecodingHCP motor 3 TRs truncation
Accuracy94.4
6
Sex ClassificationHCP motor-task fMRI (trained on resting-state) (test)
Accuracy63.2
4
Spatial Out-of-Distribution ForecastingWeather-5K 24h horizon (Spatial OOD)
Temperature Error (TMP)0.921
4
Spatial Out-of-Distribution ForecastingWeather-5K Spatial OOD 72h horizon
TMP Error1.167
4
Spatial Out-of-Distribution ForecastingWeather-5K Spatial OOD 120h horizon
Temperature Error (TMP)1.263
4
Spatial Out-of-Distribution ForecastingWeather-5K Spatial OOD 168h horizon
TMP Error1.334
4
Showing 9 of 9 rows

Other info

Follow for update