arXiv reaDer
EMボリュームでのミトコンドリアセグメンテーションのための深層学習ベースのドメイン適応
Deep learning based domain adaptation for mitochondria segmentation on EM volumes
脳の電子顕微鏡(EM)ボリュームの正確なセグメンテーションは、細胞または細胞小器官レベルでニューロン構造を特徴づけるために不可欠です。教師あり深層学習法は、過去数年間にその方向に大きな進歩をもたらしましたが、通常、大量の注釈付きデータをトレーニングする必要があり、同様の実験およびイメージング条件下で取得された他のデータではパフォーマンスが低下します。サンプル分布(またはソースドメイン)から学習したモデルは、別の分布またはターゲットドメインから抽出されたサンプルでパフォーマンスを維持するのに苦労するため、これはドメイン適応として知られる問題です。この作業では、さまざまな組織や種からのEMデータセット全体のミトコンドリアセグメンテーションのためのディープラーニングベースのドメイン適応の複雑なケースに対処します。 (1)両方のドメインの画像間の最先端のスタイルの転送に基づいて、ターゲットドメインのミトコンドリアセグメンテーションを改善するための3つの教師なしドメイン適応戦略を提示します。 (2)ラベルのないソース画像とターゲット画像を使用してモデルを事前トレーニングし、ソースラベルのみを使用してモデルを微調整する自己監視学習。 (3)ラベル付き画像とラベルなし画像の両方でエンドツーエンドでトレーニングされたマルチタスクニューラルネットワークアーキテクチャ。さらに、ソースドメインでのみ取得された形態学的事前分布に基づいた新しいトレーニング停止基準を提案します。 3つの公開されているEMデータセットを使用して、可能なすべてのクロスデータセット実験を実行しました。ターゲットデータセットで予測されたミトコンドリアセマンティックラベルで提案された戦略を評価しました。ここで紹介する方法は、ベースラインの方法よりも優れており、最先端の方法と比べて遜色ありません。検証ラベルがない場合、提案された形態ベースのメトリックを監視することは、トレーニングプロセスを停止し、平均的に最適なモデルを選択するための直感的で効果的な方法です。
Accurate segmentation of electron microscopy (EM) volumes of the brain is essential to characterize neuronal structures at a cell or organelle level. While supervised deep learning methods have led to major breakthroughs in that direction during the past years, they usually require large amounts of annotated data to be trained, and perform poorly on other data acquired under similar experimental and imaging conditions. This is a problem known as domain adaptation, since models that learned from a sample distribution (or source domain) struggle to maintain their performance on samples extracted from a different distribution or target domain. In this work, we address the complex case of deep learning based domain adaptation for mitochondria segmentation across EM datasets from different tissues and species. We present three unsupervised domain adaptation strategies to improve mitochondria segmentation in the target domain based on (1) state-of-the-art style transfer between images of both domains; (2) self-supervised learning to pre-train a model using unlabeled source and target images, and then fine-tune it only with the source labels; and (3) multi-task neural network architectures trained end-to-end with both labeled and unlabeled images. Additionally, we propose a new training stopping criterion based on morphological priors obtained exclusively in the source domain. We carried out all possible cross-dataset experiments using three publicly available EM datasets. We evaluated our proposed strategies on the mitochondria semantic labels predicted on the target datasets. The methods introduced here outperform the baseline methods and compare favorably to the state of the art. In the absence of validation labels, monitoring our proposed morphology-based metric is an intuitive and effective way to stop the training process and select in average optimal models.
updated: Tue Jul 05 2022 14:43:44 GMT+0000 (UTC)
published: Tue Feb 22 2022 09:49:25 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アソシエイト