Как уменьшить перепроизводство при использовании VGG16 для регрессии? - PullRequest
0 голосов
/ 06 ноября 2018

Я использую трансферное обучение из VGG16 для регрессионной задачи, но я очень быстро перенимаю. Я хочу уменьшить количество параметров для регрессии (последний слой), как я могу это сделать?

1 Ответ

0 голосов
/ 07 ноября 2018

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

Хотя вы должны иметь в виду, что веса VGG16 были обучены с использованием функции потерь, подходящей для задач классификации (например, категориальной кроссентропии), которая создает у меня впечатление, что ваша модель будет только угадывать длину, основываясь на том, какой объект (который может Можно даже подумать, но большому автомобилю, похожему на насекомое, можно придать меньший размер, чем маленькому автомобилю, похожему на здание).

...