$$\rightleftharpoonup{xx}$$
$$\longleftharp{xx}$$,
$$\longrightharp{xx}$$,
Bu çalışmanın amacı, bir dil modelini (LM) ince ayarlayarak özetler üreterek halüsinasyonu azaltmaktır. Halüsinasyonlar, bir modelin metni gerçek doğrulama yerine istatistiksel olasılıkla ürettiği için "kendinden emin bir şekilde yanlış veya alakasız çıktı ürettiğinde" meydana gelir. Bu, Varlık Halüsinasyon Endeksi (EHI) temelli bir ödül fonksiyonu tasarlanarak ve bunu en üst düzeye çıkarmak için pekiştirici öğrenme uygulanarak sağlanır.
Problem formülasyonu
Bir giriş belgesi Xi verildiğinde, amaç, Yi'de bahsedilen varlıkların Xi'ye bağlı olacak bir özet Yi oluşturmaktır. Geleneksel maksimum olasılık eğitimi, gerçek tutarlılığı açıkça optimize etmez. Bu eğitim yöntemleri, uydurma varlıkların dahil edilmesini cezalandırmaz. Bu nedenle, ödülün varlığın sadakatiyle orantılı olduğu, EHI ile ölçüldüğünde, ödül odaklı bir ince ayar stratejisi getirilir.
Veri seti ve yöntem
İki korpus kullanılır: birincisi, çoklu turlu toplantı transkriptleri ve soyut altın özetlerden oluşan bir diyalog transkripsiyon veri seti, ikincisi ise XSUM haber özetlemekıyaslaması 22,23. Her veri seti temizlendi, düzleştirildi ve %80 tren, %10 doğrulama ve %10 test bölümlerine bölündü. Diyaloglar gayri resmi dil ve zamirlerle zenginken, XSUM kısa haber makaleleri içerir; Bu kombinasyon, hem alan içi hem de alan dışı genellemeyi değerlendirmemize olanak tanır.
Varlık çıkarımı, hem kaynak belgeler hem de özetler üzerinde gerçekleştirilir ve Varlık Halüsinasyon İndeksi (EHI) hesaplanır. İnce ayar sırasında, bu çalışma yalnızca eğitim bölünmesini kullanırken, doğrulama EHI puanları en iyi kontrol noktasını seçmek için izlenir. Son değerlendirmeler, modellerin ince ayarından önce ve sonra sıfır atış kalınlığında kalan tutturulan test setinde rapor edilir.
Özetle, öncelikle transkriptler düzleştirilmiş bir formatta düzenlenir ve veri bölünür. Önden eğitilmiş model temel özetler üretir. İkinci olarak, EHI, trafo tabanlı NER ve beş halüsinasyon faktörü kullanılarak hesaplanır. İnce ayar sırasında model, LoRa adaptörleri aracılığıyla EHI ödülü ile güncellenir. Son olarak, özetler ince ayarlanmış model kullanılarak oluşturulur ve EHI, Entity F1 ve diğer metriklerle değerlendirilir (Şekil 2). Tüm deneyler, sağlanan kod ve yapılandırma dosyaları aracılığıyla tekrarlanabilir. Şekil 3, veri seti hazırlama, temel özetleme, EHI hesaplaması, ince ayar ve değerlendirme için kademeli süreç akışını göstermektedir.
Temel özetleme
Üç önceden eğitilmiş LLM seçilir ve onların temel özetleri ele geçirilir: Flan-T5, Mistral (Nous-Hermes-2-Mistral-7B-DPO) ve DistilBART 24,25,26. Her model, her giriş belgesi (xi) için ilk özet (Yi) oluşturmak amacıyla sıfır atış modunda çalıştırıldı. 4 ışın genişliği ve 1.0 uzunluk cezası ile ışın araması, akıcı ama özlü özetleri teşvik etmek için kullanıldı. Bu temel özetler, halüsinasyonun yaygınlığını tahmin etmeye hizmet etmiş ve ince ayar için başlangıç noktaları sağlamıştır.
Varlık çıkarımı ve EHI hesaplaması
Entity Hallucination Index (EHI) doğru hesaplaması sağlam bir NER gerektirir. Başlangıçta, Spacy NER operasyonları için kullanıldı. Transformatör tabanlı NER modellerinin daha doğru olduğu anlaşılmaktadır, ancak Spacy şu nedenlerle ilk tercih edilmiştir: (1) Transformatör tabanlı NER modelleri Spacy'den halüsinasyon yapabildiğini kanıtlamıştır. (2) Boşluk istatistiksel olması, uygulama maliyeti ve uygulama süresi açısından hesaplama verimliliği sağlar. (3) Ayrıca, daha zayıf bir NER modeliyle bile EHI'nin halüsinasyonu azaltmada sağlam olduğunu kanıtlamamıza yardımcı olur. Ancak, deneylerin yerleşik veri setleri üzerinde yapıldığı ve spaCy'nin en_core_web_sm modelinin günlük transkriptlerde kırılganolduğu göz önüne alındığında 13. Spacy, CoNLL-2003 veri setinde ince ayar edilmiş bir BERT modeli olan transformatör tabanlı bir NER modeli (dslim/bert-base-NER) ile değiştirilmiştir. BERT tabanlı NER sistemleri, alana özgü görevlerde spaCy modellerinden üstün olduğu gösterilmiştir—örneğin, bir Aviation-BERT-NER modeli 17 varlığı tanımlayarak %94,78 F1'e ulaşmış, spaCy NER ise %93,73 ile yedi varlığı27 tanımıştır. BERT NER modeli, Hugging Face Transformers28 ile yapılandırıldı ve küçük harflere duyarsız eşleştirme gerçekleştirdi. Zamirleri ve yüzey biçimi varyasyonlarını yakalamak için, bu yöntem ayrıca "Mr. Smith" ve "Smith"i aynı kanonik biçime eşleyen basit kurallar da uyguladı; Ancak, tam ko-referans çözümü gelecekteki çalışmalar olarak kalacak. XSUM veri setinden alınan 200 kayıt örnekleminde NER modellerinin karşılaştırmalı analizi Tablo 1'den çıkarılabilir.
Varlık Halüsinasyon Endeksi (EHI) şu şekilde hesaplanır:

