Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Sparsity- and Hybridity-Inspired Visual Parameter-Efficient Fine-Tuning for Medical Diagnosis

About

The success of Large Vision Models (LVMs) is accompanied by vast data volumes, which are prohibitively expensive in medical diagnosis.To address this, recent efforts exploit Parameter-Efficient Fine-Tuning (PEFT), which trains a small number of weights while freezing the rest.However, they typically assign trainable weights to the same positions in LVMs in a heuristic manner, regardless of task differences, making them suboptimal for professional applications like medical diagnosis.To address this, we statistically reveal the nature of sparsity and hybridity during diagnostic-targeted fine-tuning, i.e., a small portion of key weights significantly impacts performance, and these key weights are hybrid, including both task-specific and task-agnostic parts.Based on this, we propose a novel Sparsity- and Hybridity-inspired Parameter Efficient Fine-Tuning (SH-PEFT).It selects and trains a small portion of weights based on their importance, which is innovatively estimated by hybridizing both task-specific and task-agnostic strategies.Validated on six medical datasets of different modalities, we demonstrate that SH-PEFT achieves state-of-the-art performance in transferring LVMs to medical diagnosis in terms of accuracy. By tuning around 0.01% number of weights, it outperforms full model fine-tuning.Moreover, SH-PEFT also achieves comparable performance to other models deliberately optimized for specific medical tasks.Extensive experiments demonstrate the effectiveness of each design and reveal that large model transfer holds great potential in medical diagnosis.

Mingyuan Liu, Lu Xu, Shengnan Liu, Jicong Zhang• 2024

Related benchmarks

TaskDatasetResultRank
Medical DiagnosisCOVID19-CT
F1 Score83
16
Medical Diagnosis ClassificationChaoyang
F1 Score80.6
14
Medical Diagnosis ClassificationBUSI
F1 Score90.5
12
Medical Diagnosis ClassificationCXT3
F1 Score81
12
Medical Diagnosis ClassificationLIUMC
F1 Score72.7
12
Medical Diagnosis ClassificationOCT
F1 Score (%)95.6
12
Showing 6 of 6 rows

Other info

Follow for update