真诚友善团结专业,共建你我引以为荣之社区。《常见问题解答》
1 / 53
12月 17 日

由 ZaneL 发布于 1 天前

这是一个医疗领域的多模态大模型微调案例,希望对各位开发者和爱好者有所帮助。

简介

当前通用多模态视觉语言模型在医疗影像场景中存在显著技术瓶颈:高分辨率医学影像(如CT、MR)的细粒度特征提取能力不足,且模型推理时显存占用高、计算效率低,难以支撑临床实时分析与大规模部署需求。

患者提问:“请使用中文详细描述这张图像并给出你的诊断结果。”

这是微调前模型的回答。虽然能够识别出基本病变,但其分析存在明显不足,描述过于简略,仅关注单一病灶而忽略了图像中实际存在的双肺多发性结节,且诊断结论过于武断,直接定性为"良性肿瘤",缺乏严谨的鉴别诊断思路,临床参考价值有限。

这是微调后模型的回答。它成功化身为“严谨的放射科医生”,不仅准确定位双肺病灶,系统分析肺部结构、心脏大血管和骨骼关系,更能从病灶特征、位置分布和临床意义多个维度进行专业解读,提供完整的鉴别诊断思路,其描述精准、逻辑严密、术语规范。

通过以上对比可以直观地看到,经过高质量数据微调后的模型,成功地从一位“门外汉”进化为了可靠的“AI放射科医生”。

一、项目背景:打破医疗AI的“不可能三角”

当前,通用视觉大模型在医疗影像场景中存在三大瓶颈:

​ ● 细节捕捉弱:难以看懂高分辨率(CT/MR)影像中的微小病灶

​ ● 显存占用高:动辄数十GB的显存需求,边缘设备跑不动,难以临床部署;

​ ● 专业表述差:生成内容缺乏临床术语,可信度低,难以支撑临床实时分析需求。

二、方案设计:稀疏激活 + 高效微调

在医疗场景下,我们面临着“既要马儿跑,又要马儿少吃草”的悖论:

​ ● 要精度: 必须看懂高分辨率CT/MR,参数量不能小(30B级别)

​ ● 要成本: 医院边缘设备显存有限,跑不动庞然大物

我们选择了Qwen3-VL-30B-A3B-Instruct,正是因为它采用了 “稀疏激活(Active 3B)” 架构。它拥有300亿参数的知识储备,但推理时仅激活30亿参数——这为低成本落地埋下了伏笔。

配置参数 配置项 说明
模型 Qwen3-VL-30B-A3B-Instruct 稀疏激活架构,仅激活3B参数,支持高分辨率动态切换,极大节约计算资源
数据集 MedTrinity-25M (16k样本子集) 选取的数据集是MedTrinity-25M子集中的其中一个(16163张图片),MedTrinity-25M是当前规模最大的公开医学影像-文本对数据集,涵盖超过2500万张图像,涉及CT、MR、X-Ray等多种模态,并为65多种疾病提供了多层次的注释
GPU H800 * 4(推荐) 模型规模较大,建议配置足够显存以确保稳定高效训练
微调方法 LoRA 显著降低计算与存储成本,实现大模型的高效轻量化微调

三、训练实战:从数据到可对话的医疗专家

1、数据加工:把“医学教材”喂给AI

高质量、格式规范的数据集是成功的关键。我们通过以下流程将原始医学数据转化为模型可理解的“教材”:

​ ● 下载数据:从MedTrinity-25M数据集中精选1.6万条高质量影像-文本对

​ ● 格式转换:使用定制Python脚本,将原始数据转换为LLaMA-Factory Online支持的 ShareGPT多模态对话格式

​ ● 质量验证:通过随机抽样与基线模型测试验证数据有效性。

:light_bulb:核心代码详解:我们提供了完整的数据格式转换脚本,将原始Parquet数据转换为模型可训练的格式。

