arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2409.13345cs.CVcs.AI
收藏

多模态模型的新型自适应微调算法:遥感中高质量数据集的自优化分类与选择

A Novel Adaptive Fine-Tuning Algorithm for Multimodal Models: Self-Optimizing Classification and Selection of High-Quality Datasets in Remote Sensing

Hangzhou Institute of Technology of Xidian University(西安电子科技大学杭州研究院);Xidian University(西安电子科技大学);Key Laboratory of Coal Resources Exploration and Comprehensive Utilization, Ministry of Natural Resources(自然资源部煤炭资源勘查与综合利用重点实验室);Shaanxi Water Development Group Co., Ltd(陕西水务发展集团有限公司)

Yi Ren, Tianyi Zhang, Zhixiong Han, Weibin Li, Zhiyang Wang, Wenbo Ji, Chenhao Qin, Chenbin Liang, Licheng Jiao

AI总结:

本文提出一种针对多模态大模型的两阶段截断自适应微调算法,通过MiniBatchKMeans聚类与平移差异计算筛选高泛化数据,仅用三分之一数据即可在保持性能的同时大幅减少训练时间。

摘要:

我们提出了一种针对多模态大模型的自适应微调算法。该算法的核心步骤包含两个截断阶段。首先,将海量数据投影到语义向量空间,并使用MiniBatchKMeans算法进行自动聚类。这种分类确保了每个簇内的数据展现出高度的语义相似性。接下来,我们处理每个簇中的数据,计算在多模态大模型的向量空间中原始数据与扰动数据之间的平移差异。该差异作为数据的泛化指标。基于此指标,我们选择具有高泛化潜力的数据进行训练。我们在两块3090 GPU上,使用GeoChat多模态遥感数据集的三分之一数据,应用该算法训练了InternLM-XComposer2-VL-7B模型。结果表明,我们的算法优于各种最先进的基线方法。基于实验验证,在我们优选的三分之一数据集上训练的模型,与在完整数据集上训练的模型相比,在各种遥感指标上的性能仅下降了1%。该方法在显著保留通用能力的同时,将训练时间减少了68.2%。此外,该模型在UCMerced和AID评估数据集上分别取得了89.86和77.19的分数,分别超过GeoChat数据集5.43和5.16分。它在LRBEN评估数据集上仅表现出0.91分的平均下降。

英文摘要:

We propose an adaptive fine-tuning algorithm for multimodal large models. The core steps of this algorithm involve two stages of truncation. First, the vast amount of data is projected into a semantic vector space, and the MiniBatchKMeans algorithm is used for automated clustering. This classification ensures that the data within each cluster exhibit high semantic similarity. Next, we process the data in each cluster, calculating the translational difference between the original and perturbed data in the multimodal large model's vector space. This difference serves as a generalization metric for the data. Based on this metric, we select the data with high generalization potential for training. We applied this algorithm to train the InternLM-XComposer2-VL-7B model on two 3090 GPUs using one-third of the GeoChat multimodal remote sensing dataset. The results demonstrate that our algorithm outperforms the state-of-the-art baselines. various baselines. The model trained on our optimally chosen one-third dataset, based on experimental validation, exhibited only 1% reduction in performance across various remote sensing metrics compared to the model trained on the full dataset. This approach significantly preserved general-purpose capabilities while reducing training time by 68.2%. Furthermore, the model achieved scores of 89.86 and 77.19 on the UCMerced and AID evaluation datasets, respectively, surpassing the GeoChat dataset by 5.43 and 5.16 points. It only showed a 0.91-point average decrease on the LRBEN evaluation dataset.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