arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2307.09548cs.CV
收藏

基于器械-组织交互混合监督学习的手术动作三元组检测

Surgical Action Triplet Detection by Mixed Supervised Learning of Instrument-Tissue Interactions

ICube Laboratory(ICube实验室);University of Strasbourg(斯特拉斯堡大学);IHU Strasbourg(斯特拉斯堡大学医院研究所);University Hospital of Strasbourg(斯特拉斯堡大学医院);CNRS(法国国家科学研究中心)

Saurav Sharma, Chinedu Innocent Nwoye, Didier Mutter, Nicolas Padoy

AI总结:

针对手术动作三元组检测缺乏空间标注、易出现关联错误的问题,提出两阶段MCIT-IG网络与混合监督学习策略,在CholecT50数据集上实现性能提升,登顶MICCAI 2022相关挑战赛。

摘要:

手术动作三元组将器械-组织交互描述为(器械、动词、目标)的组合,从而支持对手术场景活动和工作流程的精细化分析。本研究聚焦于手术动作三元组检测任务,该任务比传统的三元组识别任务更具挑战性但也更精准,因为它包含两个联合任务:(1)手术器械定位;(2)识别每个已定位器械对应的手术动作三元组。由于缺乏空间三元组标注,三元组检测的复杂度极高。我们分析了器械空间标注的数量对三元组检测的影响,并观察到精准的器械定位并不一定能带来更好的三元组检测效果,因为存在动词和目标关联错误的风险。为解决这两项任务,我们提出了MCIT-IG(多类别器械感知Transformer-交互图)这一两阶段网络。网络的MCIT阶段将目标的逐类别嵌入建模为额外特征,以降低三元组错配的风险。此外,IG阶段构建了一个二分动态图,用于建模器械与目标之间以动词为表现形式的交互。我们采用混合监督学习策略训练网络,该策略结合了用于MCIT的弱目标存在标签和用于IG的伪三元组标签。我们观察到,用目标嵌入补充最少的器械空间标注可实现更优的三元组检测效果。我们在CholecT50数据集上评估了模型,结果显示模型在器械定位和三元组检测两项任务上的性能均有所提升,在MICCAI 2022的CholecTriplet挑战赛排行榜上位列第一。

英文摘要:

Surgical action triplets describe instrument-tissue interactions as (instrument, verb, target) combinations, thereby supporting a detailed analysis of surgical scene activities and workflow. This work focuses on surgical action triplet detection, which is challenging but more precise than the traditional triplet recognition task as it consists of joint (1) localization of surgical instruments and (2) recognition of the surgical action triplet associated with every localized instrument. Triplet detection is highly complex due to the lack of spatial triplet annotation. We analyze how the amount of instrument spatial annotations affects triplet detection and observe that accurate instrument localization does not guarantee better triplet detection due to the risk of erroneous associations with the verbs and targets. To solve the two tasks, we propose MCIT-IG, a two-stage network, that stands for Multi-Class Instrument-aware Transformer-Interaction Graph. The MCIT stage of our network models per class embedding of the targets as additional features to reduce the risk of misassociating triplets. Furthermore, the IG stage constructs a bipartite dynamic graph to model the interaction between the instruments and targets, cast as the verbs. We utilize a mixed-supervised learning strategy that combines weak target presence labels for MCIT and pseudo triplet labels for IG to train our network. We observed that complementing minimal instrument spatial annotations with target embeddings results in better triplet detection. We evaluate our model on the CholecT50 dataset and show improved performance on both instrument localization and triplet detection, topping the leaderboard of the CholecTriplet challenge in MICCAI 2022.

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

↑