arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.09359eess.IVcs.CV
收藏

剪枝稀疏张量神经网络实现三维超声定位显微成像的深度学习

Pruning Sparse Tensor Neural Networks Enables Deep Learning for 3D Ultrasound Localization Microscopy

Polytechnique Montréal(蒙特利尔理工学院);IVADO(IVADO研究所);TransMedTech Institute(TransMedTech研究所);Mila-Quebec artificial intelligence institute(魁北克人工智能研究所);ServiceNow(ServiceNow公司);Department of Engineering Physics(工程物理系)

Brice Rauby, Paul Xing, Jonathan Porée, Maxime Gasse, Jean Provost

AI总结:

本文提出用稀疏张量神经网络降低超声定位显微成像的内存需求,实现三维深度学习,在减少内存的同时提升高浓度下的成像性能并缩短采集时间。

摘要:

超声定位显微成像(ULM)是一种非侵入性技术,能够在体内深处以约十微米的分辨率对微血管进行成像。ULM基于对注入血流中的单个微泡进行亚分辨率定位。绘制整个血管架构需要从数千帧图像中累积微泡轨迹,这些图像通常需数分钟采集。通过增加微泡浓度可以减少ULM采集时间,但这需要更先进的算法来单独检测它们。已有多种深度学习方法被提出用于此任务,但它们仍局限于二维成像,部分原因是相关的大内存需求。在此,我们提出使用稀疏张量神经网络来减少二维内存使用,并改善深度学习架构扩展到三维时内存需求的缩放性。我们研究了多种将超声数据高效转换为稀疏格式的方法,并研究了相关信息损失的影响。在二维应用中,与密集网络相比,稀疏公式将内存需求减少了2倍,但性能略有下降。在三维中,所提出的方法将内存需求减少了两个数量级,同时在高浓度设置下大幅优于传统ULM。我们表明,三维ULM中的稀疏张量神经网络能够实现与二维ULM中基于密集深度学习的方法相同的优势,即使用更高浓度的硅内实验和减少采集时间。

英文摘要:

Ultrasound Localization Microscopy (ULM) is a non-invasive technique that allows for the imaging of micro-vessels in vivo, at depth and with a resolution on the order of ten microns. ULM is based on the sub-resolution localization of individual microbubbles injected in the bloodstream. Mapping the whole angioarchitecture requires the accumulation of microbubbles trajectories from thousands of frames, typically acquired over a few minutes. ULM acquisition times can be reduced by increasing the microbubble concentration, but requires more advanced algorithms to detect them individually. Several deep learning approaches have been proposed for this task, but they remain limited to 2D imaging, in part due to the associated large memory requirements. Herein, we propose to use sparse tensor neural networks to reduce memory usage in 2D and to improve the scaling of the memory requirement for the extension of deep learning architecture to 3D. We study several approaches to efficiently convert ultrasound data into a sparse format and study the impact of the associated loss of information. When applied in 2D, the sparse formulation reduces the memory requirements by a factor 2 at the cost of a small reduction of performance when compared against dense networks. In 3D, the proposed approach reduces memory requirements by two order of magnitude while largely outperforming conventional ULM in high concentration settings. We show that Sparse Tensor Neural Networks in 3D ULM allow for the same benefits as dense deep learning based method in 2D ULM i.e. the use of higher concentration in silico and reduced acquisition time.

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

↑