ディープラーニングを人工知能(AI)システムに適用する際の主な制限は、高品質のキュレートされたデータセットの不足です。この問題に対処するために、強力な拡張ベースの自己監視学習(SSL)手法を調査します。例として乳がんの検出を使用して、最初にマンモグラム固有の変換パラダイムを特定し、次に、さまざまなアプローチを表す4つの最近のSSLメソッドを体系的に比較します。事前にトレーニングされたモデルを、均一に並べて表示されたパッチの予測から画像全体に変換する方法と、分類パフォーマンスを向上させる注意ベースのプーリング方法を開発します。最良のSSLモデルは、ベースラインの監視ありモデルを大幅に上回っています。最高のSSLモデルは、サンプルラベリングのデータ効率をほぼ4倍向上させ、あるデータセットから別のデータセットに高度に転送可能でした。 SSLは、コンピュータービジョンの大きな進歩であり、医用画像分野のAIが、教師あり学習や希少なラベルへの依存から脱却するのに役立つ可能性があります。
A major limitation in applying deep learning to artificial intelligence (AI) systems is the scarcity of high-quality curated datasets. We investigate strong augmentation based self-supervised learning (SSL) techniques to address this problem. Using breast cancer detection as an example, we first identify a mammogram-specific transformation paradigm and then systematically compare four recent SSL methods representing a diversity of approaches. We develop a method to convert a pretrained model from making predictions on uniformly tiled patches to whole images, and an attention-based pooling method that improves the classification performance. We found that the best SSL model substantially outperformed the baseline supervised model. The best SSL model also improved the data efficiency of sample labeling by nearly 4-fold and was highly transferrable from one dataset to another. SSL represents a major breakthrough in computer vision and may help the AI for medical imaging field to shift away from supervised learning and dependency on scarce labels.