Change search
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf
Collaborative Training of Data-Driven Remaining Useful Life Prediction Models Using Federated Learning
RISE Research Institutes of Sweden, Materials and Production, Methodology, Textiles and Medical Technology. Mälardalen University, Sweden.ORCID iD: 0000-0002-1262-9143
RISE Research Institutes of Sweden, Digital Systems, Smart Hardware.ORCID iD: 0000-0001-8278-8601
RISE Research Institutes of Sweden, Digital Systems, Smart Hardware.ORCID iD: 0000-0001-8993-1335
RISE Research Institutes of Sweden, Digital Systems, Smart Hardware.ORCID iD: 0000-0002-2232-7835
Show others and affiliations
2024 (English)In: International Journal of Prognostics and Health Management, E-ISSN 2153-2648, Vol. 15, no 2Article in journal (Refereed) Published
Abstract [en]

Remaining useful life prediction models are a central aspect of developing modern and capable prognostics and health management systems. Recently, such models are increasingly data-driven and based on various machine learning techniques, in particular deep neural networks. Such models are notoriously “data hungry”, i.e., to get adequate performance of such models, a substantial amount of diverse training data is needed. However, in several domains in which one would like to deploy data-driven remaining useful life models, there is a lack of data or data are distributed among several actors. Often these actors, for various reasons, cannot share data among themselves. In this paper a method for collaborative training of remaining useful life models based on federated learning is presented. In this setting, actors do not need to share locally held secret data, only model updates. Model updates are aggregated by a central server, and subsequently sent back to each of the clients, until convergence. There are numerous strategies for aggregating clients’ model updates and in this paper two strategies will be explored: 1) federated averaging and 2) federated learning with personalization layers. Federated averaging is the common baseline federated learning strategy where the clients’ models are averaged by the central server to update the global model. Federated averaging has been shown to have a limited ability to deal with non-identically and independently distributed data. To mitigate this problem, federated learning with personalization layers, a strategy similar to federated averaging but where each client is allowed to append custom layers to their local model, is explored. The two federated learning strategies will be evaluated on two datasets: 1) run-to-failure trajectories from power cycling of silicon-carbide metal-oxide semiconductor field-effect transistors, and 2) C-MAPSS, a well-known simulated dataset of turbofan jet engines. Two neural network model architectures commonly used in remaining useful life prediction, long short-term memory with multi-layer perceptron feature extractors, and convolutional gated recurrent unit, will be used for the evaluation. It is shown that similar or better performance is achieved when using federated learning compared to when the model is only trained on local data.

Place, publisher, year, edition, pages
2024. Vol. 15, no 2
Keywords [en]
remaining useful life, federated learning, machine learning, prognostics and health management, deep learning, electronics, turbofan jet engines
National Category
Reliability and Maintenance
Identifiers
URN: urn:nbn:se:ri:diva-75674DOI: 10.36001/ijphm.2024.v15i2.3821OAI: oai:DiVA.org:ri-75674DiVA, id: diva2:1903743
Available from: 2024-10-07 Created: 2024-10-07 Last updated: 2025-09-23Bibliographically approved

Open Access in DiVA

fulltext(8436 kB)297 downloads
File information
File name FULLTEXT01.pdfFile size 8436 kBChecksum SHA-512
eb9ce8bddd1ffbf2f7f5e5a39877fd9890c0bed4d41993cd78752fbf358a5fe5f096778245836f447b46df55ec4b2444129fe2a11fd953c2797d4d5c72963e7e
Type fulltextMimetype application/pdf

Other links

Publisher's full text

Authority records

Söderkvist Vermelin, WilhelmMishra, MadhavEng, Mattias P.Andersson, Dag

Search in DiVA

By author/editor
Söderkvist Vermelin, WilhelmMishra, MadhavEng, Mattias P.Andersson, DagKyprianidis, Konstantinos
By organisation
Methodology, Textiles and Medical TechnologySmart Hardware
In the same journal
International Journal of Prognostics and Health Management
Reliability and Maintenance

Search outside of DiVA

GoogleGoogle Scholar
Total: 297 downloads
The number of downloads is the sum of all downloads of full texts. It may include eg previous versions that are now no longer available

doi
urn-nbn

Altmetric score

doi
urn-nbn
Total: 709 hits
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf