Transformer (masinõpe)

Transformer on süvaõppe mudel, mis kasutab sisemise tähelepanu mehhanismi, et differentsiaalselt kaaluda eraldi kõigi sisendandmete osatähtsust kontekstis kõigi teiste sisendandmetega. Seda kasutatakse peamiselt keeletehnoloogias ja arvutinägemises. Transformeri arhitektuuri pakkusid välja 2017. aastal Google Brain uurimisrühm artiklis "Attention Is All You Need". Nende peamine avastus seisnes selles, et lihtne närvivõrk, mis jätab ära rekurrentsuse ja konvulatsioonid ning kasutab nende asemel ainult sisemise tähelepanu mehhanisme, saab masintõlke ülesannetel paremaid tulemusi ning on rohkem paralleliseeritav, mis vähendab treenimisele kuluvat aega. Tähelepanu mehhanism on transformer närvivõrgu kõige tähtsam osa, sest see vastutab selle eest, et võrk keskenduks väljundvektorit genereerides õigele sisendvektori osale. Näiteks lauses "Pall kukkus naabri aeda, sest poiss lõi seda liiga tugevasti." viitab sõna seda sõnale pall ja mitte sõnale aed. Sisemine tähelepanu loob selle seose pall-seda, mis aitab lauset töödeldes konteksti ja loogilist järjepidavust hoida. Seosed arvutatakse kõigi kodeerija sisendi sõnade jaoks. Seda tehakse arvutades kolm maatriksi: päringukaalud W Q {\displaystyle W_{Q}} , võtmekaalud W K {\displaystyle W_{K}} ja väärtuskaalud W V {\displaystyle W_{V}} . Iga tokeni i {\displaystyle i} jaoks korrutatakse sisendi sõna x i {\displaystyle x_{i}} iga kaalumaatriksiga, et saada päringuvektor q i = x i W Q {\displaystyle q_{i}=x_{i}W_{Q}} , võtmevektor k i = x i W K {\displaystyle k_{i}=x_{i}W_{K}} ja väärtusvektor v i = x i W V {\displaystyle v_{i}=x_{i}W_{V}} . Seejärel arvutatakse päringu- ja võtmevektorite skalaarkorrutised ehk skoorid; näiteks oleks esimese tokeni esimene skoor q 1 {\displaystyle q_{1}} ja k 1 {\displaystyle k_{1}} skalaarkorrutis ning teine skoor q 1 {\displaystyle q_{1}} ja k 2 {\displaystyle k_{2}} skalaarkorrutis. Saadud skoorid jagatakse võtmevektori pikkuse ruutjuurega, d k {\displaystyle {\sqrt {d_{k}}}} , ning tulemusel kasutatakse normaliseerimiseks softmax funktsiooni. Tehte saab kokkuvõtvalt kirjutada ühe softmax funktsiooniga maatriksarvutusena A ( Q , K , V ) = s o f t m a x ( Q K T d k ) V {\displaystyle A(Q,K,V)=softmax({\frac {QK^{T}}{\sqrt {d_{k}}}})V} . Sisemist tähelepanu on edasi arendatud algses artiklis välja käidud "mitmepealise tähelepanuga" (ingl multi-headed attention), mis laiendab mudeli oskust keskenduda erinevatele positsioonidele ning lisab tähelepanukihile mitu representatsioonialamruumi . Üks kolmik maatrikse ( W Q , W K , W V ) {\displaystyle (W_{Q},W_{K},W_{V})} on üks pea ning igal kihil on mitu pead. Iga kodeerija koosneb sisemise tähelepanu mehhanismist ja edasisöödu närvivõrgust (FFN). Esimese kodeerija sisemise tähelepanu mehhanism võtab sisse sisendjärjendi koos positsioonilise informatsiooniga ning ülejäänud kodeerijad saavad eelnevast kodeerijast sisendkodeeringu, millega tekitavad kaaluseoste alusel väljundkodeeringu. Seejärel töötleb edasisööduvõrk igat väljundkodeeringut eraldi; tulemus läheb edasi järgmisesse kodeerijasse või dekodeerijasse. Kodeerija on kahesuunaline ehk tähelepanu saab suunata nii praegusest varasematele kui hilisematele tokenitele. Dekodeerija kasutab lisaks sisemisele tähelepanule ja edasisööduvõrgule ka kodeerija-dekodeerija tähelepanu kihti, mis soodustab õigele sisendjärjendi kohale keskendumist. Dekodeerijas on tähelepanu suunatud vaid varasematele tokenitele. Transformer arhitektuur on teinud suure läbilöögi keeletehnoloogias, asendades rekurrentseid närvivõrke näiteks masintõlkes ja aegridade prognoosis . Laialdast populaarsust on leidnud mitmed eeltreenitud ja avalikusele kättesaadavad keelemudelid, näiteks OpenAI GPT-2, GPT-3 ning ChatGPT või BERT ja RoBERTa. Need mudelid on näidanud lootustandvaid tulemusi mitmetel keeletehnoloogilistel ülesannetel, näiteks valdkondades: masintõlge juturobotid automaatsed kokkuvõtted DNA ja valkude järjestuste analüüs videoklassifitseerimine Transformer mudel on implementeeritud TensorFlow Tensor2Tensor teegis ja Harvardi keeletehnoloogia uurimisrühma versioonina PyTorchis. Vaswani, Ashish; Shazeer, Noam; Parmar, Niki; Uszkoreit, Jakob; Jones, Llion; Gomez, Aidan N.; Kaiser, Lukasz; Polosukhin, Illia (6. detsember 2017). "Attention Is All You Need". arXiv. Vaadatud 16. detsember 2022.{{netiviide}}: CS1 hooldus: mitu nime: autorite loend (link) He, Cheng (8. detsember 2020). "Transformer in CV". towardsdatascience.com. Vaadatud 16. detsember 2022. Wood, Thomas. "What is a Transformer Neural Network?". deepai.org. Vaadatud 16. detsember 2022. Alammar, Jay (27. juuni 2018). "The Illustrated Transformer [Blog Post]". Jay Alammar Blog. Vaadatud 18. detsember 2022. Maxime (4. jaanuar 2019). "What is a Transformer?". Inside Machine learning. Vaadatud 18. detsember 2022. Jagtap, Rohan (11. juuni 2020). "Abstractive Text Summarization Using Transformers". The Startup. Vaadatud 18. detsember 2022. Rives, Alexander; Meier, Joshua (15. detsember 2020). "Biological structure and function emerge from scaling unsupervised learning to 250 million protein sequences". bioRxiv. Vaadatud 18. detsember 2022.{{netiviide}}: CS1 hooldus: mitu nime: autorite loend (link) Wang, H.; Bertasius, G.; Torresani, L (9. veebruar 2021). "Is Space-Time Attention All You Need for Video Understanding?". arXiv. Vaadatud 18. detsember 2022.{{netiviide}}: CS1 hooldus: mitu nime: autorite loend (link)