Adapting 2D Multi-Modal Large Language Model for 3D CT Image Analysis
Organizations: Department of Electronic and Computer Engineering, The Hong Kong University of Science and Technology, Hong Kong, China. · Department of Computer Science and Engineering, The Chinese University of Hong Kong, Hong Kong, China. · Center for Artificial Intelligence and Robotics, Hong Kong Institute of Science and Innovation, Chinese Academy of Sciences, Hong Kong, China. · Institute of Medical Intelligence and XR, The Chinese University of Hong Kong, Hong Kong, China.
Abstract
3D medical image analysis is of great importance in disease diagnosis and treatment. Recently, multimodal large language models (MLLMs) have exhibited robust perceptual capacity, strong cross-modal alignment, and promising generalizability. Therefore, they have great potential to improve the performance of medical report generation (MRG) and medical visual question answering (MVQA), which serve as two important tasks in clinical scenarios. However, due to the scarcity of 3D medical images, existing 3D medical MLLMs suffer from insufficiently pretrained vision encoder and inability to extract customized image features for different kinds of tasks. In this paper, we propose to first transfer a 2D MLLM, which is well trained with 2D natural images, to support 3D medical volumetric inputs while reusing all of its pre-trained parameters. To enable the vision encoder to extract tailored image features for various tasks, we then design a Text-Guided Hierarchical MoE (TGH-MoE) framework, which can distinguish tasks under the guidance of the text prompt. Furthermore, we propose a two-stage training strategy to learn both task-shared and task-specific image features. As demonstrated empirically, our method outperforms existing 3D medical MLLMs in both MRG and MVQA tasks. Our code will be released once this paper is accepted.
Figures & tables
| Method | Model Size | Medical VQA | Report Generation | |||||||
|---|---|---|---|---|---|---|---|---|---|---|
| BLEU-1 | ROUGE-1 | METEOR | BERT | Accuracy | BLEU-1 | ROUGE-1 | METEOR | BERT | ||
| LLAVA-Med-1.5 | 7B | 34.53 | 36.54 | 25.43 | 89.06 | 64.41 | 14.25 | 16.08 | 12.74 | 82.53 |
| RadFM | 14B | 16.39 | 26.13 | 21.33 | 88.72 | 19.79 | 12.23 | 16.49 | 11.57 | 87.93 |
| M3D | 4B | 52.10 | 55.19 | 35.89 | 92.14 | 79.35 | 13.90 | 18.02 | 12.77 | 83.63 |
| Med3DVLM | 7B | 47.54 | 50.98 | 32.68 | 91.32 | 76.89 | 13.86 | 18.01 | 12.64 | 83.70 |
| Ours | 4B | 54.64 | 57.97 | 37.85 | 92.57 | 81.06 | 14.23 | 19.03 | 13.40 | 84.03 |
| No. | 2D to 3D Adap. | TGH-MoE | Two-stage Train. Stra. | Medical VQA | Report Generation | |||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| B-1 | R-1 | M | BERT | Accuracy | B-1 | R-1 | M | BERT | ||||
| 1 | 52.10 | 55.35 | 35.62 | 92.15 | 78.41 | 14.67 | 19.54 | 13.97 | 83.88 | |||
| 2 | 52.83 | 56.05 | 36.44 | 92.30 | 78.96 | 14.78 | 19.48 | 14.08 | 83.94 | |||
| 3 | 52.87 | 56.08 | 36.43 | 92.29 | 79.11 | 14.73 | 19.87 | 13.87 | 83.98 | |||
| 4 | 52.74 | 55.95 | 36.20 | 92.30 | 79.13 | 14.69 | 20.02 | 14.09 | 84.05 | |||
| 5 | 53.02 | 56.17 | 36.52 | 92.33 | 79.28 | 14.91 | 19.55 | 14.11 | 83.92 | |||
| Design Choices | Medical VQA | Report Generation | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|
| Description | MoE Type | B-1 | R-1 | M | BERT | Accuracy | B-1 | R-1 | M | BERT | |
| Only Task-shared | 1 | Token-level | 52.77 | 56.05 | 36.29 | 92.29 | 79.24 | 14.96 | 20.03 | 14.07 | 84.06 |
| Only Task-specific | 1 | TGH-MoE | 53.02 | 56.17 | 36.52 | 92.33 | 79.28 | 14.91 | 19.55 | 14.11 | 83.92 |
| Finetune LLM in Stage 2 | 2 | TGH-MoE | 52.73 | 55.98 | 36.47 | 92.26 | 79.29 | 15.10 | 19.62 | 13.91 | 83.89 |
| Task-shared and -specific (Ours) | 2 | TGH-MoE | 53.10 | 56.39 | 36.61 | 92.33 | 79.32 | 15.10 | 20.31 | 14.23 | 84.12 |
Multimodal Large Language Models
Multimodal large language models combine language models with image, video, audio, and other inputs for grounded understanding and generation.
See all 1,091 papers