IEEE International Reliability Physics Symposium proceedings

Robustness of Quantum Federated Learning (QFL) Against “Label Flipping Attacks” for Lithography Hotspot Detection in Semiconductor Manufacturing

Amandeep Singh Bhatia, Sabre Kais, Muhammad Ashraful Alam

Published Apr 14, 2024• 6 citations

Abstract

The geographical dispersion of manufacturing units introduces challenges related to latency, network connectivity, data synchronization, communication overhead, scalability, data security, and resource allocation. Manufacturing data often includes sensitive information related to proprietary processes, designs, and technologies. Companies are hesitant to share this data such as lithography hotspot data (LHD) directly due to concerns about privacy and the risk of intellectual property theft. Federated learning provides a solution to address these challenges by allowing companies to collaboratively train models without sharing raw data. It allows companies to adhere to privacy laws while benefiting from shared insights. In recent years, the convergence of quantum machine learning has demonstrated transformative potential by leveraging quantum principles to address complex problems across various industries. In this work, a concept of quantum federated learning consisting of variational quantum circuit is proposed to address the lithography hotspot sharing challenges by providing a privacy-preserving, centralized, and quantum-enhanced framework. It combines the collaborative advantages of federated learning with the unique capabilities of quantum computing, offering a solution that is secure, scalable, and adaptable to the complexities of modern manufacturing environments. The performance evaluations involve scalability and robustness, particularly in the context of label-flipping attacks, which is crucial for validating, optimizing, and enhancing the quantum federated learning framework's applicability to real-world challenges in the manufacturing sector.

DOI: 10.1109/irps48228.2024.10529306

Related papers

On the Robustness of Variational Quantum Classifier Against “Label Flipping Attacks” in Federated Learning for Semiconductor Manufacturing

Amandeep Singh Bhatia, Sabre Kais, Muhammad Ashraful Alam

Due to the widespread presence of manufacturing units, real-time data collection becomes a challenging task, especially as the scale of operations continues to expand. Moreover, manufacturing industries often deal with sensitive and proprietary data and are reluctant to share data directly in areas like lithography hotspot detection (LHD) due to concerns about IP protection and regulatory compliance. Moreover, human error can lead to incorrect labeling of parts or products. Detecting hotspots in lithography is crucial for ensuring the product quality and reliability of the manufacturing process. In today's context, Federated Learning (FL) stands out as a crucial remedy, facilitating the rapid advancement of distributed machine learning while effectively managing critical concerns regarding data privacy and governance. Moreover, FL systems must be robust against not only malicious label flipping attacks but also inadvertent human errors. In recent years, the convergence of quantum computing and machine learning represents a groundbreaking interdisciplinary approach with immense potential for revolutionizing various industries ranging from healthcare to manufacturing. In this work, we proposed a Quantum Federated Learning (QFL) framework for lithography hotspot detection and offer a unique solution to address the challenges mentioned above effectively. To demonstrate the generalization ability and robustness, we performed extensive experiments with label-flipping attacks in the proposed framework. Based on our results, the global quantum model maintained a high classification accuracy and best performance even when faced with label-flipping attacks from a specific number of clients in federated learning settings, while keeping data private across manufacturing units. Moreover, a defense strategy based on clustering is proposed to distinguish the good and bad model updates from the participants in a quantum federated framework.

IEEE International Conference on Quantum Computing and Engineering (QCE)Sep 15, 20242 citations

FedQNN: Federated Learning using Quantum Neural Networks

Nouhaila Innan, Muhammad Al-Zafar Khan, Alberto Marchisio +2 more

In this study, we explore the innovative domain of Quantum Federated Learning (QFL) as a framework for training Quantum Machine Learning (QML) models via distributed networks. Conventional machine learning models frequently grapple with issues about data privacy and the exposure of sensitive information. Our proposed Federated Quantum Neural Network (FedQNN) framework emerges as a cutting-edge solution, integrating the singular characteristics of QML with the principles of classical federated learning. This work thoroughly investigates QFL, underscoring its capability to secure data handling in a distributed environment and facilitate cooperative learning without direct data sharing. Our research corroborates the concept through experiments across varied datasets, including genomics and healthcare, thereby validating the versatility and efficacy of our FedQNN framework. The results consistently exceed 86% accuracy across three distinct datasets, proving its suitability for conducting various QML tasks. Our research not only identifies the limitations of classical paradigms but also presents a novel framework to propel the field of QML into a new era of secure and collaborative innovation.

arXiv:2403.10861International Joint Conference on Neural Networks (IJCNN)Jun 30, 202470 citations

Enhancing Quantum Federated Learning with Fisher Information-Based Optimization

Amandeep Singh Bhatia, Sabre Kais

