Training and Scaling Compute-Optimal Physiological Waveform Foundation Models
cs.LG
Submitted: 2026-10-05
Updated: 2026-10-05
Code: https://github.com/jax-ml/jax
License: http://creativecommons.org/licenses/by/4.0/
The gist: We investigate the scaling laws and compute-optimal training of physiological waveform foundation models (FMs).
Terminology
Abstract
We investigate the scaling laws and compute-optimal training of physiological waveform foundation models (FMs). We train Aether, a family of over one hundred FMs ranging from 20M to 2.1B parameters, on up to 36.3M hours of physiological waveforms. We construct eight clinical prediction tasks from MIMIC-III and evaluate the FMs through linear probing. The 720M FM outperforms all existing baseline FMs across all eight tasks. A scaling law of model size, pretraining hours, and labeled patients predicts downstream ranking error, i.e. 1-AUROC, effectively with 0.5% prediction MAE at held-out resource scales and 0.9% MAE when extrapolating to 2.1B parameters. We present three findings: (1) Compute-optimal training scales both FM size and pretraining hours. Under the fitted law, a 10.0 times increase in compute FLOPs scales model size by 1.2 times and pretraining hours by 8.2 times. (2) Larger FMs use waveform data more efficiently, and greater pretraining exposure increases the benefit of model scaling. Starting from 25M parameters and 4.8M pretraining hours, doubling FM size reduces the predicted hours needed for the same performance by 51.8%. (3) Pretraining and clinical supervision reinforce each other: more labeled patients increase the return to pretraining, while larger FMs and longer pretraining reduce labeling requirements. For the example of the 720M FM, extending pretraining from 120K to 36.3M hours reduces the predicted patient requirement by 61% at a target ranking error. These findings provide a quantitative training recipe and a promising and durable scaling path for physiological waveform modeling and downstream clinical prediction.
Sources
- Pretraining Strategies and Scaling for ECG Foundation Models: A Systematic Study
- Scaling laws for decoding images from brain activity
- Broken Neural Scaling Laws
- SOTER: A Generative Time-Series Foundation Model for Wearable Human Physiological Signals
- QualityFM: a Multimodal Physiological Signal Foundation Model with Self-Distillation for Signal Quality Challenges in Critically Ill Patients
- DIVER-1: Scaling Intracranial EEG Foundation Models for Transferable Representations
- Scaling Laws for Transfer
- Deep Learning Scaling is Predictable, Empirically
- MiniCPM: Unveiling the Potential of Small Language Models with Scalable Training Strategies
- Scaling Laws for Neural Language Models
- The Impact of Positional Encoding on Length Generalization in Transformers
- How Do Electrocardiogram Models Scale?
- Decoupled Weight Decay Regularization
- Toward World Modeling of Physiological Signals with Chaos-Theoretic Balancing and Latent Dynamics
- ECG-FM: An Open Electrocardiogram Foundation Model
- Scaling Wearable Foundation Models
- GLU Variants Improve Transformer
- sleep2vec: Unified Cross-Modal Alignment for Heterogeneous Nocturnal Biosignals
- Root Mean Square Layer Normalization
- Exploring Scaling Laws for EHR Foundation Models
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks