arXiv reaDer
生成拡散機能による 3D 階層構造の教師なし発見
Unsupervised Discovery of 3D Hierarchical Structure with Generative Diffusion Features
生成拡散モデルが意味的に意味のある表現を学習するという最近の発見に触発され、教師なしセグメンテーションを使用して生物医学 3D 画像の固有の階層構造を発見するためにそれらを使用します。 U-Net ベースのはしごのようなアーキテクチャのさまざまな段階からの拡散モデルの機能が、3D 生物医学画像のさまざまな階層レベルをキャプチャすることを示します。階層を表す意味のあるネストされたサブボリュームへの 3D ボリュームの分解を促進する予測教師なしセグメンテーション ネットワークをトレーニングするために、3 つの損失を設計します。まず、3D 拡散モデルを事前トレーニングし、サブボリューム全体でその特徴の一貫性を使用します。次に、サブボリューム間の視覚的な一貫性を使用します。第三に、正則化として測光拡張に対する不変性を使用します。私たちのモデルは、挑戦的な生物学に触発された合成データセットおよび現実世界の脳腫瘍 MRI データセットで、以前の教師なし構造発見アプローチよりも優れたパフォーマンスを達成します。
Inspired by recent findings that generative diffusion models learn semantically meaningful representations, we use them to discover the intrinsic hierarchical structure in biomedical 3D images using unsupervised segmentation. We show that features of diffusion models from different stages of a U-Net-based ladder-like architecture capture different hierarchy levels in 3D biomedical images. We design three losses to train a predictive unsupervised segmentation network that encourages the decomposition of 3D volumes into meaningful nested subvolumes that represent a hierarchy. First, we pretrain 3D diffusion models and use the consistency of their features across subvolumes. Second, we use the visual consistency between subvolumes. Third, we use the invariance to photometric augmentations as a regularizer. Our models achieve better performance than prior unsupervised structure discovery approaches on challenging biologically-inspired synthetic datasets and on a real-world brain tumor MRI dataset.
updated: Tue Oct 10 2023 04:01:38 GMT+0000 (UTC)
published: Fri Apr 28 2023 19:37:17 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アソシエイト