Multi-Task Distributed Learning using Vision Transformer with Random Patch Permutation
Not on the map. Why?
Abstract (arXiv)
The widespread application of artificial intelligence in health research is currently hampered by limitations in data availability. Distributed learning methods such as federated learning (FL) and shared learning (SL) are introduced to solve this problem as well as data management and ownership issues with their different strengths and weaknesses. The recent proposal of federated split task-agnostic (FeSTA) learning tries to reconcile the distinct merits of FL and SL by enabling the multi-task collaboration between participants through Vision Transformer (ViT) architecture, but they suffer from higher communication overhead. To address this, here we present a multi-task distributed learning using ViT with random patch permutation. Instead of using a CNN based head as in FeSTA, p-FeSTA adopts a randomly permuting simple patch embedder, improving the multi-task learning performance without sacrificing privacy. Experimental results confirm that the proposed method significantly enhances the benefit of multi-task collaboration, communication efficiency, and privacy preservation, shedding light on practical multi-task distributed learning in the field of medical imaging.
Key facts. Cited by 10 papers in the map (data through 7 Oct 2026). Builds on 37 of its 51 references that are in the map. First posted on arXiv 7 Apr 2022.
Builds on
37 of 51 references.
References · 37
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,592
- 2016Membership Inference Attacks against Machine Learning Modelscited by 2,221
- 2015Unsupervised Representation Learning with Deep Convolutional Generative Adversarial Networkscited by 2,048
- 2019Federated Learning: Challenges, Methods, and Future Directionscited by 1,531
- 2016Federated Learning: Strategies for Improving Communication Efficiencycited by 1,461
- 2016Unsupervised Learning of Visual Representations by Solving Jigsaw Puzzlescited by 1,379
- 2019Deep Leakage from Gradientscited by 1,004
- 2020Pre-Trained Image Processing Transformercited by 891
- 2018How To Backdoor Federated Learningcited by 745
- 2020Inverting Gradients -- How easy is it to break privacy in federated learning?cited by 707
25 more
- 2020The Future of Digital Health with Federated Learningcited by 625
- 2020Flower: A Friendly Federated Learning Research Frameworkcited by 449
- 2021Intriguing Properties of Vision Transformerscited by 446
- 2018Split learning for health: Distributed deep learning without sharing raw patient datacited by 391
- 2020iDLG: Improved Deep Leakage from Gradientscited by 367
- 2019Multi-Task Deep Neural Networks for Natural Language Understandingcited by 349
- 2020SplitFed: When Federated Learning Meets Split Learningcited by 308
- 2020Data Poisoning Attacks Against Federated Learning Systemscited by 266
- 2020Threats to Federated Learning: A Surveycited by 171
- 2019Privacy-preserving Federated Brain Tumour Segmentationcited by 169
- 2020Privacy and Robustness in Federated Learning: Attacks and Defensescited by 163
- 2020Unleashing the Tiger: Inference Attacks on Split Learningcited by 104
- 2020Learning to Detect Malicious Clients for Robust Federated Learningcited by 86
- 2021Federated Learning: Opportunities and Challengescited by 83
- 2020BIMCV COVID-19+: a large annotated dataset of RX and CT images from COVID-19 patientscited by 81
- 2020End-to-End Evaluation of Federated Learning and Split Learning for Internet of Thingscited by 64
- 2021Multi-Task Learning in Natural Language Processing: An Overviewcited by 61
- 2020Straggler-Resilient Federated Learning: Leveraging the Interplay Between Statistical Accuracy and System Heterogeneitycited by 56
- 2020Weakly Supervised Lesion Localization With Probabilistic-CAM Poolingcited by 35
- 2021Vision Transformer using Low-level Chest X-ray Feature Corpus for COVID-19 Diagnosis and Severity Quantificationcited by 21
- 2021Membership Inference Attacks on Deep Regression Models for Neuroimagingcited by 19
- 2021Federated Split Vision Transformer for COVID-19 CXR Diagnosis using Task-Agnostic Trainingcited by 17
- 2022Feature Space Hijacking Attacks against Differentially Private Split Learningcited by 14
- 2019Convolutional Neural Networks on Randomized Datacited by 6
- 2019Privacy-Preserving Machine Learning Using EtC Imagescited by 3
Built on by
Citing papers per year: 2022: 1; 2023: 2; 2024: 5; 2025: 2.
10 papers.
- 2024From Challenges and Pitfalls to Recommendations and Opportunities: Implementing Federated Learning in Healthcarecited by 16
- 2023Recent Methodological Advances in Federated Learning for Healthcarecited by 11
- 2022Vision Transformers in Medical Imaging: A Reviewcited by 8
- 2025Enhancing Monocular Depth Estimation with Multi-Source Auxiliary Taskscited by 2
- 2024Private Knowledge Sharing in Distributed Learning: A Surveycited by 1
- 2024Heterogeneous Federated Learning with Splited Language Modelcited by 1
- 2024Task-agnostic Decision Transformer for Multi-type Agent Control with Federated Split Trainingcited by 1
- 2023Single-round Self-supervised Distributed Learning using Vision Transformercited by 0
- 2024Privacy-Preserving Vision Transformer Using Images Encrypted with Restricted Random Permutation Matricescited by 0
- 2025MedSegNet10: A Publicly Accessible Network Repository for Split Federated Medical Image Segmentationcited by 0
- 2025MedSegNet10: A Publicly Accessible Network Repository for Split Federated Medical Image Segmentation26 Mar 2025
- 2025Enhancing Monocular Depth Estimation with Multi-Source Auxiliary Tasks22 Jan 2025
- 2024From Challenges and Pitfalls to Recommendations and Opportunities: Implementing Federated Learning in Healthcare15 Sep 2024
- 2024Privacy-Preserving Vision Transformer Using Images Encrypted with Restricted Random Permutation Matrices16 Aug 2024
- 2024Task-agnostic Decision Transformer for Multi-type Agent Control with Federated Split Training22 May 2024
- 2024Heterogeneous Federated Learning with Splited Language Model24 Mar 2024
- 2024Private Knowledge Sharing in Distributed Learning: A Survey8 Feb 2024
- 2023Recent Methodological Advances in Federated Learning for Healthcare4 Oct 2023
- 2023Single-round Self-supervised Distributed Learning using Vision Transformer5 Jan 2023
- 2022Vision Transformers in Medical Imaging: A Review18 Nov 2022