arXiv reaDer
ノイズ除去による監視
Supervision by Denoising
U-Net に基づくモデルなどの学習ベースの画像再構成モデルでは、適切な一般化が保証される場合、大量のラベル付き画像セットが必要です。ただし、一部のイメージング ドメインでは、ピクセル レベルまたはボクセル レベルのラベル精度を持つラベル付きデータは、取得コストが原因で不足しています。この問題は、単一のグラウンド トゥルース ラベルが存在しない医療画像処理のような分野ではさらに悪化し、ラベルに大量の反復変動が生じます。したがって、ラベル付けされた例とラベル付けされていない例の両方から学習することにより (半教師あり学習と呼ばれる)、再構成ネットワークをよりよく一般化するようにトレーニングすることは、実用的かつ理論的に重要な問題です。ただし、画像再構成のための従来の半教師付き学習方法では、多くの場合、特定の画像処理の問題に固有の微分可能な正則化器を手作りする必要があり、非常に時間がかかる可能性があります。この作業では、「ノイズ除去による監視」(SUD) を提案します。これは、独自のノイズ除去された出力をソフトラベルとして使用して再構成モデルを監視できるようにするフレームワークです。 SUD は、時空間ノイズ除去フレームワークの下で確率的平均化と空間ノイズ除去手法を統合し、半教師用の最適化フレームワークでノイズ除去とモデルの重み更新ステップを交互に行います。適用例として、生物医学イメージングから生じる 2 つの問題 (解剖学的脳再構成 (3D) と皮質区画 (2D)) に SUD を適用して、監視のみの確率的平均ベースラインよりも画像再構成が大幅に改善されることを示します。
Learning-based image reconstruction models, such as those based on the U-Net, require a large set of labeled images if good generalization is to be guaranteed. In some imaging domains, however, labeled data with pixel- or voxel-level label accuracy are scarce due to the cost of acquiring them. This problem is exacerbated further in domains like medical imaging, where there is no single ground truth label, resulting in large amounts of repeat variability in the labels. Therefore, training reconstruction networks to generalize better by learning from both labeled and unlabeled examples (called semi-supervised learning) is problem of practical and theoretical interest. However, traditional semi-supervised learning methods for image reconstruction often necessitate handcrafting a differentiable regularizer specific to some given imaging problem, which can be extremely time-consuming. In this work, we propose "supervision by denoising" (SUD), a framework that enables us to supervise reconstruction models using their own denoised output as soft labels. SUD unifies stochastic averaging and spatial denoising techniques under a spatio-temporal denoising framework and alternates denoising and model weight update steps in an optimization framework for semi-supervision. As example applications, we apply SUD to two problems arising from biomedical imaging -- anatomical brain reconstruction (3D) and cortical parcellation (2D) -- to demonstrate a significant improvement in the image reconstructions over supervised-only and stochastic averaging baselines.
updated: Wed Apr 19 2023 21:08:30 GMT+0000 (UTC)
published: Mon Feb 07 2022 05:29:16 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アソシエイト