← 返回 2026-06-05

MechVQA:面向机械工程图理解的多模态大语言模型基准与增强方法 MechVQA: Benchmarking and Enhancing Multimodal LLMs on Comprehensive Mechanical Drawing Understanding

Qian Kou, Xiaofeng Shi, Yulin Li, Xiaosong Qiu, Xinyang Wang, Hua Zhou, Cao Dongxing 📅 2026-05-29 👍 5 2026-07-13 08:36
DAPO 多模态大语言模型 强化学习 机械工程图理解 领域专用基准

首个机械工程图理解基准数据集与专用模型,提升多模态LLM在工程制图中的能力

前置知识

机械工程图投影原理

机械工程图采用正投影法,通过主视、俯视、侧视等多个视图来完整表达三维物体的几何形状和尺寸信息。不同视图之间必须保持投影一致性,即主视与俯视长度对齐、主视与左视高度对齐、俯视与左视宽度相等。理解投影规则是跨视图空间推理的基础,例如需要知道一个特征在主视图中的位置如何对应到侧视图。

MechVQA的Projection & Multi-view子任务要求模型理解视图间的对应关系,无法建立投影一致性就无法正确回答跨视图问题,比如'前视图中的某个圆孔在侧视图对应哪个区域'。

几何尺寸与公差(GD&T)

GD&T是一种标准化的工程符号语言,用于精确控制零件的几何特征,如位置度、圆度、平面度等。它使用基准符号(如Ⓑ、Ⓒ)、特征控制框和公差值来定义零件的可接受变化范围。例如位置度符号⌖ 0.1 Ⓑ Ⓒ表示相对于基准B和C,被测特征的位置偏差不得超过0.1毫米。GD&T比传统尺寸标注更精确,但也更复杂。

MechVQA的Dimension & Annotation和Judging子任务涉及GD&T符号解读,模型必须理解这些专业符号的含义及其与零件几何的关系,否则无法判断图纸标注的正确性或回答公差相关问题。

DAPO(Decoupled Clip and Dynamic Sampling Policy Optimization)

DAPO是一种基于组的强化学习算法,建立在GRPO(Group Relative Policy Optimization)基础上,引入了非对称裁剪、动态采样、token级策略梯度和过长奖励塑造。它通过采样一组响应(而非单个),计算组归一化优势,使用非对称裁剪将策略更新限制在合理范围内,动态采样保留正负奖励样本,避免退化组。DAPO特别适合长推理链任务,如机械工程图的多步几何计算。

MechVL模型使用DAPO进行第二阶段训练,DAPO是模型性能大幅提升的关键。理解DAPO有助于明白论文如何通过强化学习解决SFT-only基线的失效问题,如遗漏关键标注、违反多视图一致性等。

链式计算(Dimension Chain Calculation)

链式计算是通过尺寸链推导未标注尺寸或验证标注合理性的方法。尺寸链是封闭环和组成环构成的尺寸系统,封闭环的尺寸等于所有组成环的代数和。在机械设计中,当一个特征尺寸未直接标注时,需要通过相关联的已知尺寸计算得出,例如计算总长度减去两个子长度得到中间段长度。这要求理解零件的几何结构和尺寸间的依赖关系。

MechVQA的Geometric Calculation子任务涉及链式计算,如根据标注尺寸推导未标注尺寸,这是检验模型是否理解机械工程约束的关键能力,也是中等难度问题的主要类型。

研究动机

现有的多模态大语言模型在机械工程图理解任务上表现脆弱,主要表现在三个方面:首先是高密度标注问题,机械工程图包含密集的尺寸标注、符号标注、技术要求等,关键线索容易被模型忽略,导致遗漏重要信息;其次是缺乏领域知识,模型不理解机械制图标准(如GB、ISO)、投影规则、几何公差等专业概念,导致误读符号或违反投影一致性;第三是空间关系推理不可靠,在严格投影规则和几何约束下,模型无法正确建立视图间的对应关系,例如主视图的某个特征无法准确映射到侧视图,这种错误在复杂的装配图中更为明显。论文中提到,通用MLLMs在密集机械图上的主要失效模式不是光学识别错误,而是由于缺乏领域先验和不可靠的空间推理导致的结构不一致性。

