arXiv reaDer
画像から確率論的解剖学的形状へ:深い変分ボトルネックアプローチ
From Images to Probabilistic Anatomical Shapes: A Deep Variational Bottleneck Approach
3D医療画像から直接統計的形状モデリング(SSM)は、病理を検出し、病気を診断し、人口レベルの形態分析を行うための十分に活用されていないツールです。ディープラーニングフレームワークは、従来のSSMワークフローで専門家主導の手動および計算のオーバーヘッドを削減することにより、医療現場でSSMを採用する可能性を高めました。ただし、ニューラルネットワークは、機密性の高い臨床的意思決定では信頼できない自信過剰な予測を生成する可能性があるため、このようなフレームワークを臨床診療に変換するには、不確実性の調整が必要です。気晴らし(データ依存)の不確実性を伴う形状を予測するための既存の手法は、モデルトレーニングから分離して計算された主成分分析(PCA)ベースの形状表現を利用します。この制約は、学習タスクを3D画像から事前定義された形状記述子のみを推定するように制限し、この形状表現と出力(つまり形状)空間の間に線形関係を課します。この論文では、形状記述子の監視されたエンコーディングなしで画像から直接解剖学的構造の確率的形状を予測しながら、これらの仮定を緩和するために変分情報ボトルネック理論に基づく原理的なフレームワークを提案します。ここでは、潜在的な表現が学習タスクのコンテキストで学習され、データの非線形性をより適切にキャプチャする、よりスケーラブルで柔軟なモデルが得られます。さらに、このモデルは自己正規化されており、限られたトレーニングデータが与えられるとより一般化されます。私たちの実験は、提案された方法が、最先端の方法よりも改善された精度とより良い較正された気管不確実性推定を提供することを示しています。
Statistical shape modeling (SSM) directly from 3D medical images is an underutilized tool for detecting pathology, diagnosing disease, and conducting population-level morphology analysis. Deep learning frameworks have increased the feasibility of adopting SSM in medical practice by reducing the expert-driven manual and computational overhead in traditional SSM workflows. However, translating such frameworks to clinical practice requires calibrated uncertainty measures as neural networks can produce over-confident predictions that cannot be trusted in sensitive clinical decision-making. Existing techniques for predicting shape with aleatoric (data-dependent) uncertainty utilize a principal component analysis (PCA) based shape representation computed in isolation from the model training. This constraint restricts the learning task to solely estimating pre-defined shape descriptors from 3D images and imposes a linear relationship between this shape representation and the output (i.e., shape) space. In this paper, we propose a principled framework based on the variational information bottleneck theory to relax these assumptions while predicting probabilistic shapes of anatomy directly from images without supervised encoding of shape descriptors. Here, the latent representation is learned in the context of the learning task, resulting in a more scalable, flexible model that better captures data non-linearity. Additionally, this model is self-regularized and generalizes better given limited training data. Our experiments demonstrate that the proposed method provides improved accuracy and better calibrated aleatoric uncertainty estimates than state-of-the-art methods.
updated: Fri May 13 2022 19:39:08 GMT+0000 (UTC)
published: Fri May 13 2022 19:39:08 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト