Self-Supervised Pretraining for PHM: What Transfers, What Does Not, and What Federating It Costs
Organizations: Department of Computer Science and Artificial Intelligence, Universidad de Sevilla, Seville, Spain
Abstract
Self-supervised pretraining has produced capable time-series foundation models, but its value for Prognostics and Health Management (PHM) has not been measured under leakage-controlled conditions. We measure it on a corpus of 47 public PHM datasets, harmonized to a single tensor contract and partitioned so that 11 datasets are withheld from pretraining in full. Transfer turns out to be task-dependent: a compact channel-independent Transformer encoder (801,808 parameters) pretrained with masked patch prediction raises fault-classification macro-F1 from 0.350 to 0.812 on CWRU and from 0.569 to 0.951 on HSG18, yet on CMAPSS remaining-useful-life estimation no pretrained checkpoint reliably improves upon a from-scratch baseline. We pretrain under two regimes---centralized, and simulated cross-silo federated with ten clients grouped by physical domain---holding architecture, objective, source data and optimization budget fixed. Two further findings are negative. At matched budget the centralized encoder leads by 15.5--20.2 macro-F1 points in four of four comparisons, although the comparison is confounded by a learning-rate schedule that favours the centralized arm and rests on a single pretraining run per regime; on RUL the two regimes are indistinguishable. And the intrinsic SSL loss is not a valid selection criterion: the centralized encoder more than halves the federated reconstruction loss without a corresponding gap on RUL, and a server-momentum sweep moves downstream validation macro-F1 by up to 0.29 while the SSL loss varies by 0.021. Transfer also hinges on the backbone learning rate: the same checkpoint yields 0.169 or 0.829 macro-F1 on CWRU depending on a single order of magnitude. We release the corpus pipeline, the protocol and the code.
Figures & tables
| Quantity | Value |
|---|---|
| Candidate datasets in catalogue | 55 |
| Downloaded | 53 |
| Audited | 51 |
| Accepted into the protocol | 47 |
| Source (pretraining only) | 36 |
| Target (evaluation only) | 11 |
| Checkpoint | Regime | Budget (steps) | SSL val. loss |
|---|---|---|---|
| Central-100k | centralized | 100,000 | — |
| Central-25k | centralized | 25,000 | 0.3467 |
| FedAvg | federated | 25,000 | 0.7844 |
| FedAvgM | federated | 25,000 | 0.7636 |
| Configuration | Backbone LR | CWRU | HSG18 |
|---|---|---|---|
| From scratch (1 seed) | — | 0.350 | 0.569 |
| Linear probing, Central-100k (1 seed) | frozen | 0.705 | 0.906 |
| Central-100k | / | ||
| Central-25k | / | ||
| FedAvg | / | ||
| FedAvgM | / |
| Dataset | Comparison | df | |||
|---|---|---|---|---|---|
| CWRU | Central-25k FedAvg | 2.31 | 2.14 | 0.139 | |
| CWRU | Central-25k FedAvgM | 4.25 | 2.30 | 0.040 | |
| HSG18 | Central-25k FedAvg | 6.01 | 2.11 | 0.023 | |
| HSG18 | Central-25k FedAvgM | 3.92 | 2.06 | 0.057 |
| Checkpoint | Adaptation | Validation | Test |
|---|---|---|---|
| From scratch (1 seed) | linear head, full training | 39.08 | 43.67 |
| Central-100k (1 seed) | linear probing | 43.21 | 53.86 |
| Central-100k (1 seed) | full fine-tuning | 39.54 | 44.33 |
| Central-100k | MLP head, frozen | ||
| Central-25k | MLP head, frozen | ||
| FedAvg | MLP head, frozen |
| Mode | Test RMSE | Asymmetric score | |
|---|---|---|---|
| From scratch | 43.67 | 754,754 | |
| Linear probing | 53.86 | 1,002,485 | |
| Full fine-tuning | 44.33 | 645,389 |
| SSL validation loss ( ) | 0.784 | 0.778 | 0.765 | 0.764 | 0.768 |
|---|---|---|---|---|---|
| CWRU, linear probing ( ) | 0.163 | 0.160 | 0.217 | 0.371 | 0.273 |
| HSG18, full fine-tuning ( ) | 0.713 | 0.464 | 0.750 | 0.724 | 0.462 |
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | Status | Reason |
|---|---|---|
| CWRU | final, | classification, body of the paper |
| HSG18 | final, | classification, body of the paper |
| CMAPSS_RUL | final, | RUL, body of the paper |
| PBCP16 | diagnostic, | negative case, Appendix B |
| PHM18 | excluded from body | run with uncapped batch; pending re-execution |
| PHMAP23 | documented negative | extreme class imbalance, noise floor |
| Class | Support | Precision | Recall | F1 |
|---|---|---|---|---|
| 0 | 477 | 0.480 | 0.998 | 0.648 |
| 1 | 2378 | 0.997 | 0.799 | 0.887 |
| 2 | 476 | 0.998 | 0.910 | 0.952 |
| 3 | 0 | — | — | undefined |
| Macro-F1 over the three supported classes | 0.829 | |||
| Parameter | Value |
|---|---|
| Window / stride / patch | 512 / 256 / 16 |
| Patches per window ( ) | 32 |
| Mask ratio | 0.30 of valid patches |
| / layers / heads | 128 / 4 / 4 |
| Normalization | pre-norm; per-window, per-channel instance norm |
| Trainable parameters | 801,808 |
Self-Supervised Pre-Training
Self-supervised pre-training learns transferable representations from unlabeled data to reduce the labels needed for downstream tasks.
See all 251 papers