本文的目标是本文旨在构建一个综合性的机械工程图理解基准,并建立领域专用的强基线模型。具体目标包括:创建一个包含真实零件图和装配图的基准数据集,系统性地评估MLLMs在工程制图理解方面的能力;设计细粒度的任务分类(识别、推理、判断)和难度分级,从直接感知到专家级推理进行分层评估;开发领域专用的训练方法,通过多阶段后训练提升模型在高密度标注、跨视图一致性、约束敏感性等挑战下的性能;最终为机械设计和检验工作流中的MLLM部署提供可复用的技术基础。

与已有工作不同的是,本文的独特切入角度在于系统性地填补了机械工程图理解基准的空白。现有的工程领域多模态基准虽然存在,但范围有限:AECV-Bench关注建筑工程绘图和空间OCR,BlueprintSymVL专注蓝图符号识别,CReFT-CAD探索三视图推理,PHT-CAD聚焦参数化图元分析,MechBench通过示意图探测物理定律,DesignQA集成专业文档。这些基准要么只覆盖工程图理解的切片(如符号识别),要么工程图分析仅是更大套件中的小部分且不公开。更重要的是,缺乏一个统一评估机械零件和复杂装配图理解的基准,这些基准未能同时涵盖结构化感知、多视图一致性和工程级推理。MechVQA是首个全面覆盖机械工程图理解的数据集,从真实来源构建,包含10个细粒度子任务,这是与现有工作的本质区别。

核心方法

方法整体思路分为两个层面:数据集构建和模型训练。数据集层面,采用半自动化构建流水线,从公开来源(机械教材、专业手册、设计平台)收集工程图,经过专家过滤和预处理,然后通过多轮交叉模型问答生成、多数投票质量检查和专家审计,最终得到高质量的问答对。模型层面,基于Qwen3-VL-4B-Instruct初始化,采用两阶段训练范式:第一阶段是监督微调(SFT),在MechVQA训练集上学习指令遵循和领域知识,冻结视觉编码器和投影层,仅对语言模型模块进行全参数微调;第二阶段是强化学习(RL),使用DAPO算法在自我博弈方式下进一步提升可靠性,通过精心设计的复合奖励(准确率、格式、质量)直接优化答案正确性、输出格式合规性和解释质量。

核心创新点有两个:其一是构建了首个系统性的机械工程图理解基准MechVQA,涵盖3个能力维度(识别、推理、判断)和10个细粒度子任务,建立了分层能力评估框架;其二是采用与分类学对齐的奖励设计进行强化学习,复合奖励结合了语义容忍的准确率奖励(通过LLM-as-a-Judge评估)、严格的格式奖励和基于逻辑性、专业性、简洁性的质量奖励,这种设计直接针对密集机械图上的常见失效模式(遗漏关键标注、跨视图不一致、违反约束的错误答案),使RL训练能够显著提升模型的工程制图理解能力。这与传统仅基于字符串匹配或token级F1的奖励有本质区别。

方法步骤详情

