arXiv reaDer
IKD+: 網膜症分類のための信頼性の高い複雑性の低いディープ モデル
IKD+: Reliable Low Complexity Deep Models For Retinopathy Classification
網膜症のディープ ニューラル ネットワーク (DNN) モデルでは、90% から高い予測精度が推定されています。ただし、次の側面は未解決のままです。予測の信頼性は大きく異なります。この論文では、これらの側面に焦点を当て、サイズ、精度、信頼性のトレードオフを組み込んだ IKD+ と呼ばれる反復知識蒸留 (IKD) の形式を提案します。モデルのキャリブレーションを推定するために広く使用されている 2 つの手法 (Platt スケーリングと温度スケーリング) を使用して、IKD+ の機能を調査します。温度スケーリングを備えた IKD+ は、精度を大幅に損なうことなく、元のアンサンブルよりもパラメーター数が最大約 500 倍減少するモデルをもたらすことを示しています。さらに、IKD+ モデルのキャリブレーション スコア (信頼性) は、基本モードと同等かそれ以上です。
Deep neural network (DNN) models for retinopathy have estimated predictive accuracies in the mid-to-high 90%. However, the following aspects remain unaddressed: State-of-the-art models are complex and require substantial computational infrastructure to train and deploy; The reliability of predictions can vary widely. In this paper, we focus on these aspects and propose a form of iterative knowledge distillation(IKD), called IKD+ that incorporates a tradeoff between size, accuracy and reliability. We investigate the functioning of IKD+ using two widely used techniques for estimating model calibration (Platt-scaling and temperature-scaling), using the best-performing model available, which is an ensemble of EfficientNets with approximately 100M parameters. We demonstrate that IKD+ equipped with temperature-scaling results in models that show up to approximately 500-fold decreases in the number of parameters than the original ensemble without a significant loss in accuracy. In addition, calibration scores (reliability) for the IKD+ models are as good as or better than the base mode
updated: Sat Mar 04 2023 03:59:06 GMT+0000 (UTC)
published: Sat Mar 04 2023 03:59:06 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アソシエイト