#多模态数据格式转换代码 import os import json import random from tqdm import tqdm import datasets def save_images_and_json(ds, ratio=0.1, output_dir="mllm_data"): """ 保存数据集中的图像,并且构建多模态训练集和验证集。 参数: ds: 数据集对象,包含图像和描述。 ratio: 验证集比例,默认为 0.1。 output_dir: 输出目录,默认为 "mllm_data"。 """ # 创建输出目录 os.makedirs(output_dir, exist_ok=True) all_train_data = [] # 多模态训练数据 all_val_data = [] # 多模态验证数据 total_samples = len(ds) val_index = set(random.sample(range(total_samples), int(ratio * total_samples))) # 遍历数据集中的每个项目 for idx, item in tqdm(enumerate(ds), total=total_samples, desc="Processing"): img_path = os.path.join(output_dir, f"{item['id']}.jpg") image = item["image"] # 保存图像 image.save(img_path) sample = { "messages": [ { "role": "user", "content": "<image>图片中的诊断结果是怎样?" }, { "role": "assistant", "content": item["caption"] # 从数据集中获取的描述 } ], "images": [img_path] # 图像文件路径 } if idx in val_index: all_val_data.append(sample) else: all_train_data.append(sample) # 将数据保存到 JSON 文件 train_json_path = os.path.join(output_dir, "mllm_train_data.json") val_json_path = os.path.join(output_dir, "mllm_val_data.json") with open(train_json_path, "w", encoding="utf-8") as f: json.dump(all_train_data, f, ensure_ascii=False, indent=2) with open(val_json_path, "w", encoding="utf-8") as f: json.dump(all_val_data, f, ensure_ascii=False, indent=2) if __name__ == "__main__": # 加载数据集 ds = datasets.load_dataset("parquet", data_files="../dataset/data_sft/train-00000-of-00010.parquet")["train"] # 保存图像并构建多模态训练/验证集 save_images_and_json( ds, ratio=0.2, output_dir="/workspace/user-data/codelab/Qwen3-VL/dataset/mllm_data" )

转换后的数据示例如下:

[ { "messages": [ { "role": "user", "content": "<image>图片中的诊断结果是怎样?" }, { "role": "assistant", "content": "这是一张非增强脑部计算机断层扫描(CT)图像,图中未见任何医疗设备。位于图像中央的兴趣区域可见密度改变,提示脑内出血。该异常区域与周围脑组织分界清晰,可能为脑实质内血肿或出血。其位置及表现可能与邻近脑组织存在关联,进而产生占位效应或导致颅内压升高。" } ], "images": [ "/workspace/user-data/codelab/Qwen3-VL/dataset/mllm_data/8031efe0-1b5c-11ef-8929-000066532cad.jpg" ] }, { "messages": [ { "role": "user", "content": "<image>图片中的诊断结果是怎样?" }, { "role": "assistant", "content": "这是一张脑部非增强计算机断层扫描(CT)图像,显示双侧大脑半球,图中无医疗器械。感兴趣区域位于脑中央偏下方,约占图像面积的 1.1%,表现异常,提示可能存在病变:其密度或纹理改变符合颅内出血特征。该区域与周围脑结构紧邻,可能对邻近组织产生压迫,或受邻近组织影响,提示病变可能正在扩展,并可能影响周边组织功能。" } ], "images": [ "/workspace/user-data/codelab/Qwen3-VL/dataset/mllm_data/803201d1-1b5c-11ef-bba0-000066532cad.jpg" ] }, ]

2、模型训练:找到医疗影像的“学习密码”

训练一个专业模型,不仅是“跑起来”,更要“学得好”。参数调优就是寻找最佳“学习方案”的过程。我们通过严谨的对比实验,揭示了影响医疗影像学习效果的关键因素。

(1)DeepSpeed Stage选择是性能关键

在微调30B级别大模型时,很多人的第一反应是无脑开DeepSpeed Stage 3以节省显存。但在医疗影像这种需要极高精度的任务中,我们通过实战验证了一个残酷的真相:

