arXiv:2502.14149cs.CVcs.AI2025-02被引 6

针对垂体手术视觉问答,提出高效微调方法与新数据集

PitVQA++: Vector Matrix-Low-Rank Adaptation for Open-Ended Visual Question Answering in Pituitary Surgery

  • 用向量排序分配参数,让早期网络层获更多调整权重
  • 在10万+帧数据上表现超越现有基线,减少灾难性遗忘
  • 适合医学影像、手术辅助系统研究者使用

视觉语言模型(VLMs)在视觉问答(VQA)中为术中决策、直观交互和外科教学提供新可能。然而,受限于数据集小及全量微调易过拟合、灾难性遗忘等问题,其在手术VQA中的应用面临挑战。本文提出PitVQA++,包含约101,803帧的开放型垂体手术数据集,以及向量矩阵低秩适配(Vector-MoLoRA)方法,用于适配GPT-2模型。该方法结合LoRA与MoRA思想,通过向量排名机制将更多参数分配给早期层(学习通用特征),后期逐层递减。在开放型PitVQA与EndoVis18-VQA数据集上的验证显示,该方法有效缓解灾难性遗忘,并显著提升性能。风险覆盖分析表明其在不确定预测下更具可靠性与可信度。代码与数据集已开源。

原文摘要 · Abstract (English)

Vision-Language Models (VLMs) in visual question answering (VQA) offer a unique opportunity to enhance intra-operative decision-making, promote intuitive interactions, and significantly advancing surgical education. However, the development of VLMs for surgical VQA is challenging due to limited datasets and the risk of overfitting and catastrophic forgetting during full fine-tuning of pretrained weights. While parameter-efficient techniques like Low-Rank Adaptation (LoRA) and Matrix of Rank Adaptation (MoRA) address adaptation challenges, their uniform parameter distribution overlooks the feature hierarchy in deep networks, where earlier layers, that learn general features, require more parameters than later ones. This work introduces PitVQA++ with an open-ended PitVQA dataset and vector matrix-low-rank adaptation (Vector-MoLoRA), an innovative VLM fine-tuning approach for adapting GPT-2 to pituitary surgery. Open-Ended PitVQA comprises around 101,803 frames from 25 procedural videos with 745,972 question-answer sentence pairs, covering key surgical elements such as phase and step recognition, context understanding, tool detection, localization, and interactions recognition. Vector-MoLoRA incorporates the principles of LoRA and MoRA to develop a matrix-low-rank adaptation strategy that employs vector ranking to allocate more parameters to earlier layers, gradually reducing them in the later layers. Our approach, validated on the Open-Ended PitVQA and EndoVis18-VQA datasets, effectively mitigates catastrophic forgetting while significantly enhancing performance over recent baselines. Furthermore, our risk-coverage analysis highlights its enhanced reliability and trustworthiness in handling uncertain predictions. Our source code and dataset is available at~\url{https://github.com/HRL-Mike/PitVQA-Plus}.

视觉问答手术辅助低秩微调

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。