arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2201.11957eess.IVcs.RO
收藏

面向手术场景理解的全局推理多任务学习模型

Global-Reasoned Multi-Task Learning Model for Surgical Scene Understanding

Biomedical Image Analysis Group, Imperial College London(伦敦帝国理工学院生物医学图像分析组);Dept. of Biomedical Engineering, National University of Singapore(新加坡国立大学生物医学工程系);Dept. of Electronics and Communication Engineering, National Institute of Technology, Tiruchirappalli(蒂鲁吉拉帕利国家技术学院电子与通信工程系);Dept. of Electronic Engineering, The Chinese University of Hong Kong(香港中文大学电子工程系)

Lalithkumar Seenivasan, Sai Mitheran, Mobarakol Islam, Hongliang Ren

摘要:

全局和局部关系推理使场景理解模型能够执行类似于人类的场景分析和理解。场景理解能够更好地进行语义分割和对象间交互检测。在医学领域,一个强大的手术场景理解模型允许自动化的手术技能评估,实时监控外科医生的表现和手术后的分析。本文介绍了一种全局推理的多任务手术场景理解模型,能够执行器械分割和工具-组织交互检测。在这里,我们将全球关系推理的潜在的相互作用空间,并在坐标空间中引入多尺度的局部(邻域)推理,以改善分割。利用多任务模型设置,通过全局推理进一步提高视觉语义图注意网络在交互检测中的性能。将来自分割模块的全局交互空间特征引入到图网络中,使其能够基于节点到节点和全局交互推理来检测交互。与运行两个独立的单任务模型相比,我们的模型通过共享公共模块降低了计算成本,这对于实际应用是不可或缺的。使用顺序优化技术,所提出的多任务模型在MICCAI内窥镜视觉挑战2018数据集上的表现优于其他最先进的单任务模型。此外,我们还观察了使用知识蒸馏技术训练时多任务模型的性能。官方代码实现在GitHub上提供。

英文摘要:

Global and local relational reasoning enable scene understanding models to perform human-like scene analysis and understanding. Scene understanding enables better semantic segmentation and object-to-object interaction detection. In the medical domain, a robust surgical scene understanding model allows the automation of surgical skill evaluation, real-time monitoring of surgeon's performance and post-surgical analysis. This paper introduces a globally-reasoned multi-task surgical scene understanding model capable of performing instrument segmentation and tool-tissue interaction detection. Here, we incorporate global relational reasoning in the latent interaction space and introduce multi-scale local (neighborhood) reasoning in the coordinate space to improve segmentation. Utilizing the multi-task model setup, the performance of the visual-semantic graph attention network in interaction detection is further enhanced through global reasoning. The global interaction space features from the segmentation module are introduced into the graph network, allowing it to detect interactions based on both node-to-node and global interaction reasoning. Our model reduces the computation cost compared to running two independent single-task models by sharing common modules, which is indispensable for practical applications. Using a sequential optimization technique, the proposed multi-task model outperforms other state-of-the-art single-task models on the MICCAI endoscopic vision challenge 2018 dataset. Additionally, we also observe the performance of the multi-task model when trained using the knowledge distillation technique. The official code implementation is made available in GitHub.

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

↑