Parameter Efficient Fine-tuning for Domain-specific Gastrointestinal Disease Recognition
Organizations: Auburn University · Tribhuvan University · The University of Texas at Austin · Georgia State University · Middle Tennessee State University
Abstract
Despite recent advancements in the field of medical image analysis with the use of pretrained foundation models, the issue of distribution shifts between cross-source images largely remains adamant. To circumvent that issue, investigators generally train a separate model for each source. However, this method becomes expensive when we fully fine-tune pretrained large models for a single dataset, as we must store multiple copies of those models. Thus, in this work, we propose using a low-rank adaptation (LoRA) module for fine-tuning downstream classification tasks. LoRAs learn lightweight task-specific low-rank matrices that perturb pretrained weights to optimize those downstream tasks. For gastrointestinal tract diseases, they exhibit significantly better results than end-to-end finetuning with improved parameter efficiency. Code is available at: github.com/sanjay931/peft-gi-recognition.
Figures & tables
| Train \ Test | GastroVision | Kvasir-Capsule |
|---|---|---|
| GastroVision [ 11 ] | 86.31 | 8.04 |
| Kvasir-Capsule [ 16 ] | 10.12 | 99.05 |
| GastroVision | Kvasir-Capsule | ||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|
| Method | #Params. | Prec. | Rec. | F1 | Acc. | MCC | Prec. | Rec. | F1 | Acc. | MCC |
| MobileNetV2 [ 15 ] | 3.5M | 74.74 | 76.67 | 74.53 | 76.67 | 73.88 | 95.34 | 95.49 | 95.22 | 95.49 | 90.51 |
| Inception V3 [ 18 ] | 25.4M | 79.87 | 80.44 | 79.31 | 80.44 | 78.17 | 97.27 | 97.41 | 97.32 | 97.41 | 94.57 |
| DenseNet121 [ 9 ] | 7.5M | 82.57 | 83.17 | 82.12 | 83.17 | 81.14 | 97.42 | 97.59 | 97.47 | 97.59 | 94.93 |
| DenseNet201 [ 9 ] | 19.1M | 83.14 | 82.91 | 81.82 | 82.91 | 80.92 | 98.62 | 98.60 | 98.57 | 98.60 | 97.07 |
| ConvNeXtV2 [ 21 ] + Swin-T [ 13 ] | 28.3M | 73.92 | 74.77 | 73.87 | 74.77 | 71.75 | 95.92 | 95.97 | 95.80 | 95.97 | 91.51 |