Új módszerrel gyorsítja a 3D-s GCNN-eket az ArXiv kutatása
A módszer a geometriai felbontást függetleníti a memória- és feldolgozási költségektől, így a betanítás és az inferencia is hatékonyabbá válik.
Jelentősen felgyorsíthatja a 3D geometriai adatok feldolgozását a csoport-konvolúciós neurális hálózatokban (GCNN) egy új módszer, amely a mintavételezést a jellemzőtérbe helyezi át — írja az ArXiv-en megjelent tanulmány.
A GCNN-ek kulcsfontosságúak a szimmetria bevezetésében a mélytanulásban, mivel minden lineáris rétegben sűrűn mintavételezik a transzformációs csoportokat, és különböző pózokban korrelálják az adatokat és a szűrőket. Ez a megközelítés azonban rendkívül költséges, különösen a 3D-s adatok, például a fordítások és forgatások esetén, ahol a költségek exponenciálisan nőnek a szabadsági fokok számával.
A számítási költségek labirintusa
A kutatók szerint a probléma megoldása az, hogy a geometriailag sűrű mintákat jellemző hasonlóság alapján kiválasztott reprezentatív mintákkal helyettesítik. Ez a megközelítés lehetővé teszi a geometriai felbontás és a számítási költségek szétválasztását, ami eddig akadályozta a GCNN-ek gyakorlati alkalmazását a komplex 3D-s feladatokban.
A hatékonyság új távlatban
Az új módszerrel a GCNN-ek sokkal hatékonyabban alkalmazhatók olyan területeken, mint a 3D-s molekuláris struktúrák elemzése, ahol a SchNet és DimeNet korábbi architektúrái már bevezették a folyamatos szűrős konvolúciós rétegeket, és a kutatás 2024. januárjában kerül befejezésre az ArXiv kutatócsoport által.