Как изменить предобученную нейросеть с классификации на регрессию
✅ Ответ: используем transfer learning — перенос знаний с одной задачи на другую.
Что делаем?
🔹 Заменяем последний полносвязный слой и Softmax (отвечающий за классификацию) на один нейрон (или новый полносвязный слой) для регрессии.
🔹 Опционально замораживаем первые слои, если данных мало или нужна быстрая сходимость.
🔹 Обучаем сеть на новых данных с функцией потерь для регрессии.
Таким образом, мы сохраняем мощные фичи первых слоев, обученные на огромных датасетах, но адаптируем выход под задачу регрессии.
Post #830
1.02K

- ❤ 8
- 👍 3