Nhungamiro yehunyanzvi

Stochastic Weight Avhareji

Stochastic Weight Averaging (SWA) inotora avhareji yakapusa yehuremu hwemodhi kubva akati wandei mapoinzi kunonoka mukudzidziswa pane kungochengeta mufananidzo wekupedzisira.

2 min verengaLast update

Pfupiso

This cheap trick often lands the model in a flatter, wider region of the loss landscape, which tends to generalize noticeably better on unseen data.

Kudzika Kwakadzika

Yakaunzwa naIzmailov, Wilson nevamwe vaaishanda navo muna 2018, SWA inoshandisa fungidziro yekuti SGD ine chisingaperi kana cyclical kudzidza mwero haichinji kune imwe nzvimbo - inosvetuka ichitenderedza kumucheto kwemupata wakafara, wakafuratira. Panzvimbo pekutora imwe yeidzo nzvimbo dzekumisa dzine ruzha, SWA inomhanya zvine mwero (kazhinji inogara kana kutenderera) mwero wekudzidza wenguva yekupedzisira uye avhareji huremu hwainoshanyira, kazhinji nguva yega yega. Huremu hweavhareji hunogara padyo nepakati penzvimbo yakati sandara. Nekuda kwekuti batch-normalization statistics inoverengerwa kune chaiwo huremu, SWA inoda imwe yekuwedzera kumberi kupfuura data kudzoreredza BN inomhanya nzira uye misiyano yeavhareji modhi. Mutengo wacho ndewemahara, uye kuwana kwechokwadi kunopindirana mumhando dzemifananidzo uye nekupfuura.

Technical Insight

SWA inochengetedza avhareji inomhanya w_SWA = (n·w_SWA + w_i)/(n+1) yakagadziridza kutenderera kwega kwega, nepo mhenyu yeSGD modhi inoramba ichiongorora nemwero wakakura wekudzidza. Avhareji yehuremu nzvimbo inokwana ensemble munzvimbo yebasa asi inodhura imwe modhi pakunongedza, kwete mazhinji. Iyo yakakosha meshini ndeyekuti flat minima yakasimba kune huremu kukanganisa, saka iyo yekudzidzira / bvunzo yekurasikirwa nzvimbo inogara yakabatana, ichidzikisa gaka rekuita.

Strategic Impact

Mutengo uye bhajeti

Zvisarudzo zvezvivakwa zvinotyaira kuita uye mutengo wekushandisa kwemakore.

Sarudzo dzakajeka

Dzidzo yehunyanzvi inobatsira zvikwata kusarudza murwi wakakodzera, kwete iwo mutsva chete.

Kudzora kwemhando yepamusoro

Sarudzo dzeinjiniya dziri nani dzinoderedza zviitiko zvekuvimbika mukugadzira.

Ramangwana reStochastic Weight Averaging

SWA yakaburitsa akasiyana seSWA-Gaussian (SWAG) yekusagadzikana kweBayesi yakachipa, uye iyo pfungwa yeavhareji ikozvino inosimbisa Exponential Moving Average matipi anoshandiswa zvakanyanya mumhando dzekuparadzira, kudzidza kwekuzvitarisira, uye hombe-modhi yekudzidzira. Tarisira kuti huremu hurambe huri 'yemahara yekudya kwemasikati' mukudzidzira mabikirwo, netsvakiridzo inoiwedzera kusvika pakubatanidza mamodheru akadzidzira (modhi soups) uye kuvandudza kuenzanisa padivi pechokwadi mbishi.

Real-World Implementation

Kuwedzera bvunzo kurongeka kweResNet uye DenseNet mufananidzo classifiers paCIFAR uye ImageNet pasina imwe yekuwedzera mutengo.

SWAG (SWA-Gaussian) inogadzira fungidziro yekusavimbika yakayerwa yekufembera inotarisisa kubva kune imwechete kudzidziswa kumhanya.

EMA-ye-huremu inodzikamisa iyo sampling network mukuparadzira mifananidzo jenareta seStable Diffusion.

Kugadzira 'soups yemuenzaniso' neaverage yakawanda-yakanyatso gadziridzwa nzvimbo dzekutarisa kuti uvandudze kusimba pasina kudzidziswazve.

Njodzi & Guardrails

Kugadzirisa imwe bhenji kunogona kuvanza yakafara system kushaya simba.

Infrastructure uye mari yekugadzirisa inowanzotarisirwa pasi.

Chengetedzo uye kucherechedzwa mapundu anogona kukura sezvo masisitimu anowedzera kuoma.

Implementation Roadmap

1

Tsanangura latency, mhando, uye mutengo zvinangwa usati waitwa.

2

Benchmark pasi pechokwadi mutoro uye data mamiriro.

3

Chishandiso chekutarisa zvikanganiso, kudonha, uye mushandisi maitiro.

4

Gadzirira nzira dzekudzosera kumashure uye dzezviitiko usati wawedzera.

Ramba Uchiongorora

Free newsletter

Get the daily AI briefing

Three verified AI stories every weekday morning, written in plain English. Free forever, no ads.

One email each weekday. Unsubscribe in one click. We never sell or share your address.

Test yourself

Take the Stochastic Weight Averaging quiz

Instant feedback on every answer, and a shareable certificate with a verifiable ID once you pass a course.

Tanga mibvunzo

Support free AI education. AI Understanding is a 501(c)(3) nonprofit — no ads, no paywall, ever. Make a donation

Gaidhi rinotevera

Weight Initialization

Mibvunzo inowanzo bvunzwa

What is Stochastic Weight Averaging?

Stochastic Weight Averaging (SWA) inotora avhareji yakapusa yehuremu hwemodhi kubva akati wandei mapoinzi kunonoka mukudzidziswa pane kungochengeta mufananidzo wekupedzisira. Uhu hunyengeri hwakachipa hunowanzo dzika modhi munzvimbo yakapfava, yakafara nzvimbo yekurasika, iyo inowanzo wedzera zvinooneka zvirinani pane isingaonekwe data.

Chii chinonzi Stochastic Weight Averaging chaizvo?

SWA inoyera mapeji emuenzaniso (zviremu) zvakaunganidzwa panzvimbo dzinoverengeka dzekutarisa panguva yekupedzisira yekudzidziswa, kwete kufanotaura kana gradients.

Sei SWA ichida kuvandudza generalization?

Avhareji inofambisa mhinduro yakananga pakati penzvimbo yakati sandara yenzvimbo yekurasikirwa, uye flat minima generalize zvirinani nekuti chitima uye kurasikirwa kwebvunzo kunogara kwakabatana.

Nderipi rimwe danho rinodiwa neSWA kune network inoshandisa batch normalization?

Nekuti huwandu hweBN hunoenderana nehuremu chaihwo, iyo yeavhareji modhi inoda imwe yekuwedzera kupfuura data kuti idzokorore nzira dzekumhanya uye kusiyana.

Ndeapi maitiro ekudzidza-chiyero anowanzo shandiswa panguva yeavhareji yeSWA chikamu?

SWA inochengeta mwero wakakwira zvine mwero kana cyclic yekudzidza kuitira kuti SGD ienderere mberi ichiongorora nzvimbo yakafara furati ine mapoinzi anobva apinzwa pakati.

Mari yekufungidzira yeSWA inofananidzwa sei neyechinyakare ensemble yeN modhi?

SWA inodonhedza macheki akawanda kuita avhareji huremu seti, saka kufungidzira kunodhura zvakafanana nemhando imwe chete kwete N.