Parameter-efficient Quantum Multi-task Learning
Organizations: Department of Electrical and Electronic Engineering, University of Melbourne, Parkville, 3010, VIC, Australia. · CSIRO, Sydney, NSW, Australia.
Abstract
Multi-task learning (MTL) improves generalization and data efficiency by jointly learning related tasks through shared representations. In the widely used hard-parameter-sharing setting, a shared backbone is combined with task-specific prediction heads. However, task-specific parameters can grow rapidly with the number of tasks. Therefore, designing multi-task heads that preserve task specialization while improving parameter efficiency remains a key challenge. In Quantum Machine Learning (QML), variational quantum circuits (VQCs) provide a compact mechanism for mapping classical data to quantum states residing in high-dimensional Hilbert spaces, enabling expressive representations within constrained parameter budgets. We propose a parameter-efficient quantum multi-task learning (QMTL) framework that replaces conventional task-specific linear heads with a fully quantum prediction head in a hybrid architecture. The model consists of a VQC with a shared, task-independent quantum encoding stage, followed by lightweight task-specific ansatz blocks enabling localized task adaptation while maintaining compact parameterization. Under a controlled and capacity-matched formulation where the shared representation dimension grows with the number of tasks, our parameter-scaling analysis demonstrates that a standard classical head exhibits quadratic growth, whereas the proposed quantum head parameter cost scales linearly. We evaluate QMTL on three multi-task benchmarks spanning natural language processing, medical imaging, and multimodal sarcasm detection, where we achieve performance comparable to, and in some cases exceeding, classical hard-parameter-sharing baselines while consistently outperforming existing hybrid quantum MTL models with substantially fewer head parameters. We further demonstrate QMTL's executability on noisy simulators and real quantum hardware, illustrating its feasibility.
Figures & tables
| Dataset | Head variant | Head detail | Training method |
| GLUE (Frozen BERT backbone) | Classical MTL head | 1 linear layer per task | Task-sampled MTL (with capped batches and LR scheduler) Classification: BCEWithLogitsLoss , CrossEntropyLoss , Regression: MSELoss |
| HQNN (4 or 10 qubits) | VQC ( Phukan et al., 2024 ) followed by 1 linear layer per task | ||
| QMTL (10 qubits) | QMTL head ( & ) | ||
| CheXpert (DenseNet121 backbone) | Classical MTL head | 1 linear layer per task | Parallel tasks MTL Masked loss for missing labels; Class-weighted focal loss (Appendix Eq. 24 ) 3-class train, binary evaluations |
| HQNN (4 or 10 qubits) | VQC ( Phukan et al., 2024 ) followed by 1 linear layer per task | ||
| QMTL (10 qubits) | QMTL head ( for all tasks) |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | Modality | # tasks | Task Type | # samples | Labeling scheme |
| GLUE Benchmark | Text (NLP) | 9 | Binary, Multiclass, Regression | per task | Exactly 1-labeled task per sample |
| CheXpert | Chest X-ray (image) | 5 | Binary | k total | labeled task per sample |
| Extended MUStARD | Multimodal (audio, text, visual) | 5 | Binary, Multiclass | total | Fully labeled |
| Dataset | Task | Metrics |
| CheXpert | Atelectasis † | Acc, P, R, F1 |
| Cardiomegaly † | Acc, P, R, F1 | |
| Consolidation † | Acc, P, R, F1 | |
| Edema † | Acc, P, R, F1 | |
| Pleural Effusion † | Acc, P, R, F1 | |
| GLUE | COLA † | Matthews_corr, Acc |