3D磁気共鳴(MR)画像での椎体(VB)と椎間板(IVD)の自動セグメンテーションは、脊椎疾患の診断と治療に不可欠です。ただし、VBとIVDを同時にセグメント化することは簡単ではありません。さらに、異方性分解能、高い計算コスト、クラス間類似性とクラス内変動、およびデータの不均衡によって引き起こされるぼやけたセグメンテーションを含む問題が存在します。正確な同時VBおよびIVDセグメンテーションを実現することにより、これらの問題に対処するために、半教師ありハイブリッドスパインネットワーク(SSHSNet)という名前の2段階アルゴリズムを提案しました。最初の段階では、クロス疑似監視を使用して2D半教師ありDeepLabv3 +を構築し、スライス内の特徴と粗いセグメンテーションを取得しました。第2段階では、3Dフル解像度パッチベースのDeepLabv3+が構築されました。このモデルを使用して、スライス間情報を抽出し、最初の段階で提供された粗いセグメンテーションとスライス内の機能を組み合わせることができます。さらに、2Dおよび3Dネットワークから別々に生成されたスライス間およびスライス内情報の損失を補償するためにクロストライアテンションモジュールが適用され、それによって特徴表現能力が向上し、満足のいくセグメンテーション結果が達成されました。提案されたSSHSNetは、公開されている脊椎MR画像データセットで検証され、驚くべきセグメンテーションパフォーマンスが達成されました。さらに、結果は、提案された方法がデータ不均衡問題に対処する上で大きな可能性を持っていることを示しています。以前の報告に基づいて、脊椎のセグメンテーションのためのクロスアテンションメカニズムを備えた半教師あり学習戦略を組み込んだ研究はほとんどありません。したがって、提案された方法は、脊椎のセグメンテーションに有用なツールを提供し、脊椎疾患の診断および治療を臨床的に支援する可能性があります。コードはhttps://github.com/Meiyan88/SSHSNetで公開されています。
Automatic segmentation of vertebral bodies (VBs) and intervertebral discs (IVDs) in 3D magnetic resonance (MR) images is vital in diagnosing and treating spinal diseases. However, segmenting the VBs and IVDs simultaneously is not trivial. Moreover, problems exist, including blurry segmentation caused by anisotropy resolution, high computational cost, inter-class similarity and intra-class variability, and data imbalances. We proposed a two-stage algorithm, named semi-supervised hybrid spine network (SSHSNet), to address these problems by achieving accurate simultaneous VB and IVD segmentation. In the first stage, we constructed a 2D semi-supervised DeepLabv3+ by using cross pseudo supervision to obtain intra-slice features and coarse segmentation. In the second stage, a 3D full-resolution patch-based DeepLabv3+ was built. This model can be used to extract inter-slice information and combine the coarse segmentation and intra-slice features provided from the first stage. Moreover, a cross tri-attention module was applied to compensate for the loss of inter-slice and intra-slice information separately generated from 2D and 3D networks, thereby improving feature representation ability and achieving satisfactory segmentation results. The proposed SSHSNet was validated on a publicly available spine MR image dataset, and remarkable segmentation performance was achieved. Moreover, results show that the proposed method has great potential in dealing with the data imbalance problem. Based on previous reports, few studies have incorporated a semi-supervised learning strategy with a cross attention mechanism for spine segmentation. Therefore, the proposed method may provide a useful tool for spine segmentation and aid clinically in spinal disease diagnoses and treatments. Codes are publicly available at: https://github.com/Meiyan88/SSHSNet.