arXiv reaDer
クラス不均衡下でのニューラルネットの過剰適合:セグメンテーションの分析と改善
Overfitting of neural nets under class imbalance: Analysis and improvements for segmentation
  ディープラーニングの過剰適合は最近の多くの研究の焦点でしたが、ニューラルネットワークの動作に対するその正確な影響はよく理解されていません。この研究では、ロジットの分布がモデルの過剰適合に関連してどのように変化するかを調べることにより、過剰適合を分析します。具体的には、少数のデータサンプルでトレーニングする場合、過少表現クラスの見えないテストサンプルを処理するときのロジットアクティベーションの分布は、決定境界に向かって、さらには決定境界を越えてシフトする傾向がありますが、過大表現クラスは影響を受けないようです。画像のセグメンテーションでは、前景のサンプルが頻繁に過小表現されます。モデルの感度はオーバーフィッティングの結果として低下しますが、精度はほとんど安定しています。私たちの分析に基づいて、既存の損失関数と大きなマージン損失、焦点損失、敵対者訓練、および混同を含む正則化器の非対称的な修正を導き出します。これは、未表示のクラスの見えないサンプルを埋め込むときに観察されるシフトを減らすことを特に目的としています。脳腫瘍コアのバイナリセグメンテーションのケースを研究し、我々の提案された単純な変更が対称バリアントを介して大幅に改善されたセグメンテーションパフォーマンスにつながることを示します。
Overfitting in deep learning has been the focus of a number of recent works, yet its exact impact on the behavior of neural networks is not well understood. This study analyzes overfitting by examining how the distribution of logits alters in relation to how much the model overfits. Specifically, we find that when training with few data samples, the distribution of logit activations when processing unseen test samples of an under-represented class tends to shift towards and even across the decision boundary, while the over-represented class seems unaffected. In image segmentation, foreground samples are often heavily under-represented. We observe that sensitivity of the model drops as a result of overfitting, while precision remains mostly stable. Based on our analysis, we derive asymmetric modifications of existing loss functions and regularizers including a large margin loss, focal loss, adversarial training and mixup, which specifically aim at reducing the shift observed when embedding unseen samples of the under-represented class. We study the case of binary segmentation of brain tumor core and show that our proposed simple modifications lead to significantly improved segmentation performance over the symmetric variants.
updated: Tue Oct 01 2019 13:22:58 GMT+0000 (UTC)
published: Thu Jul 25 2019 11:47:12 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト