CNNで精度の高いラベル上位nこを取り出す方法

Question

CNNで例えばcifar10をデータセットとして使ったとします。
この時出力としては最も精度が高い画像が返ってくるはずですが、他のラベルも精度何%で判別されたはずです。
この時、精度が上位3つとか精度が高い順に複数のラベルを返す事例はないのでしょうか？

何か参考になるサイトでも意見、同様の手法なんでも構いません。
アドバイスをいただけないでしょうか？
ご教授お願いします。

Answer 1

一般的にsoftmax関数を使って、各クラスである確率を計算し、その後にargmaxを使ってクラスを決定するというコードになっていると思うので、softmax直後の値をsess.runを使って取り出せば自由に加工できます。
tensorflow内の関数を使って加工した結果をsess.runで取得する手もありますが、こうした加工はtesorflowの外側のアプリの役割と思っているのでお勧めしません

Comments

[mkgrei]

ただ、top_kを損失関数に含めて学習させることによって、より上位3つのうちに正解が含まれるモデルを作ることができるようになるかもしれません。
top_kを取り出すことが目的なのか、top_kに正解が含まれることが目的なのかによりますが。

[R.Shigemori]

そういう使い方ならtensorflowの関数を使うべきですね。

Answer 2

精度というか確率でしょうか（出力層にsoftmaxを使った場合の話）。
出力層の生の出力を得るのは簡単にできるでしょうし、上位幾つかがほしいのであればソートして紐付いたラベルを返せば良いでしょう。
やり方はフレームワーク依存ですが、たとえばtensorflowならtop_kという関数でできる、らしいです（自分でやったことはないので実際どんなものなのかはよくわかりません、悪しからず）。自分で書いても大して難しくはないでしょう。