Learning Regularization Structure for Biosignal Template Estimation

Abstract

Estimating event-locked templates from bio-signal recordings via regularized least-squares requires choosing both the regularization structure and its magnitude, choices that are typically made heuristically. We develop a data-driven framework based on Stein's Unbiased Risk Estimate (SURE) that jointly optimizes both. By parameterizing the regularization operator as a convolution kernel, our method learns the penalty structure directly from the data, combining smoothness enforcement with ridge-like shrinkage in a way that cannot be achieved by scaling a fixed difference operator. While standard SURE assumes white noise, biosignal noise exhibits temporal autocorrelation. We therefore extend SURE to colored noise by replacing its scalar trace term with a structured correction based on the noise covariance matrix. For AR(1) noise, this correction requires only two parameters, the noise variance and the lag-1 autocorrelation, both estimable from pre-event baselines. Cross-modality validation on auditory event-related potentials, P300 brain--computer interface data, and ECG morphology demonstrates consistent gains compared to alternative methods, all at $K=5$ events per class - the regime most relevant for rapid calibration and personalization.

Abstract: arXiv descriptive metadata (CC0). Thank you to arXiv for use of its open access interoperability. This service was not reviewed or approved by, nor does it necessarily express or reflect the policies or opinions of, arXiv.