burada PH, EF, OF, NH, LF, madde i için sırasıyla Pozitif Halüsinasyon (PH), Çıkarma Faktörü (EF), Negatif Halüsinasyon (NH), Aşırı Odaklanmış Varlıklar (OF) ve Kayıp Odak (LF) puanlarını temsil eder.
Halüsinasyon faktörlerinin detayları:
Pozitif Halüsinasyon (PH): Yeni tanıtılan varlıkların gerçeklere doğru ve faydalı ölçümleri.
Çıkarma Faktörü (EF): Ölçüm varlıkları, giriş belgesinden özete doğru şekilde çıkarılır.
Negatif Halüsinasyon (NH): Yanlış veya girdiye bağlı olmayan halüsinasyon varlıklarını yakalar.
Aşırı Odaklanmış (OF): Çeşitliliği eksik bırakan, dar bir alt küme odaklanan özetleri cezalandırır.
Lost Focus (LF): Girdide bulunan önemli varlıkları atlayan özetleri cezalandırır.
Yukarıdaki faktörlerin hesaplanması, örnek hesaplama ile Şekil 4'te ayrıntılı olarak gösterilmiştir. Giriş Belgesi Varlıkları örneğini ele alarak (I):"John"," AI", konferans" Referans Özet Varlıkları (R):"John", AI" Oluşturulan Özet Varlıklar (G):"John", "AI", technology". EHI'nin daha yüksek değerleri, daha iyi varlık sadakatini gösterir; hem çıkarıcı hem de olumlu halüsinasyonlu özetleri ödüllendirir (faydalı ama sadık varlıkları tanıtır), ancak temelsiz, aşırı veya eksikvarlık 20'yi cezalandırır. Bu faktörler, tespit edilen varlıkların toplam sayısıyla normalleştirilerek 0 ile 1 arasında bir EHI puanı elde edildi. En iyi kontrol noktası seçilmek için eğitim sırasında EHI doğrulaması izlendi. PH ve EF faydalı yeni varlıkları ödüllendirir ve doğru çıkarma sağlarken, OF, NH ve LF tekrar, uydurma ve ihmal cezalarını alır. Mükemmel 1.0 puan, özetteki tüm varlıkların topraklanmış veya faydalı halüsinasyonlarda olduğu anlamına gelirken, 0 puanı özette isimli bir varlığın kaynakta görünmediğini gösterir.
RL ile ince ayar prosedürü
İnce ayar için ayrıntılı hiperparametre yapılandırması, Mistral-7B, DistilBart ve Flan-T5 için optimizator türleri, öğrenme oranları, parti boyutları, donanım özellikleri ve diğer yapılandırma detaylarını gösteren Tablo 2, Tablo 3 ve Tablo 4'te ayrı ayrı olarak sunulmaktadır. Buradaki amaç, üretilen özetler arasında beklenen Varlık Halüsinasyon İndeksi (EHI) ödülünü en üst düzeye çıkararak model parametreleri θ'yi ince ayarlamaktır. Resmi olarak, bir giriş belgesi Xi, oluşturulmuş bir özet Yi ve model parametreleri θ için beklenen ödül hedefi şu şekilde tanımlanır:
(2)
burada EHI (y, x), üretilen özet Yi ile girdi Xi arasında hesaplanan Varlık Halüsinasyon İndeksini gösterir.
REINFORCEalgoritması 25'e göre, bu hedefin gradyanı şu şekilde tahmin edilir:
(3)
Pratikte, modellerden özetler alındı, karşılık gelen EHI puanları hesaplandı ve politika gradyanı güncellemeleri uygulanarak daha yüksek ödüllü çıktıları teşvik edildi. Parametre verimli ince ayarlamayı mümkün kılmak için bu çalışma Düşük Seviye Uyumu (LoRA) kullandı. Sadece LoRA adaptörleri güncellenirken, temel model ağırlıkları dondurulmuş kaldı. İnce ayarlama, küçük öğrenme hızı (örneğin, 5 × 10-6), parti büyüklüğü 4 ve gradyan birikimiyle Adam optimizer29 ile 8 adımda yapılan bir GPU üzerinde yapıldı. Her 500 güncellemede bir özetler yenilenerek ödül tahminleri yenilendi ve eğitim, doğrulama EHI'ye ulaşana kadar devam etti. REINFORCE temel çizgisi, varyansı azaltmak için son ödüllerin ortalamasına ayarlandı. EHI, Entity F12'ye ek olarak, genel kaliteyi izlemek için ROUGE-1/2/L puanları kaydedildi. Özetler, ince ayar sırasında model iyileştirmelerini yansıtmak için periyodik olarak yeniden oluşturulur.
Değerlendirme metrikleri
Çıktılar, Rouge-1, Rouge-25 gibi Bilgilendiricilik metrikleri kullanılarak değerlendirildi. Rouge LCS5 gibi akıcılık metrikleri. Varlık metrikleri Varlık F1 ile EHI, FactCC ve QAGS 2,6,18,30 gibi halüsinasyon metrikleriyle örtüşüyor. FactCC, kaynağa bağlı veya çelişkili cümleleri etiketlemek için sentetik çelişki verileri üzerine eğitilmiş bir sınıflandırıcı kullanır. FactCC iki puan veriyor; notlar, etiketin GEÇERLİ olduğu yerde kaydedildi. Öte yandan, QAGS LLM üzerinde çalışan bir soru-cevap yöntemi kullanır. Metin 31 üzerinde soru oluşturmak için hafif model T5 small kullanıldı. Roberta_base_Squad2, çıktımetni 32 üzerinden cevaplar üretmek için kullanıldı. Bu modeller, uygulama için pratik hesaplama maliyeti nedeniyle seçildi.