ディープラーニングは、乳がんのマンモグラフィベースのコンピュータ支援診断(CAD)を促進できますが、一般的にサンプルサイズが小さいという問題があります。自己教師あり学習(SSL)は、限られたトレーニングサンプルでの医療画像分析でその有効性を示しています。ただし、ネットワークモデルは、口実タスクと微調整メカニズムの制限により、従来のSSLフレームワークで十分に事前トレーニングできない場合があります。この作業では、マンモグラフィベースのCADの分類モデルのパフォーマンスを向上させるために、タスク駆動型の自己教師ありバイチャネルネットワーク(TSBN)フレームワークを提案します。特に、新しいグレースケール画像マッピング(GSIM)が口実タスクとして設計されており、マンモグラムのクラスラベル情報を画像復元タスクに埋め込んで、識別可能な特徴表現を改善します。次に、提案されたTSBNは、画像復元ネットワークや分類ネットワークなど、さまざまなネットワークアーキテクチャを統合SSLフレームワークに革新的に統合します。バイチャネルネットワークモデルを共同でトレーニングし、診断精度を向上させて、知識を口実タスクネットワークからダウンストリームタスクネットワークに共同で転送します。提案されたTSBNは、公開されているINbreastマンモグラムデータセットで評価されます。実験結果は、限られたサンプルで乳がんを診断するための従来のSSLおよびマルチタスク学習アルゴリズムよりも優れていることを示しています。
Deep learning can promote the mammography-based computer-aided diagnosis (CAD) for breast cancers, but it generally suffers from the small sample size problem. Self-supervised learning (SSL) has shown its effectiveness in medical image analysis with limited training samples. However, the network model sometimes cannot be well pre-trained in the conventional SSL framework due to the limitation of the pretext task and fine-tuning mechanism. In this work, a Task-driven Self-supervised Bi-channel Networks (TSBN) framework is proposed to improve the performance of classification model the mammography-based CAD. In particular, a new gray-scale image mapping (GSIM) is designed as the pretext task, which embeds the class label information of mammograms into the image restoration task to improve discriminative feature representation. The proposed TSBN then innovatively integrates different network architecture, including the image restoration network and the classification network, into a unified SSL framework. It jointly trains the bi-channel network models and collaboratively transfers the knowledge from the pretext task network to the downstream task network with improved diagnostic accuracy. The proposed TSBN is evaluated on a public INbreast mammogram dataset. The experimental results indicate that it outperforms the conventional SSL and multi-task learning algorithms for diagnosis of breast cancers with limited samples.