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

基于未分割医学图像的弱监督贝叶斯形状建模

Weakly Supervised Bayesian Shape Modeling from Unsegmented Medical Images

Scientific Computing and Imaging Institute, University of Utah(犹他大学科学计算与成像研究所);Kahlert School of Computing, University of Utah(犹他大学卡勒特计算学院)

Jadie Adams, Krithika Iyer, Shireen Elhabian

AI总结:

针对全监督统计形状建模依赖繁琐传统流程生成训练数据的问题,提出基于点云弱监督的贝叶斯形状预测方法,在保持准确率与不确定性估计性能的同时提升了训练可行性。

摘要:

解剖形状分析在临床研究与假设检验中发挥着关键作用,其中形态与功能的关联至关重要。基于对应关系的统计形状建模(SSM)可支撑群体层面的形态计量分析,但其构建流程繁琐,还可能引入偏差。近期的深度学习进展通过直接从未分割医学图像预测SSM,简化了推理阶段的流程。然而,现有方法均为全监督模式,需要借助传统SSM构建流程生成训练数据,因此继承了相关的负担与局限性。为解决这些问题,我们提出一种弱监督深度学习方法,利用点云监督从图像中预测SSM。具体而言,我们降低了当前最优的全贝叶斯变分信息瓶颈DeepSSM(BVIB-DeepSSM)模型所需的监督强度。BVIB-DeepSSM是一种有效且具备理论基础的框架,可从图像中预测概率性解剖形状,同时量化偶然不确定性与认知不确定性。原始BVIB-DeepSSM方法需要以真实对应点形式提供强监督,而本文提出的方法通过点云表面表征实现弱监督,这类监督数据更易获取。此外,该方法以完全数据驱动的方式学习对应关系,无需对形状队列的预期变异性做出先验假设。我们的实验表明,该方法在准确率与不确定性估计方面与全监督场景表现相当,同时大幅提升了SSM构建中模型训练的可行性。

英文摘要:

Anatomical shape analysis plays a pivotal role in clinical research and hypothesis testing, where the relationship between form and function is paramount. Correspondence-based statistical shape modeling (SSM) facilitates population-level morphometrics but requires a cumbersome, potentially bias-inducing construction pipeline. Recent advancements in deep learning have streamlined this process in inference by providing SSM prediction directly from unsegmented medical images. However, the proposed approaches are fully supervised and require utilizing a traditional SSM construction pipeline to create training data, thus inheriting the associated burdens and limitations. To address these challenges, we introduce a weakly supervised deep learning approach to predict SSM from images using point cloud supervision. Specifically, we propose reducing the supervision associated with the state-of-the-art fully Bayesian variational information bottleneck DeepSSM (BVIB-DeepSSM) model. BVIB-DeepSSM is an effective, principled framework for predicting probabilistic anatomical shapes from images with quantification of both aleatoric and epistemic uncertainties. Whereas the original BVIB-DeepSSM method requires strong supervision in the form of ground truth correspondence points, the proposed approach utilizes weak supervision via point cloud surface representations, which are more readily obtainable. Furthermore, the proposed approach learns correspondence in a completely data-driven manner without prior assumptions about the expected variability in shape cohort. Our experiments demonstrate that this approach yields similar accuracy and uncertainty estimation to the fully supervised scenario while substantially enhancing the feasibility of model training for SSM construction.

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

↑