DICE: Decentralized Incentive-driven Federated Learning for Efficient & Scalable Customization of Pretrained Models in Web3
Organizations: The State Key Laboratory of Blockchain and Data Security, Zhejiang University Hangzhou High-Tech Zone (Binjiang) Institute of Blockchain&Data Security
Abstract
Customizing Pretrained Models (PMs) for downstream tasks demands task-specific data, which burgeoning Web3 applications provide abundantly. Blockchain-powered federated learning (BCFL) offers a secure and privacy-preserving paradigm to customize PMs among self-interested and untrusted Web3 participants. Yet, existing BCFL algorithms face severe efficiency and scalability challenges when applied to large PMs, due to their substantial computational and memory requirements and the hardware constraints of many participants. This paper introduces DICE, a novel BCFL algorithm that addresses the efficiency and scalability challenges of customizing large PMs in Web3 via two sets of synergistic techniques: (i) a dynamic progressive learning scheme and a model slice verification scheme to enable a wider range of honest participants, including those unable to deploy the full model, to engage in customizing PMs, while preserving model utility; (ii) a tree-structured committee consensus protocol and an efficiency-driven incentive scheme to address the dominant bottleneck that hinders existing BCFL algorithms from scaling to a larger number of data holders for customizing data-hungry large PMs, while maintaining security. Theoretical analysis and extensive experimental results on customizing vision and language PMs demonstrate DICE's efficiency, effectiveness, and security.
Figures & tables
| Batch Size | 8 | 16 | 32 | 64 | # Workers | 10 | 40 | 70 | 100 |
| (GB) | 11 | 17 | 30 | 59 | (GB) | 18 | 71 | 125 | 170 |
| Symbol | Definition |
| , , | global round index; total number of rounds; learning rate |
| , | worker’s and validator’s addresses on the blockchain network |
| , | the complete sets of eligible workers and eligible validators |
| workers that have submitted updates in round | |
| scheduled workers; replaced by when training completes | |
| validators that submit VA results in round |
| Settings | Vision Tasks | Language Tasks |
| Type | Image Classification | NER and RE |
| Model | ViT-base (default), DINOv2-large | BERT-base (default), Qwen2.5-0.5B |
| Dataset | FedISIC ( , 2022 ) ‡ , CIFAR100 ( Sharma et al., 2018 ) , DermaMNIST ( Yang et al., 2021 ; Yang et al., 2023b ) | 2018n2c2 ( Lee et al., 2020 ) ‡ , BC5-disease (BC5-d) ( Lee et al., 2020 ) , JNLPBA ( Kim et al., 2004 ) , GAD ( Lee et al., 2020 ) , DDI ( Herrero-Zazo et al., 2013 ) , ChemProt ( Taboureau et al., 2010 ) |
| Data Partition | FedISIC: ; all others: | |
| Further Pretraining | MedMNIST (supervised image classification) | PubMed (self-supervised language modeling) |
| Workers / Validators ( ) | 100 / 1,173 (1,036) | 50 / 678 (599) |
| Hardware Ref. | Hardware Ref. | ||||
| 24 | 1.8 | 16.4 | RTX 4090 24GB | 122.2 | 2 H100 80GB |
| 20 | 1.6 | 13.8 | RX 9070 XT 16GB | 108.3 | 2 H100 80GB |
| 16 | 1.4 | 11.2 | RTX 5070 12GB | 94.5 | RTX Pro 6000 96GB |
| 12 | 1.2 | 8.6 | ARC B570 10GB | 80.6 | 2 RTX A6000 48GB |
| Methods | ACC | F1 | F1 | |||||||||||||
| Vision Tasks | FedISIC | CIFAR100 | DermaMNIST | NER Tasks | n2c2 | JNLPBA | BC5-d | RE Tasks | DDI | GAD | ChemProt | |||||
| BCFL | ✗ | 100.0% | 100.0% | 79.5% | 94.0% | 82.9% | 100.0% | 100.0% | 90.7% | 98.5% | 89.5% | 100.0% | 100.0% | 44.3% | 92.7% | 13.8% |
| +LoRA | 101.6% | 1.7% | 75.7% | 91.5% | 78.3% | 101.2% | 1.6% | 86.4% | 93.5% | 85.2% | 101.2% | 2.2% | 43.8% | 87.4% | 13.1% | |
| +BitFit | 100.0% | 0.2% | 71.5% | 94.8% | 75.2% | 100.0% | 0.1% | 81.6% | 59.3% | 84.5% | 100.0% | 0.8% | 42.2% | 76.3% | 23.0% | |
| +Prune | ✓ | 50.4% | 50.0% | 95.9% | 90.9% | 99.6% | 60.9% | 50.0% | 97.4% | 98.2% | 98.5% | 61.1% | 50.4% | 93.3% | 99.6% | 78.8% |
| +KD | 3.3% | 3.1% | 91.7% | 80.3% | 91.2% | 97.1% | 99.0% | 99.7% | 92.8% | 99.9% | 97.6% | |||||
| Methods | ACC/F1 | ||||||
| FedISIC | CIFAR100 | DermaMNIST | |||||
| DICE | 50.4% | 41.6% | 50.4% | 3.5% | 98.3% | 97.9% | 99.1% |
| +LoRA | 51.0% | 0.6% | 50.4% | 94.0% | 94.3% | 94.6% | |
| +BitFit | 50.4% | 0.1% | 50.4% | 89.1% | 89.2% | 89.7% | |
| +Prune | 50.4% | 50.0% | 50.4% | 96.1% | 90.9% | 98.6% | |
| +KD | 3.3% | 3.1% | 3.3% | 91.9% | 80.1% | 91.2% | |
| Methods | FedISIC (ACC) | n2c2 (F1) | DDI (F1) |
| T-Only | 73.8% | 97.2% | 81.6% |
| B-Only | 95.9% | 96.1% | 86.9% |
| OT | 87.1% | 96.9% | 76.2% |
| DyTB | 91.7% | 97.2% | 93.1% |
| DyBT | 96.0% | 96.1% | 86.9% |
| DyOT | 91.3% | 96.9% | 88.5% |
| FedISIC (ACC) | n2c2 (F1) | DDI (F1) | |
| 0 | 55.3% | 67.1% | 63.8% |
| 1 (Default) | 98.3% | 97.9% | 96.1% |
| 2 | 98.4% | 97.1% | 96.3% |
| Methods | ACC/F1 | |||||
| FedISIC | n2c2 | DDI | FedISIC | n2c2 | DDI | |
| Full-Verify | 98.5% | 98.5% | 96.6% | 0.0% | 0.0% | 0.0% |
| MSV | 98.3% | 97.9% | 96.1% | 0.0% | 0.0% | 0.0% |
| Methods | ACC/F1 | |||||
| FedISIC | n2c2 | DDI | FedISIC | n2c2 | DDI | |
| Fixed | 92.7% | 95.4% | 92.5% | 11.2% | 8.7% | 9.6% |
| Dynamic | 98.3% | 97.9% | 96.1% | 0.0% | 0.0% | 0.0% |
| Methods | All Honest | 20% Malicious | ||||
| FedISIC | n2c2 | DDI | FedISIC | n2c2 | DDI | |
| TCC-only | 6.3% | 10.7% | 10.5% | 90.5% | 88.9% | 89.9% |
| TCC+EDI | 6.1% | 10.5% | 10.6% | 6.1% | 10.6% | 10.5% |