用批量增强和单模态微调提升超声与文本联合检测胎儿器官的准确率
Batch Augmentation with Unimodal Fine-tuning for Multimodal Learning
- 先用医学图像数据预初始化模型,再通过批量增强和单模态微调优化特征提取
- 在FPU23和UPMC Food-101数据集上达到接近顶尖性能,尤其在食品图像多模态任务中表现突出
- 适合医学影像与多模态学习研究者,代码开源便于复现与改进
本文提出一种结合批量增强与单模态微调的多模态学习方法,用于从超声图像及其关联临床文本中检测胎儿器官。首先,利用医学图像数据对模型初始层进行迁移初始化;随后,在图像批次中应用批量增强技术,并对初始层进行微调以提取特征。同时,从图像描述中提取文本信息,与图像特征融合后训练分类头层。设计专用数据加载器,为每批数据引入随机增强策略,提升泛化能力。实验基于FPU23超声与UPMC Food-101多模态数据集,所提方法在多模态大语言模型上取得最佳性能,在UPMC Food-101上达到接近当前最优(SOTA)结果。相关代码已公开于github.com/dipuk0506/multimodal。
原文摘要 · Abstract (English)
This paper proposes batch augmentation with unimodal fine-tuning to detect the fetus's organs from ultrasound images and associated clinical textual information. We also prescribe pre-training initial layers with investigated medical data before the multimodal training. At first, we apply a transferred initialization with the unimodal image portion of the dataset with batch augmentation. This step adjusts the initial layer weights for medical data. Then, we apply neural networks (NNs) with fine-tuned initial layers to images in batches with batch augmentation to obtain features. We also extract information from descriptions of images. We combine this information with features obtained from images to train the head layer. We write a dataloader script to load the multimodal data and use existing unimodal image augmentation techniques with batch augmentation for the multimodal data. The dataloader brings a new random augmentation for each batch to get a good generalization. We investigate the FPU23 ultrasound and UPMC Food-101 multimodal datasets. The multimodal large language model (LLM) with the proposed training provides the best results among the investigated methods. We receive near state-of-the-art (SOTA) performance on the UPMC Food-101 dataset. We share the scripts of the proposed method with traditional counterparts at the following repository: github.com/dipuk0506/multimodal
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。