通过生成硬负例提升大模型对几何细节的理解能力
MMGeoLM: Hard Negative Contrastive Learning for Fine-Grained Geometric Understanding in Large Multimodal Models
- 用代码扰动生成图像硬负例,结合规则与检索构造文本硬负例
- 7B模型在三个几何推理任务上超越开源模型,媲美GPT-4o
- 验证了硬负例类型与训练配置对几何细粒度理解的关键作用
大型多模态模型(LMMs)通常基于视觉变换器(如CLIP),但其使用随机批内负例的训练方式难以捕捉细微视觉差异,尤其在几何场景中表现不足。为此,我们提出一种新颖的硬负例对比学习框架,用于视觉编码器:结合基于生成的图像硬负例(通过扰动图表生成代码获得)和基于规则的文本硬负例(修改几何描述)及基于检索的文本负例(根据标题相似性选择)。我们采用该方法训练视觉编码器(CLIP),得到MMCLIP,并进一步训练用于几何问题求解的LMM。实验表明,所训练的模型MMGeoLM在三个几何推理基准测试中显著优于其他开源模型,即使仅7B参数量,也能媲美GPT-4o等闭源强模型。我们还进行了消融研究,分析三类关键因素:硬负例类型、图像硬负例效率以及训练配置,为优化视觉编码器在细粒度几何推理任务中的训练流程提供了重要洞察。
原文摘要 · Abstract (English)
Large Multimodal Models (LMMs) typically build on ViTs (e.g., CLIP), yet their training with simple random in-batch negatives limits the ability to capture fine-grained visual differences, particularly in geometric scenarios. To address this challenge, we propose a novel hard negative contrastive learning framework for the vision encoder, which combines image-based contrastive learning using generation-based hard negatives created by perturbing diagram generation code, and text-based contrastive learning using rule-based negatives derived from modified geometric descriptions and retrieval-based negatives selected based on caption similarity. We train a vision encoder (CLIP) using our hard negative training method, namely MMCLIP (Multimodal Math CLIP), and subsequently train an LMM for geometric problem-solving. Experiments show that our trained model, MMGeoLM, significantly outperforms other open-source models on three geometric reasoning benchmarks. Even with a size of 7B, it can rival powerful closed-source models like GPT-4o. We further conduct ablation studies to analyze three key factors: hard negative types, the efficiency of image-based negatives, and training configurations. These analyses yield important insights into optimizing the training pipeline of vision encoder for fine-grained geometric reasoning tasks. https://github.com/THU-KEG/MMGeoLM.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。