方法步骤的完整描述如下:数据构建阶段,首先从公开来源收集工程图并过滤低质量、不完整、扫描质量差的样本,由领域专家人工审查,得到3,281张高质量图纸。接着使用先进OCR模型提取文本内容(如表格条目),并利用强闭源MLLMs推断其他原始元数据字段,机械专业研究生进行二次验证,检查字段包括视图类别定义、工作流程和不确定案例说明。问答生成采用三种来源:VQA自由生成使用闭源MLLMs根据子任务定义生成候选问题,通过迭代交叉模型检查(不同模型作为验证器检查问题是否可回答)和多模型回答多数投票;基于模板的生成(无真值)通过将模板实例化为图纸内容生成问题,如定位多个尺寸标注后绑定符号类型和引用特征;基于模板的生成(有真值)使用经专家验证的元数据生成确定性问题(如查询视图数量)或包含专家编辑的问题(如CAD工具注入不一致性)。所有答案都附带详细解释和最终简短答案格式。最后按8:1:1比例划分训练/验证/测试集,在绘制级别严格分离,同一图纸的所有问答记录分配到同一划分,通过融合CLIP表示计算相似度并进行分层聚类分配,减少近重复重叠。模型训练阶段,首先对Qwen3-VL-4B-Instruct进行全参数SFT,训练实例为(图纸图像x,问题q,目标响应y*)三元组,优化因果语言建模目标损失LSFT = -∑log πθ(y*t | x, q, y*<t),使模型学习将制图线索锚定到机械语义并遵循要求的响应模式。然后进行两阶段DAPO自我博弈RL:首先在完整MechVQA训练集上RL,然后在增强低表现子任务比例的采样子集上继续训练,保持相同目标和奖励。DAPO每次更新从旧策略πθold采样一组G个候选响应{yi},每个响应获得奖励Ri = r(u, yi),计算组归一化优势Âi,t = (Ri - mean{Rj}) / (std{Rj} + εA),定义token级重要性比率ri,t(θ) = πθ(yi,t | u, yi,<t) / πθold(yi,t | u, yi,<t),优化PPO风格裁剪代理LDAPO = -E,{yi} (1/G) ∑|yi| (1/|yi|) ∑t min(ri,t(θ)Âi,t, clip(ri,t(θ), 1-εlow, 1+εhigh)Âi,t)。复合奖励R(x, q, y) = λacc racc + λfmt rfmt + λqual rqual,其中racc通过LLM-as-a-Judge评估语义正确性,rfmt为二元格式奖励(响应可解析且包含解释和答案部分为1,否则0),rqual评估逻辑性、专业性、简洁性三个维度。

技术新颖性

技术新颖性体现在多个方面:在数据构建方面,采用了半自动化流水线结合专家验证的质量控制机制,通过多轮交叉模型检查、多数投票、专家审计确保问答质量,而不是简单的大规模合成。在任务设计方面,首次提出了涵盖识别、推理、判断三个能力维度的分层分类法,并将每个维度细分为具体子任务(如识别维度包括识别计数、尺寸标注、文本表格、物品定位;推理维度包括结构理解、几何计算、装配关系、投影多视图;判断维度包括异常检测、一致性判断),为工程制图理解提供了系统性的评估框架。在奖励设计方面,采用与分类学对齐的复合奖励,而非简单的字符串匹配,语义容忍的准确率奖励允许表达不同但语义正确的答案获得非零奖励,质量奖励通过LLM-as-a-Judge评估逻辑性、专业性和简洁性,直接针对密集机械图上的失效模式进行优化。在训练策略方面,采用两阶段自我博弈RL,先全数据DAPO再目标子任务上采样DAPO,这种设计减少了能力不平衡,从消融实验可以看到目标RL将推理能力从70.75提升到77.04,识别能力从86.26提升到89.70。

MechVQA dataset construction and analysis.
Figure 2: MechVQA dataset construction and analysis.

实验结果