​ ● 误区(DeepSpeed Stage 3): 虽然显存占用低,但在医疗细粒度特征上,Loss 下降缓慢。原因在于Stage 3的“参数延迟+梯度噪声”机制,干扰了模型对微小病灶的学习

​ ● 正解(DeepSpeed Stage 2): 虽然显存占用稍高,但loss曲线如丝般顺滑,收敛更彻底

:heart:独家心法:在LLaMA-Factory Online配置时,若显存允许(如使用 H800),请果断选择Stage 2。如果必须用Stage 3,请务必配合“放大Global Batch Size+拉长Warmup”的组合拳来弥补性能损失。

(2)参数配置对比实验与分析

为验证上述发现,在任务模式下,我们对模型进行了两组微调实验(参数一和参数二),以评估不同配置的效果。两组实验的变量仅为 per_device_train_batch_size(32,4)和DeepSpeed(3,2)参数,其他条件完全相同。具体参数差异如下表所示:

配置参数 参数说明 参数一 参数二
基础配置
model 训练用的基模型 Qwen3-VL-30B-A3B-Instruct Qwen3-VL-30B-A3B-Instruct
dataset 训练使用的数据集名称 mllm_train_data mllm_train_data
stage 训练方式 sft sft
finetuning_type 微调方法 lora lora
进阶配置
LR Scheduling Type 动态调整学习率的方式 cosine cosine
Max Gradient Norm 梯度裁剪的最大范数,用于防止梯度爆炸 1.0 1.0
训练配置
Learning Rate 学习率 5e-05 5e-05
Epochs 训练轮数 2 2
per_device_train_batch_size 单GPU批处理大小 32 4
Gradient Accumulation 梯度累计,将一个完整批次的梯度计算拆分为多个小批次,逐步累积梯度,最后统一更新模型参数 8 8
Save steps 训练过程中每隔多少个训练步保存一次模型 200 200
Warmup Ratio 将学习率从零增加到初始值的训练步数比例 0 0
Chat Template 基模型的对话模版,训练和推理时构造prompt的模版 qwen3 qwen3
效率与性能配置
Mixed Precision Train 混合精度训练,模型在训练或推理时所使用的数据精度格式,如 FP32、FP16或BF16 bf16 bf16
分布式配置
DeepSpeed Deepspeed Stage是DeepSpeed中ZeRO(Zero Redundancy Optimizer)优化技术的阶段参数,其范围是none、2、3。参数越大,意味着模型状态的分片程度越高,每个GPU的内存占用越少,但同时通信开销也可能越大 3 2
数据参数配置
Max Sample Size 每个数据集的最大样本数:设置后,每个数据集的样本数将被截断至指定的max_samples 100000 100000
Cutoff Length 输入的最大token数,超过该长度会被截断 2048 2048
Preprocess Workers 预处理时使用的进程数量 32 32
日志配置
Logging Steps 日志打印步数 5 5
LoRA配置
LoRARank LoRA微调的本征维数 r,r 越大可训练的参数越多 8 8
LoRAScalling Factor LoRA缩放系数。一般情况下为 lora_rank * 2 16 16
Random dropout LoRA微调中的dropout率 0 0
LoRAModules Lora作用模块 all all

通过任务模式完成两组参数配置的模型微调后,从loss对比结果来看,相同硬件与数据集条件下,deepseed 3(参数一)方案训练速度更快,但微调阶段loss显著上升;deepseed 2(参数二)方案虽训练速度略有下降,却能更有效地压低loss。具体来看:

​ ● deepseed 3训练速度的提升,核心得益于 “小块通讯 + 微批次自动放大” 带来的带宽优化;

​ ● deepseed 3微调loss上涨的本质,是 “参数延迟 + 梯度噪声” 导致模型收敛效果变差;

