Înțelegerea modului de calcul al numărului de parametri într-o rețea neurală este esențială pentru proiectarea și analiza modelelor. Parametrii includ greutăți și prejudecăți pe care rețeaua le învață în timpul formării. Acest ghid explică pașii de bază pentru a determina numărul total de parametri într-un model de rețea neuronală.

Parametrii în straturi complet conectate

Într-un strat complet conectat (dens), fiecare intrare se conectează la fiecare neuron cu o greutate, și fiecare neuron are un termen de prejudecată. Numărul total de parametri într-un astfel de strat este calculat ca:

Numărul de parametri = (Numărul de intrări × Numărul de neuroni) + numărul de neuroni

Calcularea parametrilor în straturile de evoluție

Pentru straturile de convoluție, parametrii depind de dimensiunea filtrului, numărul de filtre și canalele de intrare. Formula este:

Numărul de parametri = (înălțime de filtrare × lățime de filtru × Numărul de canale de intrare × Numărul de filtre) + Numărul de filtre

Parametrii totali în model

Pentru a găsi numărul total de parametri într-o rețea neurală, suma parametrii tuturor straturilor individuale. Aceasta oferă o măsură generală a complexității și capacității modelului.

Calculul de exemplu

Luați în considerare un model cu un strat complet conectat care primește 100 de intrări și produce 50 de ieșiri. Numărul de parametri este:

  • (100 intrări × 50 ieșiri) + 50 prejudecată = 5,050 parametri

Acest calcul ajută la înțelegerea cerințelor de dimensiune și formare ale modelului.