实验结果显示,MechVL-4B-RL在MechVQA上实现了最佳总体得分84.85,超越最强开源基线GLM-4.6V(78.91)5.94个百分点,超越最强闭源基线Gemini-3-Pro-Preview(77.28)7.57个百分点。相对于MechVL-4B-SFT(76.36),RL带来+8.49的大幅增益,表明自我博弈RL对密集机械图和约束敏感任务至关重要。MechVL-4B-RL在多个子任务上达到最佳得分:DA(90.70)、IL(82.01)、SU(83.33)、AR(84.00)、PM(64.00)、AD(86.94),表明领域后训练保持了强感知能力,同时大幅加强了投影一致推理和标准感知判断。难度分层分析显示,所有模型从简单到困难的准确率都有明显下降,表明更难的问题需要更可靠的跨视图对应、更严格的约束满足和超越直接阅读的多步推理。MechVL-4B-RL在三个难度级别上实现最平衡的性能,达到94%(简单)、79%(中等)、75%(困难),相比SFT在中等准确率上从70%提升到79%,困难准确率上从53%提升到75%,而简单性能保持相似(92%到94%),这表明RL主要提高了更高推理和一致性需求下的鲁棒性。能力维度比较显示,MechVL-4B-RL在三个能力轴上都有一致提升,平均相应子任务得分达到89.70(识别)、77.04(推理)、82.81(判断)。与GLM-4.6V相比,对应增益为+5.68、+6.54、+11.00;与Gemini-3-Pro-Preview相比,增益为+8.14、+19.62、+2.29,这些结果表明后训练一致性地加强了感知重阅读、投影一致推理和标准感知决策。

Comparison of VQA Datasets: General vs. CAD/Mechanical Domains.
Table 1: Comparison of VQA Datasets: General vs. CAD/Mechanical Domains.
Evaluation results of MechVQA on open-source, closed-source MLLMs and our MechVL models.
Table 2: Evaluation results of MechVQA on open-source, closed-source MLLMs and our MechVL models.
Combined ablation results on training stages, RL algorithms, and reward design.
Table 3: Combined ablation results on training stages, RL algorithms, and reward design.
Evaluation results on difficulty levels
Figure 3: Evaluation results on difficulty levels
Response length dynamics under different reward designs during RL training.
Figure 4: Response length dynamics under different reward designs during RL training.
查看结构化数据
任务指标本文基线提升
MechVQA总体得分 总分 84.85 GLM-4.6V (78.91) +5.94
MechVQA总体得分 总分 84.85 Gemini-3-Pro-Preview (77.28) +7.57
Dimension & Annotation 准确率 90.70 Qwen3-VL-32B (86.68) +4.02
Anomaly Detection 准确率 86.94 Gemini-3-Pro-Preview (78.37) +8.57
困难级别 准确率 75 Qwen3-VL-Plus (66) +9
简单级别 准确率 94 GLM-4.6V (92) +2
推理能力平均 准确率 77.04 Gemini-3-Pro-Preview (57.42) +19.62
判断能力平均 准确率 82.81 GLM-4.6V (71.81) +11.00

局限与改进

论文作者承认的局限性包括:数据来源范围有限,MechVQA从公开教育及专业材料(教材、手册、设计平台)构建,而非专有工业档案,因此可能无法完全捕捉真实工厂图纸的变异性、遗留蓝图或公司特定制图惯例;专注于2D图纸理解,虽然MechVQA包含多视图推理和涉及等轴测图的问题,但基准以2D机械图纸的图纸基础理解为中心,不旨在解决完整的3D CAD重建或直接生成STEP、IGES等工程文件格式;依赖OCR和视觉清晰度,基准构建流水线部分依赖OCR、元数据提取和专家验证,虽然应用多阶段验证、语义投票和专家审计,但在标注极端拥挤、扫描质量差或视觉局部区域模糊的图纸上性能可能仍会下降;残留污染风险,通过强制图纸级别划分和感知分配明确缓解基准内部泄漏,但由于基准从公共来源构建且现代基础模型在广泛网络语料库上训练,无法排除绝对污染。本文观察到的额外局限性:基准验证仍不完整,目前不报告人类专家上限或正式标注者间一致性统计,虽然标注过程由书面标注手册和结构化工作流程指导而非临时标注,但未来版本应包含人类一致性分析以更好量化基准难度、标注一致性与模型和领域专家之间的剩余差距;发布和许可限制,由于MechVQA从公开可访问的教育和专业来源构建,发布将遵循底层材料的再分发许可,当直接再分发原始图纸受限时,将在允许时发布相应注释、元数据、划分信息、提示词和来源引用。