:star:选型建议:若显存充足,优先选择deepseed 2方案以追求更优指标;若显存不足需使用deepseed 3,则需同步通过放大global batch、拉长 warmup时长、降低学习率(lr)的方式弥补收敛性能。

通过反复实验,我们总结出了一套适用于Qwen3-VL医疗微调的参数心法

​ ● LR Scheduler(学习率调度): 放弃Linear!在多模态图文对齐任务中,Linear衰减表现平平。请选择 Cosine + Warmup,它能更好地适配视觉特征的学习节奏

​ ● Epoch(训练轮数):在16k数据场景下,3个Epoch是性能拐点;第4个Epoch起训练Loss仍降,但验证指标不再上升,属于典型过拟合;5k 小数据场景下可拉到6~8Epoch

​ ● LoRARank:医疗影像细节极多(如微小结节、毛刺征),低Rank(如8以下)表达能力不足。Rank 32是效果与成本的性价比拐点

​ ● Alpha值: 死磕公式Alpha = Rank×2,稳定性最佳

​ ● dropout:数据量 ≤ 10k时,设置dropout=0.05 可有效防过拟合;数据 > 10k:可直接设为0

3、效果验证:从“业余”到“专业”的飞跃

经过精心的微调,模型的性能实现了质的飞跃。我们通过量化指标和定性分析,全方位评估其提升效果。

(1)指标对比:数十倍至上千倍的提升

下面的数据清晰地展示了模型在微调前后的巨大变化。其中,参数二(DeepSpeed Z2方案) 在各项文本生成质量指标上达到了最优水平。

评估指标 微调前(原生模型) 参数一微调后 (Z3方案) 参数二微调后(Z2方案)
BLEU-4 0.806 27.653 92.375
ROUGE-1 2.778 38.069 96.114
ROUGE-2 0.006 16.363 94.036
ROUGE-L 2.013 20.695 94.286

指标解读

​ ● BLEU-4衡量生成文本与专业参考答案在词组和表达上的匹配度

​ ● ROUGE-1/2/L综合评估生成内容的关键词覆盖、短语搭配和句法连贯性

结论一目了然:采用Z2方案微调的模型(参数二),其生成质量远超原生模型和Z3方案,在专业术语、句式结构和临床逻辑上都与标准医学描述高度一致。

(2)生成质量:从“无法使用”到“专业优秀”

​ ● 微调前(原生模型):各项指标极低,生成内容与参考答案关联性微弱,逻辑混乱,完全无法满足专业场景需求

​ ● 微调后(参数二模型)

​ ○ BLEU-4高达92.37,意味着模型能精准复现医学报告中的专业词汇与表达

​ ○ ROUGE系列指标均超过94,代表其在关键词捕捉、专业短语运用和长篇报告的连贯性上表现出色

​ ○ 生成文本的质量已达到优秀级别,具备临床应用的潜力

(3)效率提升:速度与精度的双重胜利

除了生成质量,推理效率也得到显著优化。

评估指标 微调前(原生模型) 参数一微调后 (Z3方案) 参数二微调后(Z2方案)
predict_samples_per_second 0.773 0.057 0.194
predict_steps_per_second 0.048 0.002 0.048
predict_runtime 4179.834 56431.560 16668.369

微调不仅解决了原生模型生成质量“不可用”的核心问题,更在效率上实现了超越。最终得到的模型在专业性准确性响应速度上取得了完美平衡,可立即投入医学影像报告生成、辅助诊断等严肃多模态场景。

4、实战对话:真正的“AI放射科医生”

模型性能的最终检验标准在于实战。我们对比了参数一(Z3方案)参数二(Z2方案)微调后的模型对同一张胸部CT影像的分析,结果显示两者均达到专业水准,但在分析的全面性细致程度诊断深度上存在显著差异。

通过对比分析,我们验证了一个重要结论,参数二(Z2方案) 在以下方面表现显著更优:

​ ● 观察敏锐度:能够发现图像中的多个病灶,避免漏诊

