Розуміння кількості параметрів в нейромережі є важливим для проектування та аналізу моделей. Параметри включають ваги та упередження, які мережа навчається під час тренувань. Цей посібник пояснює основні кроки для визначення загальної кількості параметрів в моделі нейромережі.

Параметри в повнорозмірних шарах

У повністю підключеному (щільному) шарі кожен вхід з'єднує до кожного нейрона з вагою, і кожен нейрон має дворівневий термін. Загальна кількість параметрів в такому шарі розраховується як:

Кількість параметрів = (Кількість входів × Кількість нейронів) + Кількість нейронів

Розрахунок параметрів в конв'юнктурних шарах

Для конвюктивних шарів, параметрів залежать від розміру фільтра, кількості фільтрів, і вхідних каналів. Формула:

Кількість параметрів = (Висота фільтра × Ширина фільтра × Кількість вхідних каналів × Кількість фільтрів) + Кількість фільтрів

Всього параметрів в моделі

Щоб знайти загальну кількість параметрів в нейромережі, підведіть параметри всіх окремих шарів. Це забезпечує загальний вимір складності моделі та ємності.

Приклад розрахунку

Розглянемо модель з одним повністю підключеним шаром, що отримує 100 вхідних і виробляє 50 виходів. Кількість параметрів:

  • (100 вводів × 50 виходів) + 50 ваза = 5,050 параметри

Цей розрахунок допомагає розуміння розмірів та вимог до підготовки моделі.