独立分析的弱点

独立分析的弱点包括:数据规模相对较小,虽然经过严格质量控制,但3,281张图纸和20,778个问答对相比通用VQA基准(如VQA有25万张图像、75万个问题)规模有限,可能限制了模型的泛化能力和对罕见工程图类型的覆盖;模型初始化依赖特定基线,MechVL基于Qwen3-VL-4B-Instruct初始化,虽然这是合理的选择,但未探索其他视觉语言架构作为初始化,可能存在架构偏见;评估指标相对单一,主要使用准确率作为评估指标,虽然涵盖10个子任务,但对于工程制图理解的其他重要方面(如可解释性、与人类工程思维的一致性)缺乏更细粒度的评估;训练计算成本较高,两阶段训练(SFT+RL)和DAPO算法需要大量计算资源,尤其是自我博弈RL阶段采样多组候选响应,与仅SFT的方法相比成本显著增加。改进方向:扩大数据规模,纳入更多样化的工程图类型(如不同行业标准、不同年代的图纸、不同复杂度的装配图);探索多架构初始化,比较不同视觉语言架构(如LLaVA架构、InternVL架构)作为初始基线的效果;引入更全面的评估指标,包括可解释性评估、与人类专家思维路径一致性评估、实际工程场景中的效用评估;优化训练效率,探索参数高效的RL方法(如PEFT结合RL)或减少采样组大小的策略。

未来方向

作者提出的未来研究方向包括:报告人类专家上限和标注者间一致性统计,更好量化基准难度和模型与领域专家之间的剩余差距;扩展基准以包含遗留蓝图和公司特定制图惯例;开发3D CAD重建能力或直接生成工程文件格式(如STEP或IGES)的能力。基于本文成果可延伸的未来方向:将MechVL集成到实际的机械设计和检验工作流中,作为设计审查和检验流程的决策支持助手;开发针对特定工程领域的专用版本(如汽车、航空航天、模具设计等),每个领域有其独特的制图惯例和符号体系;探索 MechVL在工程教育中的应用,如辅助学生理解复杂工程图、自动生成练习题和解答;研究跨模态知识迁移,将从MechVQA学到的机械制图知识迁移到其他工程图类型(如电路图、液压图、建筑结构图等);开发实时交互版本,允许工程师在查看图纸时通过自然语言交互获取解释、验证标注、计算尺寸等;结合实际工程数据(如CAD模型、制造数据)构建更全面的工程理解系统,连接图纸理解与下游工程应用。

复现评估

复现评估:论文提供了相对详细的实现细节,包括训练框架(LLaMA-Factory for SFT、EasyR1 for RL)、模型初始化(Qwen3-VL-4B-Instruct)、奖励设计公式、DAPO算法细节、数据构建流水线、评估协议等。超参数配置在附录C.1和C.2中提供。然而,论文未提供完整的代码仓库、预训练模型或数据集的直接下载链接,这是复现的主要障碍。数据集发布计划说明将遵循底层材料的再分发许可,当直接再分发原始图纸受限时将发布相应注释、元数据、划分信息、提示词和来源引用,但具体发布时间表和数据访问方式未明确。模型方面,论文未说明是否计划发布MechVL的检查点。算力方面,虽然未提供具体的GPU数量和训练时长,但从Qwen3-VL-4B的规模和DAPO的自我博弈特性推断,训练需要相当可观的计算资源(可能需要多个A100或H100 GPU)。总体而言,论文在方法论描述上较为详细,但由于数据集和模型可用性未完全明确,复现难度为中等偏高,如果数据集和模型按计划发布,复现难度将降低。