畳み込みニューラルネットワーク(CNN)は、医用画像を含むさまざまな分野の画像分類に広く使用されています。ほとんどの研究は、そのようなタスクの損失関数としてクロスエントロピーを展開していますが、対照的な学習ベースの損失のファミリーに目を向けるアプローチが増えています。 CNN分類器の評価には、精度、感度、特異性などのパフォーマンスメトリックが定期的に使用されますが、これらの分類器が実際に学習する機能が特定されることはめったになく、配布外のテストサンプルの分類パフォーマンスへの影響については十分に調査されていません。この論文では、肺結節分類の実際のタスクに動機付けられて、制御された変動モードを持つ合成データセットのさまざまな分布でトレーニングおよびテストされたときにCNNが学習する機能を調査します。さまざまな損失関数がさまざまな機能の学習につながり、その結果、見えないデータに対する分類器の一般化能力に影響を与えることを示します。この研究は、医用画像タスクの深層学習ソリューションの設計に関するいくつかの重要な洞察を提供します。
Convolutional Neural Networks (CNNs) are widely used for image classification in a variety of fields, including medical imaging. While most studies deploy cross-entropy as the loss function in such tasks, a growing number of approaches have turned to a family of contrastive learning-based losses. Even though performance metrics such as accuracy, sensitivity and specificity are regularly used for the evaluation of CNN classifiers, the features that these classifiers actually learn are rarely identified and their effect on the classification performance on out-of-distribution test samples is insufficiently explored. In this paper, motivated by the real-world task of lung nodule classification, we investigate the features that a CNN learns when trained and tested on different distributions of a synthetic dataset with controlled modes of variation. We show that different loss functions lead to different features being learned and consequently affect the generalization ability of the classifier on unseen data. This study provides some important insights into the design of deep learning solutions for medical imaging tasks.