AdaLoRA-QAT: Adaptive Low-Rank and Quantization-Aware Segmentation
Organizations: International Institute of Information Technology (IIIT-H), Hyderabad, India · Nizam’s Institute of Medical Sciences (NIMS), Hyderabad, India · The Alan Turing Institute, London and University College London, United Kingdom
Abstract
Chest X-ray (CXR) segmentation is an important step in computer-aided diagnosis, yet deploying large foundation models in clinical settings remains challenging due to computational constraints. We propose AdaLoRA-QAT, a two-stage fine-tuning framework that combines adaptive low-rank encoder adaptation with full quantization-aware training. Adaptive rank allocation improves parameter efficiency, while selective mixed-precision INT8 quantization preserves structural fidelity crucial for clinical reliability. Evaluated across large-scale CXR datasets, AdaLoRA-QAT achieves 95.6% Dice, matching full-precision SAM decoder fine-tuning while reducing trainable parameters by 16.6\times and yielding 2.24\times model compression. A Wilcoxon signed-rank test confirms that quantization does not significantly degrade segmentation accuracy. These results demonstrate that AdaLoRA-QAT effectively balances accuracy, efficiency, and structural trust-worthiness, enabling compact and deployable foundation models for medical image segmentation. Code and pretrained models are available at: https://prantik-pdeb.github.io/adaloraqat.github.io/
Figures & tables
| Model | DSC | IOU | NSD |
|---|---|---|---|
| nnU-Net | 0.9485 0.119 | 0.9184 0.148 | 0.8118 0.233 |
| DeepLabv3+ | 0.9536 0.117 | 0.9269 0.142 | 0.8405 0.213 |
| SegFormer | 0.9228 0.174 | 0.8890 0.202 | 0.7622 0.237 |
| SAM Decoder | 0.9555 0.0664 | 0.9210 0.0998 | 0.9279 0.0956 |
| SAM Ada-LoRA + D-QAT ∗ | 0.9563 0.0651 | 0.9224 0.0984 | 0.9293 0.0947 |
| SAM Ada-LoRA + Full QAT † | 0.9559 0.0647 | 0.9216 0.0983 | 0.9286 0.0938 |
| Model | Total | Trainable | Reduction | DSC |
|---|---|---|---|---|
| (M) | (M) | (%) | ||
| nnU-Net | 30.0 | 30.0 | – | 94.85 |
| DeepLabV3+ | 41.0 | 41.0 | – | 95.36 |
| SegFormer | 84.6 | 84.6 | – | 92.28 |
| SAM Decoder FT | 89.7 | 3.8 | 23.6× | 95.55 |
| Ours (SAM Ada-LoRA) | 89.7 | 5.4 | 16.6× | 95.60 |
| Method | Encoder | Decoder | Train (%) | Dice (%) |
|---|---|---|---|---|
| Decoder-only FT | Frozen | Full | 4.2 | 95.55 |
| Enc.-only LoRA (r=8) | LoRA | Frozen | 1.9 | 70.86 |
| Hybrid (Stage 1) | AdaLoRA | Full | 6.1 | 95.60 |
| Hybrid + D-QAT (Stage 2) | AdaLoRA | INT8 | 6.1 | 95.63 |
| Hybrid + Full QAT | AdaLoRA and INT8 | INT8 | 6.1 | 95.59 |
Medical Image Segmentation
Medical image segmentation delineates organs, lesions, and other structures in scans to support measurement, diagnosis, treatment planning, and research.
See all 788 papers