Skewness-Guided Pruning of Multimodal Swin Transformers for Federated Skin Lesion Classification on Edge Devices
Not on the map. Why?
Abstract (arXiv)
In recent years, high-performance computer vision models have achieved remarkable success in medical imaging, with some skin lesion classification systems even surpassing dermatology specialists in diagnostic accuracy. However, such models are computationally intensive and large in size, making them unsuitable for deployment on edge devices. In addition, strict privacy constraints hinder centralized data management, motivating the adoption of Federated Learning (FL). To address these challenges, this study proposes a skewness-guided pruning method that selectively prunes the Multi-Head Self-Attention and Multi-Layer Perceptron layers of a multimodal Swin Transformer based on the statistical skewness of their output distributions. The proposed method was validated in a horizontal FL environment and shown to maintain performance while substantially reducing model complexity. Experiments on the compact Swin Transformer demonstrate approximately 36% model size reduction with no loss in accuracy. These findings highlight the feasibility of achieving efficient model compression and privacy-preserving distributed learning for multimodal medical AI on edge devices.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 20 of its 38 references that are in the map. First posted on arXiv 9 Dec 2025.
Builds on
20 of 38 references.
References · 20
- 2017Attention Is All You Needcited by 59,932
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,082
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,592
- 2018SNIP: Single-shot Network Pruning based on Connection Sensitivitycited by 802
- 2019Skin Lesion Analysis Toward Melanoma Detection 2018: A Challenge Hosted by the International Skin Imaging Collaboration (ISIC)cited by 657
- 2021Intriguing Properties of Vision Transformerscited by 446
- 2019Model Pruning Enables Efficient Federated Learning on Edge Devicescited by 219
- 2021Chasing Sparsity in Vision Transformers: An End-to-End Explorationcited by 181
- 2021Vision Transformer Pruningcited by 77
- 2021Global Vision Transformer Pruning with Hessian-Aware Saliencycited by 69
- 2023X-Pruner: eXplainable Pruning for Vision Transformerscited by 66
8 more
- 2020Pruning Algorithms to Accelerate Convolutional Neural Networks for Edge Applications: A Surveycited by 26
- 2024FedMef: Towards Memory-efficient Federated Dynamic Pruningcited by 14
- 2023Complement Sparsification: Low-Overhead Model Pruning for Federated Learningcited by 12
- 2022Federated Pruning: Improving Neural Network Efficiency with Federated Learningcited by 8
- 2023Model Pruning Enables Localized and Efficient Federated Learning for Yield Forecasting and Data Sharingcited by 7
- 2022Exploring Advances in Transformers and CNN for Skin Lesion Diagnosis on Small Datasetscited by 4
- 2024VTrans: Accelerating Transformer Compression with Variational Information Bottleneck based Pruningcited by 3
- 2025Enhancing Fairness in Skin Lesion Classification for Medical Diagnosis Using Prune Learningcited by 2
Built on by
Citing papers per year: 2026: 1.
1 paper.