用指令微调让模型学会识别新关系,提升视觉关系预测泛化能力。
ART: Adaptive Relation Tuning for Generalized Relation Prediction
- 通过指令微调和自适应采样,让模型聚焦关键关系。
- 在未见过的关系上表现优于基线方法,支持新概念推理。
- 适合需要泛化能力的复杂场景理解任务。
视觉关系检测(VRD)旨在识别图像中物体之间的关系。仅在关系检测数据上训练的模型难以泛化到未见关系。虽然提示微调被用于适配视觉-语言模型(VLM)进行VRD,但依赖人工设计提示,对新关系或复杂关系表现不佳。本文认为,通过多样化指令数据进行指令微调更具优势。因此提出ART框架,通过指令微调与策略性实例选择,适配VLM用于VRD。将VRD数据集转换为指令格式,并采用自适应采样算法,引导VLM关注信息量高的关系,同时保持泛化能力。重点关注关系分类任务:给定主体-客体边界框,模型预测两者间的谓词。在保留集上微调,在多个不同复杂度的保留数据集上评估。结果表明,该方法显著优于基线,可推断未见关系概念,这是主流方法不具备的能力。进一步通过预测关系实现复杂场景分割,验证了其实际价值。
原文摘要 · Abstract (English)
Visual relation detection (VRD) is the task of identifying the relationships between objects in a scene. VRD models trained solely on relation detection data struggle to generalize beyond the relations on which they are trained. While prompt tuning has been used to adapt vision-language models (VLMs) for VRD, it uses handcrafted prompts and struggles with novel or complex relations. We argue that instruction tuning offers a more effective solution by fine-tuning VLMs on diverse instructional data. We thus introduce ART, an Adaptive Relation Tuning framework that adapts VLMs for VRD through instruction tuning and strategic instance selection. By converting VRD datasets into an instruction tuning format and employing an adaptive sampling algorithm, ART directs the VLM to focus on informative relations while maintaining generalizability. Specifically, we focus on the relation classification, where subject-object boxes are given and the model predicts the predicate between them. We tune on a held-in set and evaluate across multiple held-out datasets of varying complexity. Our approach strongly improves over its baselines and can infer unseen relation concepts, a capability absent in mainstream VRD methods. We demonstrate ART's practical value by using the predicted relations for segmenting complex scenes.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。