Требуется ли для изучения передачи изображений классификация негативных примеров? - PullRequest
3 голосов
/ 21 февраля 2020

Задача - определить, к какому из 3 классов принадлежит изображение или нет.

Я получил готовую модель. Для Efficien tNet B4 с весами Imag eNet было применено трансферное обучение для определения 4 классов: 3 целевых и 4-го «Нет». Последние были обучены на примерах случайных изображений, не содержащих ни одного из целевых объектов.

Вопрос в том, правильный ли подход - нужен ли 4-й класс?

Моя интуиция заключается в том, что net должно быть тренируется только на 3 целевых классах. Если выходные вероятности остаются ниже некоторого порога (90%?), Изображение следует рассматривать как НЕ содержащее каких-либо целевых объектов. Я прав?

1 Ответ

4 голосов
/ 21 февраля 2020

Из-за характера функции softmax и способа обучения сети вам необходим четвертый класс.

Давайте рассмотрим конкретный пример: вы обучаете свою сеть различать guish между яблоками, апельсинами и бананами. Тем не менее, вы каким-то образом получаете фотографию сливы.

Вы можете удивиться с первого взгляда, но вам нужен другой класс в вашем наборе данных. Нет гарантии, что использование порогового значения поможет вам исключить другой класс.

Вы можете ожидать следующие два случая:

  1. Вероятность выхода гарантированно будет равна 1 / N для неизвестного класса, учитывая, что вы тестируете в неизвестном классе N + 1 .
  2. Определенный порог, после которого (как вы и предполагали) <90%, это не класс. </li>

Допустим следующие случаи:

  1. Что если у вас есть случай, когда яблоко действительно выглядит как апельсин, и ваша модель правильно предсказывает 40% яблока, 30% апельсина, 30% банана, но поскольку вы применили свой порог, правильно идентифицированное яблоко (True Positive) исключается? Простой случай, когда вы исключаете хороший результат работы вашей сети
  2. У вас все еще может быть присвоение классу 91%, хотя новое прибытие 'fruit' не является частью вашего набора данных; это связано с внутренними вычислениями и тем, как работает softmax.

Личный опыт: однажды я обучил сеть различать guish между многими типами знаков трафика c , Из чистого любопытства я привел в пример один стул для гостиной. Я ожидал того же, что и вы (порог), но, к моему большому удивлению, это был 85% «Путь доходности».

...