Тренинг CNN для 3 классов, однако общие классы 4

Question

Я тренирую CNN для классификации изображений.В общем, мои изображения в 4 разных категориях.Однако образцы для последнего класса действительно редки.

Мне было интересно, можно ли обучить его для 3 классов и сказал, что если он не принадлежит к этим группам, то он принадлежит к группе 4?

Answer 1

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

Не тренироваться для него довольно сложно, потому что даже для Imagenet (1000 классов) трудно тренировать"не из вышеупомянутого" класса.Что представляет собой «не яблоко»?

Другое решение состоит в том, чтобы избавиться от вашего слоя softmax и просто использовать выходные данные (возможно, с exp), а затем интерпретировать значения (если оно низкое в первые 3 ")может быть "это 4?)

Answer 2

Вы можете тренировать CNN 3 класса.Когда вы делаете прогноз на новом изображении, вы можете проверить уровень достоверности прогноза, и если он меньше порогового значения, вы можете классифицировать изображение как принадлежащее к 4-му классу.Надеюсь, это поможет!