A comprehensive inference-time augmentation framework in physiological signals: application to PPG-based AF detection.

Abstract

Objective.Accurate classification of physiological signals in real-world deployments is challenged by sensor noise, motion artifacts, and distribution shifts between training and deployment data. Inference-time augmentation (ITA), which applies augmentations during inference rather than retraining, offers a simple, model-agnostic mechanism to improve robustness. However, ITA application to physiological signals has remained narrow in scope, relying on limited augmentation methods with fixed, unoptimized parameters. This work proposes a unified ITA framework to address that gap.Approach.The framework incorporates 13 augmentation methods spanning time-domain, amplitude-domain, frequency-domain, and artifact-injection transformations, with hyperparameters systematically optimized via Bayesian optimization. We evaluate the framework on atrial fibrillation (AF) detection from 30 s photoplethysmography (PPG) signals using two deep learning architectures, generative pre-trained transformer (GPT)-PPG (in two sizes) and ResNet, across five datasets comprising more than 400 patients and ∼9800 h of PPG recording. Two evaluation strategies are assessed: standard ITA applied to all inputs, and selective ITA applied to initially positive predictions.Main results.Standard ITA consistently improved the area under the receiver operating characteristic curve (AUROC, up to 8.5% for GPT-PPG and 0.7% for ResNet) and the area under the precision-recall curve (AUPRC, up to 10.6% for GPT-PPG and 0.8% for ResNet) across all model-dataset combinations. Selective ITA further reduced average FPR by up to 4.4% (GPT-PPG) and 1.3% (ResNet) on non-AF PPG datasets.Significance.These findings establish ITA as a practical, model-agnostic approach for improving the reliability of PPG-based AF classification in deployment settings where retraining is not feasible, with broader applicability to physiological signal analysis.