Federated Learning (FL) has become increasingly popular across different sectors, offering a way for clients to work together to train a global model without sharing sensitive data. It involves multiple rounds of communication between the global model and participating clients, which introduces several challenges like high communication costs, heterogeneous client data, prolonged processing times, and increased vulnerability to privacy threats. In recent years, the convergence of federated learning and parameterized quantum circuits has sparked significant research interest, with promising implications for fields such as healthcare and finance. By enabling decentralized training of quantum models, it allows clients or institutions to collaboratively enhance model performance and outcomes while preserving data privacy. Recognizing that Fisher information can quantify the amount of information that a quantum state carries under parameter changes, thereby providing insight into its geometric and statistical properties. We intend to leverage this property to address the aforementioned challenges. In this work, we propose a Quantum Federated Learning (QFL) algorithm that makes use of the Fisher information computed on local client models, with data distributed across heterogeneous partitions. This approach identifies the critical parameters that significantly influence the quantum model's performance, ensuring they are preserved during the aggregation process. Our research assessed the effectiveness and feasibility of QFL by comparing its performance against other variants, and exploring the benefits of incorporating Fisher information in QFL settings. Experimental results on ADNI and MNIST datasets demonstrate the effectiveness of our approach in achieving better performance and robustness against the quantum federated averaging method.

arXiv:2507.17580IEEE International Conference on Quantum Computing and Engineering (QCE)Aug 30, 20251 citation

Quantum Federated Learning: Bridging Quantum Computing and Distributed AI

Tariq Qayyum, Manzoor Ahmed Khan, Asadullah Tariq +4 more

In recent years, the growing demand for secure, efficient, and scalable machine learning has highlighted significant challenges in traditional Federated Learning (FL) systems. These challenges include computational inefficiencies in handling large-scale, high-dimensional data, privacy concerns, and the limitations of classical hardware in solving complex optimization problems. Additionally, the increasing complexity of data and the need for more robust privacy-preserving methods have pushed the boundaries of classical FL. To address these issues, this paper presents a detailed framework for Quantum Federated Learning (QFL), where quantum computing and FL converge to enable distributed quantum model training with enhanced privacy preservation. The system is modeled using local quantum models on clients and a global aggregation strategy at the server. We develop quantum gradient-based optimization with quantum cross-entropy loss, privacy preservation through encryption, and robustness to quantum noise. A set of theorems is provided to prove the correctness and efficiency of the model, supported by rigorous mathematical formulations. We used IBM Qiskit for the simulation and compared our proposed QFL with classical FL. The results demonstrate that our QFL outperformed classical FL in terms of accuracy, precision, recall, and F1 score.

IEEE/ACM International Conference on Utility and Cloud Computing (UCC)Dec 16, 20244 citations

Revolutionizing Machine Learning Security: The Role of Quantum-Enhanced Federated Learning

E. Kannan, S Ravikumar, Carmel Mary Belinda M J +3 more

In today's age of collaborative machine learning where data sharing happens across organizations and privacy becomes a real issue. Old-school methods of securing systems work, but tend to lack against popular new attacks. The research uses a novel approach that uses quantum computing to reimagine what privacy means. An article which deals with the combination of quantum technologies and federated learning systems, specifically discussing how it may set a new standard for data privacy. The plan is to use quantum key distribution and entanglement-based protocols in order to fortify the defences of collaborative machine learning systems against adversarial attacks or unauthorized access, according a consortium statement. The approach to quantum -enhanced federated learning presented provides a solution not only with secure improvements but also implementing more extreme circumstances in decentralised learning scenarios, such as challenges due to different data available and regulatory demands. This paper provides a theoretical benchmark for quantum federated learning and focuses on the underlying concepts leading to excellent security guarantees. Applications and simulations show that this framework can be used in practice with effectiveness. Furthermore, this work benchmarks the scalability of quantum federated learning for different classical and common machine-learning models. The results are a major step towards the goal of making more sophisticated privacy solutions available and allow to create secure, private-first workspace for collaborative machine learning. By leveraging the latest in quantum computing along with specialized tactics designed to combat rapidly evolving data protection threats, this approach will provide a strong foundation for future applications of shared machine learning within distributed quantum environments.

International Conference on Emerging Research in Computational Science (ICERCS)Dec 12, 20244 citations

Communication-efficient Quantum Federated Learning Optimization for Multi-Center Healthcare Data

Amandeep Singh Bhatia, Mandeep Kaur Saggi, Sabre Kais

In the healthcare sector, the scarcity of data and privacy concerns present formidable challenges to the widespread adoption of machine learning. In the present-day scenario, Federated Learning (FL) emerges as a pivotal solution, fostering the rapid evolution of distributed machine learning paradigms while adeptly addressing the problem of data governance and privacy. It allows distributed clients to collaboratively train a global model by synchronizing their local updates without sharing private data. In recent years, federated learning and quantum computing have individually shown great promise to revolutionize various sectors, including healthcare, finance, and manufacturing, where privacy protection is paramount. In this article, we propose a communication-efficient Quantum Federated Learning (QFL) framework based on a variational circuit that enables clients to efficiently train and transmit quantum model parameters, thereby reducing communication rounds significantly and enhancing QFL performance using quantum natural gradient descent (QNGD) optimization. This paper demonstrates the feasibility of a QFL framework for predicting the presence of coronary heart disease, diagnosing whether a patient is suffering from diabetes or not, and differentiating malignant and benign cancer by distributing the UCI datasets unbalanced among healthcare institutions. The proposed framework has the potential to incorporate privacy, security, and the expedited processing of distributed data. QNGD outperformed classical GD by reducing communication rounds by a range of 5% to 60%. In addition to reducing the communication rounds by optimizing the QFL training algorithm and achieving quicker convergence, it also determines the important features regardless of the data imbalance among the clients.

IEEE-EMBS International Conference on Biomedical and Health Informatics (BHI ...)Nov 10, 202410 citations