arXiv reaDer
行列積状態テンソルネットワークを使用したパッチベースの医療画像セグメンテーション
Patch-based Medical Image Segmentation using Matrix Product State Tensor Networks
テンソルネットワークは、高次元テンソルを低次テンソルのネットワークに効率的に因数分解します。それらは、量子多体システムのエンタングルメントをモデル化するために最も一般的に使用されており、最近では、教師あり機械学習でのアプリケーションの増加を目の当たりにしています。この作業では、テンソルネットワークを使用した教師あり設定で画像セグメンテーションを作成します。重要なアイデアは、最初に画像パッチ内のピクセルを指数関数的に高次元の特徴空間に持ち上げ、線形決定超平面を使用して入力ピクセルを前景クラスと背景クラスに分類することです。高次元線形モデル自体は、行列積状態(MPS)テンソルネットワークを使用して近似されます。 MPSは、重複しない画像パッチ間で重みが共有され、ストライドテンソルネットワークモデルになります。提案されたモデルのパフォーマンスは、3つの2Dおよび1つの3D生物医学画像データセットで評価されます。提案されたテンソルネットワークセグメンテーションモデルのパフォーマンスは、関連するベースライン方法と比較されます。 2D実験では、テンソルネットワークモデルは、リソース効率を高めながら、ベースライン手法と比較して競争力のあるパフォーマンスを実現します。
Tensor networks are efficient factorisations of high-dimensional tensors into a network of lower-order tensors. They have been most commonly used to model entanglement in quantum many-body systems and more recently are witnessing increased applications in supervised machine learning. In this work, we formulate image segmentation in a supervised setting with tensor networks. The key idea is to first lift the pixels in image patches to exponentially high-dimensional feature spaces and using a linear decision hyper-plane to classify the input pixels into foreground and background classes. The high-dimensional linear model itself is approximated using the matrix product state (MPS) tensor network. The MPS is weight-shared between the non-overlapping image patches resulting in our strided tensor network model. The performance of the proposed model is evaluated on three 2D- and one 3D- biomedical imaging datasets. The performance of the proposed tensor network segmentation model is compared with relevant baseline methods. In the 2D experiments, the tensor network model yields competitive performance compared to the baseline methods while being more resource efficient.
updated: Wed Feb 23 2022 14:01:56 GMT+0000 (UTC)
published: Wed Sep 15 2021 07:54:05 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アソシエイト