Search papers, labs, and topics across Lattice.
This paper introduces BeatRhythm-TTA, a test-time adaptation framework specifically designed for ECG classification that addresses performance degradation due to domain shifts. By employing a Signal Quality Index (SQI)-gated adaptation scheme, the model selectively filters out low-quality ECG signals, thereby mitigating the risk of harmful updates during adaptation. The method also enforces dual-level consistency to maintain the integrity of beat morphology and rhythm dynamics, achieving a notable average improvement of +2.70% in Macro-F1 score across multiple adaptation protocols compared to existing methods.
Adapting ECG classification models at inference time can yield a significant performance boost, even in the presence of noisy data and domain shifts.
Deep learning models for electrocardiogram (ECG) classification often suffer from significant performance degradation when deployed in unseen domains due to shifts in acquisition devices and patient populations. Test-time adaptation (TTA) offers a practical solution by adapting models using only unlabeled data at inference time. However, existing TTA methods often underperform on ECG tasks, since naive online updates ignore the hierarchical beat-rhythm structure of cardiac cycles and are vulnerable to signal artifacts, which leads to unstable adaptation and model drift. We propose BeatRhythm-TTA, an ECG-tailored TTA framework that explicitly accounts for ECG's noisy observations and structured beat-rhythm semantics under domain shift. First, to handle pervasive ECG artifacts, we introduce a Signal Quality Index (SQI)-gated adaptation scheme that selectively filters out low-quality signals to prevent harmful updates. Second, to leverage ECG's beat-rhythm semantics, we enforce dual-level consistency so the model preserves beat morphology and rhythm dynamics while adapting to shifted acquisition conditions. Extensive experiments on multi-label ECG diagnosis across three adaptation protocols, using PTB-XL as the source domain and CPSC2018/Georgia as two target domains, demonstrate the effectiveness of our method, yielding an average +2.70% relative improvement in Macro-F1 over the best competing method.
