arXiv reaDer
SegPrompt: セグメンテーション マップをより良いプロンプトとして使用して、腎臓結石分類用のディープ モデルを微調整する
SegPrompt: Using Segmentation Map as a Better Prompt to Finetune Deep Models for Kidney Stone Classification
最近、深層学習は、内視鏡画像を使用した腎臓結石の分類に有望な結果をもたらしました。ただし、注釈付きトレーニング データの不足は、トレーニング済みモデルのパフォーマンスと一般化能力を向上させる上で深刻な問題を引き起こします。したがって、手元にある限られたデータを最大限に活用することが重要です。この論文では、セグメンテーション マップを 2 つの側面から活用することで、データ不足の問題を軽減する SegPrompt を提案します。まず、SegPrompt はセグメンテーション マップを統合して分類トレーニングを容易にし、分類モデルが関心領域を認識できるようにします。提案された方法により、画像トークンとセグメンテーショントークンが相互に対話して、セグメンテーションマップ情報を完全に利用できます。次に、セグメンテーション マップをプロンプトとして使用して、事前トレーニング済みのディープ モデルを調整します。その結果、通常の微調整よりもトレーニング可能なパラメーターがはるかに少なくなります。収集された腎臓結石データセットに対して広範な実験を行います。結果は、SegPrompt がモデル フィッティング能力と一般化能力の間で有利なバランスを達成できることを示しており、最終的には限られたトレーニング データで効果的なモデルを導きます。
Recently, deep learning has produced encouraging results for kidney stone classification using endoscope images. However, the shortage of annotated training data poses a severe problem in improving the performance and generalization ability of the trained model. It is thus crucial to fully exploit the limited data at hand. In this paper, we propose SegPrompt to alleviate the data shortage problems by exploiting segmentation maps from two aspects. First, SegPrompt integrates segmentation maps to facilitate classification training so that the classification model is aware of the regions of interest. The proposed method allows the image and segmentation tokens to interact with each other to fully utilize the segmentation map information. Second, we use the segmentation maps as prompts to tune the pretrained deep model, resulting in much fewer trainable parameters than vanilla finetuning. We perform extensive experiments on the collected kidney stone dataset. The results show that SegPrompt can achieve an advantageous balance between the model fitting ability and the generalization ability, eventually leading to an effective model with limited training data.
updated: Wed Mar 15 2023 01:30:48 GMT+0000 (UTC)
published: Wed Mar 15 2023 01:30:48 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アソシエイト