一种融合视觉Transformer和扩散模型的单视点内窥镜手术光场重建方法(特邀)

作     者：韩晨明 吴高昌 Han Chenming;Wu Gaochang

作者机构：东北大学流程工业综合自动化国家重点实验室辽宁沈阳110819 

出 版 物：《激光与光电子学进展》 (Laser & Optoelectronics Progress)

年 卷 期：2024年第61卷第16期

页      面：183-193页

核心收录：

学科分类：0808[工学-电气工程] 070207[理学-光学] 07[理学] 08[工学] 0803[工学-光学工程] 0702[理学-物理学] 

基　　金：国家自然科学基金(62103092,61991404) 教育部中央高校基础研究基金项目(N2108001,N2424004) 辽宁省辽河实验室研究计划(LLL23ZZ-05-01) 

主　　题：光场重建 视觉Transformer 多平面图像表示 条件扩散模型 

摘      要：针对内窥镜手术中单一视角图像深度估计不确定性与遮挡导致的手术场景三维感知难题,提出了一种融合视觉Transformer和条件扩散模型的单视点多平面图(MPI)表征方法,用以进行内窥镜手术光场重建。该方法首先利用视觉Transformer将输入的单视角图像令牌化,从而分解为多个图像块,并通过多头注意力机制提取局部与全局相结合的关联特征。然后,利用多尺度卷积解码器将图像块特征从粗到细进行重组与融合,生成初始MPI。最后,为了解决单视点内窥镜手术中组织之间的遮挡问题,引入了一个基于条件扩散模型的背景预测模块,根据初始MPI获取遮挡掩模,并以遮挡掩模和输入视角作为条件,预测被遮挡区域的分布,有效解决了单视点输入引起的光场内视角不连贯的问题。所提方法将基于视觉Transformer所分解的初始MPI与基于条件扩散模型预测的背景区域相结合,得到优化后的MPI,从而渲染出内窥镜手术光场中的各子视点图像。在达芬奇手术机器人的真实内窥镜手术数据集上的实验验证表明,所提方法在视觉和客观评价指标上均优于现有的单视图光场重建方法。