Middle — Senior
როგორ იძლევა SBERT ტიპის მოდელი ტრიპლეტ ლოსს?
sobes.tech AI
პასუხი AI-სგან
SBERT (Sentence-BERT) მოდელი ტრიპლეტული დანაკარგის გამოყენებით სწავლება, რათა ისწავლოს წინადადებების წარმოდგენა ვექტორულ სივრცეში ისე, რომ მსგავსი მნიშვნელობის წინადადებები იყოს ახლოს ერთმანეთთან, განსხვავებული კი — შორს.
ტრიპლეტული დანაკარგი მუშაობს სამ მაგალითზე: ანკორი (anchor), დადებითი მაგალითი (positive), რომელიც მსგავსია ანკორის, და უარყოფითი მაგალითი (negative), რომელიც განსხვავებულია. მიზანია მინიმუმამდე დაიყვანოს ანკორის და დადებითი მაგალითის შორის მანძილი და მაქსიმუმამდე — ანკორის და უარყოფითი მაგალითის შორის, გარკვეული მარკაჟით.
სწავლება შემდეგი წესით მიმდინარეობს:
- თითოეული ტრიპლეტისათვის SBERT-ის გამოყენებით გამოთვლილია ემბედინგები.
- გამოთვლილია მანძილი (მაგ., კოსინუსი) ანკორის და დადებითი, და ანკორის და უარყოფითი შორის.
- გამოითვლება ტრიპლეტული დანაკარგი: ( \max(0, d(ანკორი, დადებითი) - d(ანკორი, უარყოფითი) + მარკაჟი) ).
- მოდელის პარამეტრები განახლდება გრადიენტული დაშვებით, რათა შემცირდეს დანაკარგი.
ამ გზით, მოდელი სწავლობს შექმნას ისეთი ემბედინგები, რომლებიც სასარგებლოა ტექსტების ძიებისა და შედარების დავალებებისთვის.