Federated Quantum-Train with Batched Parameter Generation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Liu, Chen-Yu, Chen, Samuel Yen-Chi
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912014626979840
author Liu, Chen-Yu
Chen, Samuel Yen-Chi
author_facet Liu, Chen-Yu
Chen, Samuel Yen-Chi
contents In this work, we introduce the Federated Quantum-Train (QT) framework, which integrates the QT model into federated learning to leverage quantum computing for distributed learning systems. Quantum client nodes employ Quantum Neural Networks (QNNs) and a mapping model to generate local target model parameters, which are updated and aggregated at a central node. Testing with a VGG-like convolutional neural network on the CIFAR-10 dataset, our approach significantly reduces qubit usage from 19 to as low as 8 qubits while reducing generalization error. The QT method mitigates overfitting observed in classical models, aligning training and testing accuracy and improving performance in highly compressed models. Notably, the Federated QT framework does not require a quantum computer during inference, enhancing practicality given current quantum hardware limitations. This work highlights the potential of integrating quantum techniques into federated learning, paving the way for advancements in quantum machine learning and distributed learning systems.
format Preprint
id arxiv_https___arxiv_org_abs_2409_02763
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Federated Quantum-Train with Batched Parameter Generation
Liu, Chen-Yu
Chen, Samuel Yen-Chi
Quantum Physics
In this work, we introduce the Federated Quantum-Train (QT) framework, which integrates the QT model into federated learning to leverage quantum computing for distributed learning systems. Quantum client nodes employ Quantum Neural Networks (QNNs) and a mapping model to generate local target model parameters, which are updated and aggregated at a central node. Testing with a VGG-like convolutional neural network on the CIFAR-10 dataset, our approach significantly reduces qubit usage from 19 to as low as 8 qubits while reducing generalization error. The QT method mitigates overfitting observed in classical models, aligning training and testing accuracy and improving performance in highly compressed models. Notably, the Federated QT framework does not require a quantum computer during inference, enhancing practicality given current quantum hardware limitations. This work highlights the potential of integrating quantum techniques into federated learning, paving the way for advancements in quantum machine learning and distributed learning systems.
title Federated Quantum-Train with Batched Parameter Generation
topic Quantum Physics
url https://arxiv.org/abs/2409.02763