A blockchain-empowered secure federated domain generalization framework for machinery fault diagnosisShow others and affiliations
2024 (English)In: Advanced Engineering Informatics, ISSN 1474-0346, E-ISSN 1873-5320, Vol. 62, article id 102756Article in journal (Refereed) Published
Abstract [en]
The digitization transformation of traditional machinery and advances in artificial intelligence have led to the development of data-driven machinery fault diagnosis methods. However, limited by the number of machinery equipment, it is challenging for small or medium sized manufacturing enterprises (SMEs) to collect sufficient data to support the effective execution of these methods. In addition, due to the potential conflicts of interest and risks of privacy leakage, direct sharing of raw data between enterprises is often impractical. To this end, a blockchain-empowered secure federated domain generalization (FDG) framework is proposed in this paper, aiming to achieve distributed collaborative machinery fault diagnosis. In this framework, blockchain technology is first employed to replace the central server in federated learning (FL) system, effectively mitigating the single-point-of-failure issue of the FL system. Second, a committee-based consensus mechanism is designed to verify the correctness of the global model. To achieve domain generalization (DG) in federated setting, two regularizers are incorporated into the proposed framework, which restrict the information contained in representation and perform implicit distribution alignment. Experimental studies on two datasets demonstrate that the proposed method outperforms state-of-the-art FDG methods in terms of diagnosis accuracy. The high reliability of the proposed framework makes it more suitable for practical industrial scenarios.
Place, publisher, year, edition, pages
Elsevier BV , 2024. Vol. 62, article id 102756
Keywords [en]
Blockchain, Domain generalization, Fault diagnosis, Federated learning
National Category
Computer Sciences Computer Systems
Identifiers
URN: urn:nbn:se:kth:diva-351889DOI: 10.1016/j.aei.2024.102756ISI: 001292592900001Scopus ID: 2-s2.0-85200644802OAI: oai:DiVA.org:kth-351889DiVA, id: diva2:1890105
Note
QC 20240829
2024-08-192024-08-192024-09-05Bibliographically approved