医療画像のセグメンテーションに Transformer ベースのアーキテクチャを採用することへの関心が爆発的に高まっています。ただし、大規模なアノテーション付き医療データセットが不足しているため、自然画像と同等のパフォーマンスを達成することが困難になります。対照的に、畳み込みネットワークは誘導バイアスが高いため、簡単にトレーニングして高いパフォーマンスを得ることができます。最近、ConvNeXt アーキテクチャは、Transformer ブロックをミラーリングすることで標準 ConvNet を最新化しようとしました。この研究では、これを改良して、データ不足の医療現場の課題に合わせてカスタマイズされた、最新化されたスケーラブルな畳み込みアーキテクチャを設計します。 Transformer からインスピレーションを得た大規模なカーネル セグメンテーション ネットワークである MedNeXt を紹介します。これは、1) 医療画像セグメンテーションのための完全な ConvNeXt 3D エンコーダ/デコーダ ネットワーク、2) スケール全体でセマンティックな豊かさを維持するための残余 ConvNeXt アップおよびダウンサンプリング ブロック、3) 新しい技術を導入します。小規模なカーネル ネットワークをアップサンプリングすることでカーネル サイズを繰り返し増加し、限られた医療データでのパフォーマンスの飽和を防ぎます。 4) MedNeXt の複数のレベル (深さ、幅、カーネル サイズ) での複合スケーリング。これにより、CT および MRI モダリティとさまざまなデータセット サイズに関する 4 つのタスクで最先端のパフォーマンスが実現され、医療画像セグメンテーションのための最新のディープ アーキテクチャが実現します。私たちのコードは https://github.com/MIC-DKFZ/MedNeXt で公開されています。
There has been exploding interest in embracing Transformer-based architectures for medical image segmentation. However, the lack of large-scale annotated medical datasets make achieving performances equivalent to those in natural images challenging. Convolutional networks, in contrast, have higher inductive biases and consequently, are easily trainable to high performance. Recently, the ConvNeXt architecture attempted to modernize the standard ConvNet by mirroring Transformer blocks. In this work, we improve upon this to design a modernized and scalable convolutional architecture customized to challenges of data-scarce medical settings. We introduce MedNeXt, a Transformer-inspired large kernel segmentation network which introduces - 1) A fully ConvNeXt 3D Encoder-Decoder Network for medical image segmentation, 2) Residual ConvNeXt up and downsampling blocks to preserve semantic richness across scales, 3) A novel technique to iteratively increase kernel sizes by upsampling small kernel networks, to prevent performance saturation on limited medical data, 4) Compound scaling at multiple levels (depth, width, kernel size) of MedNeXt. This leads to state-of-the-art performance on 4 tasks on CT and MRI modalities and varying dataset sizes, representing a modernized deep architecture for medical image segmentation. Our code is made publicly available at: https://github.com/MIC-DKFZ/MedNeXt.