arXiv reaDer
BT-Unet:U-NetモデルでBarlowTwinsを使用した生物医学画像セグメンテーションのための自己監視学習フレームワーク
BT-Unet: A self-supervised learning framework for biomedical image segmentation using Barlow Twins with U-Net models
ディープラーニングは、医用画像の描写プロセスを自動化するための生物医学画像のセグメンテーションに最も大きな貢献をもたらしました。このようなタスクを実行するには、バイナリマスクを使用して関心領域を強調表示する、注釈付きまたはラベル付きの大量のデータを使用してモデルをトレーニングする必要があります。ただし、このような膨大なデータの注釈を効率的に生成するには、専門の生物医学アナリストと手作業による多大な労力が必要です。これは退屈で費用のかかる作業ですが、人為的ミスに対しても脆弱です。この問題に対処するために、教師なし学習フレームワークであるBT-Unetが提案されています。これは、Barlow Twinsアプローチを使用して、データ表現を学習するための監視されていない方法で冗長性の削減を介してU-Netモデルのエンコーダーを事前トレーニングします。後で、完全なネットワークが実際のセグメンテーションを実行するように微調整されます。 BT-Unetフレームワークは、限られた数の注釈付きサンプルでトレーニングできますが、注釈なしのサンプルは多数あります。これは、ほとんどの場合、実際の問題に当てはまります。このフレームワークは、標準の評価指標を使用して限られた数のラベル付きサンプルのシナリオを生成することにより、さまざまなデータセットの複数のU-Netモデルで検証されます。徹底的な実験試行により、BT-Unetフレームワークは、そのような状況下でかなりのマージンを持ってU-Netモデルのパフォーマンスを向上させることが観察されています。
Deep learning has brought the most profound contribution towards biomedical image segmentation to automate the process of delineation in medical imaging. To accomplish such task, the models are required to be trained using huge amount of annotated or labelled data that highlights the region of interest with a binary mask. However, efficient generation of the annotations for such huge data requires expert biomedical analysts and extensive manual effort. It is a tedious and expensive task, while also being vulnerable to human error. To address this problem, a self-supervised learning framework, BT-Unet is proposed that uses the Barlow Twins approach to pre-train the encoder of a U-Net model via redundancy reduction in an unsupervised manner to learn data representation. Later, complete network is fine-tuned to perform actual segmentation. The BT-Unet framework can be trained with a limited number of annotated samples while having high number of unannotated samples, which is mostly the case in real-world problems. This framework is validated over multiple U-Net models over diverse datasets by generating scenarios of a limited number of labelled samples using standard evaluation metrics. With exhaustive experiment trials, it is observed that the BT-Unet framework enhances the performance of the U-Net models with significant margin under such circumstances.
updated: Tue Dec 07 2021 17:14:15 GMT+0000 (UTC)
published: Tue Dec 07 2021 17:14:15 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アソシエイト