Donate to arXiv

ápr 30, 2021
admin

Title:Deep Residual Learning for Image Recognition

Authors:Kaiming He, Xiangyu Zhang, Shaoqing Ren, Jian Sun

Download PDF

Abstract: Deeper neural networks are more difficult to train. Bemutatunk egy maradéktanulási keretrendszert, amely megkönnyíti a korábban használtaknál lényegesen mélyebb hálózatok képzését. Kifejezetten újrafogalmazzuk a rétegeket úgy, hogy a rétegek bemeneteire hivatkozva reziduális függvényeket tanulunk, ahelyett, hogy nem hivatkozott függvényeket tanulnánk. Átfogó empirikus bizonyítékokat szolgáltatunk arra, hogy ezeket a reziduális hálózatokat könnyebb optimalizálni, és a jelentősen megnövelt mélységből is pontosságot lehet nyerni. Az ImageNet adathalmazon akár 152 réteg mélységű reziduális hálókat értékelünk – 8x mélyebbeket, mint a VGG-hálók, de még mindig alacsonyabb komplexitással. E maradék hálók együttesével 3,57%-os hibát érünk el az ImageNet tesztkészleten. Ez az eredmény az 1. helyet szerezte meg azILSVRC 2015 osztályozási feladatán. Bemutatjuk a 100 és 1000 rétegű CIFAR-10-en végzett elemzést is.
A reprezentációk mélysége központi jelentőségű számos vizuálisfelismerési feladatban. Kizárólag a rendkívül mély reprezentációinknak köszönhetően 28%-os relatív javulást értünk el a COCO tárgyfelismerési adathalmazon. A mély reziduális hálók az ILSVRC & COCO 2015 versenyekre benyújtott pályázataink alapját képezik,ahol szintén 1. helyezést értünk el az ImageNet detektálási, ImageNetlokalizációs, COCO detektálási és COCO szegmentálási feladatokban.

Vélemény, hozzászólás?

Az e-mail-címet nem tesszük közzé.