​ ● 分析系统性:提供从解剖结构到病变特征的完整分析框架

​ ● 诊断严谨性:基于医学证据进行推理,给出合理的鉴别诊断

​ ● 临床实用性:回答具有直接临床参考价值

这一结果与我们之前的实验数据高度吻合——Z3方案虽然在训练速度上稍慢,但能够学习到更丰富的医学知识结构和诊断逻辑,最终生成的影像报告更接近资深放射科医生的专业水准。

阅读时间 4 分钟

由 yqyan 发布于 1 天前

由 Vincent_Henry 发布于 1 天前

由 wangtianyi 发布于 1 天前

由 byl520 发布于 1 天前

标注一下,我也打算微调一个来做毕业论文 奈何是小白啥也不会

由 handsome 发布于 1 天前

由 crane 发布于 1 天前

由 RerrentLinden 发布于 1 天前

太强了佬友,
之前站内有医生(记得是甲状腺相关的)也有医疗影像相关科研需求

由 zone 发布于 1 天前

由 Ceia 发布于 1 天前

由 kill 发布于 23 小时前

由 iheeleme 发布于 23 小时前

由 Bamboo_OvO 发布于 23 小时前

由 blacsheep 发布于 23 小时前

好奇问一下, 30B-A3B, 虽然说是只激活3B, 但是实际场合不是30B也都得加载到显存吗? 只不过说推理的时候我们只需要3B参数, 推理速度理论和3B模型类似, 但是显存占用应该是没有节省吧

由 deep_water 发布于 23 小时前

由 user2151 发布于 23 小时前

由 ZaneL 发布于 23 小时前

参数是都要加载到显存中的。感谢指正,文中描述是有不准确的地方。

由 blacsheep 发布于 23 小时前

由 ZaneL 发布于 17 小时前

先找个简单的、手把手的教程,跟着做一遍找找感觉 :laughing:

由 ZaneL 发布于 17 小时前

由 VincentChen 发布于 17 小时前

由 chengzi2832 发布于 17 小时前

由 linuxbi 发布于 17 小时前

由 ndjasnd 发布于 16 小时前

佬友,项目有再github开源吗? 想去学习一下

由 ZaneL 发布于 16 小时前

还没有开源成项目哈,其实核心内容都在这里了。
如果要更细操作步骤的可以去翻下LlamaFactory Online的最佳实践。

由 sunfly 发布于 16 小时前

让我来学习一下,后面动手撸一把:nerd_face:

由 deep_water 发布于 16 小时前

由 sunfly 发布于 16 小时前

佬友有没有微调数据集公开呀:hugs:

由 Melo 发布于 15 小时前

感谢佬的无私分享,太吓人了,bleu和rouge指标掉这么多。请问佬有在其他微调案例中做过关于batch size或者说zero参数的对比实验吗?特别是文本模型训练,我一直用小batch size+zero3省显存。

由 XiaoheiA 发布于 15 小时前

由 bosunjiang 发布于 15 小时前

由 ZaneL 发布于 15 小时前

由 ZaneL 发布于 15 小时前

由 ZaneL 发布于 15 小时前

语言模型好像没专门做过对比实验。你是什么场景呢,我们可以找一个跑跑看。

由 mlm404 发布于 15 小时前

由 ByteBender 发布于 15 小时前

由 jovin 发布于 15 小时前

感谢佬分享,请问一下这个需要多大的显存呢?大模型微调有没有入门的好教程~

由 sunfly 发布于 14 小时前

由 wei33 发布于 14 小时前

感谢佬友分享,正好为现在的研究提供了下思路

由 Twickenham 发布于 14 小时前

tada

这是 Twickenham 首次发帖 - 让我们欢迎他/她加入社区吧!

在使用 Offloading 技术的情况下能够通过牺牲推理速度来换取在更小显存设备上部署,这种情况是可以降低显存占用的

加载下方更多的帖子