arXiv reaDer
コンピュータ支援疾患評価における画像回帰の適応コントラスト
Adaptive Contrast for Image Regression in Computer-Aided Disease Assessment
骨塩密度(BMD)の推定や左心室駆出率(LVEF)の予測など、医療アプリケーションの画像回帰タスクは、コンピューター支援疾患の評価において重要な役割を果たします。ほとんどの深回帰法は、MSEやL1損失などの単一の回帰損失関数を使用してニューラルネットワークをトレーニングします。本論文では、深部画像回帰のための最初の対比学習フレームワーク、すなわちAdaConを提案します。これは、新しい適応マージン対比損失を介した特徴学習ブランチと回帰予測ブランチで構成されます。私たちの方法は、学習した特徴表現の一部としてラベル距離の関係を組み込んでいます。これにより、ダウンストリームの回帰タスクでのパフォーマンスが向上します。さらに、既存の回帰メソッドのパフォーマンスを向上させるためのプラグアンドプレイモジュールとして使用できます。 2つの医用画像回帰タスク、つまりX線画像からの骨ミネラル密度推定と心エコービデオからの左心室駆出率予測に対するAdaConの有効性を示します。 AdaConは、最先端のBMD推定およびLVEF予測方法に比べて、MAEでそれぞれ3.3%および5.9%の相対的な改善をもたらします。
Image regression tasks for medical applications, such as bone mineral density (BMD) estimation and left-ventricular ejection fraction (LVEF) prediction, play an important role in computer-aided disease assessment. Most deep regression methods train the neural network with a single regression loss function like MSE or L1 loss. In this paper, we propose the first contrastive learning framework for deep image regression, namely AdaCon, which consists of a feature learning branch via a novel adaptive-margin contrastive loss and a regression prediction branch. Our method incorporates label distance relationships as part of the learned feature representations, which allows for better performance in downstream regression tasks. Moreover, it can be used as a plug-and-play module to improve performance of existing regression methods. We demonstrate the effectiveness of AdaCon on two medical image regression tasks, ie, bone mineral density estimation from X-ray images and left-ventricular ejection fraction prediction from echocardiogram videos. AdaCon leads to relative improvements of 3.3% and 5.9% in MAE over state-of-the-art BMD estimation and LVEF prediction methods, respectively.
updated: Wed Dec 22 2021 07:13:02 GMT+0000 (UTC)
published: Wed Dec 22 2021 07:13:02 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アソシエイト