LoRA berbasis nol untuk menyempurnakan LLM setelah RL tanpa kehilangan kemampuan bernalar

5 Oktober 202611 tampilan

Para penulis mengusulkan metode adaptasi yang efisien secara parametrik untuk model yang telah dilatih dengan pembelajaran penguatan, dengan memproyeksikan pembaruan LoRA ke ruang nol perkiraan dari aktivasi penalaran. Pendekatan ini mempertahankan akurasi penalaran pada tingkat sebelum fine-tuning, sekaligus menyamai kualitas adaptasi LoRA biasa.

LoRA berbasis nol untuk menyempurnakan LLM setelah RL tanpa kehilangan kemampuan bernalar

Mengapa SFT setelah RL mengancam kemampuan bernalar

Reinforcement learning mengembangkan kemampuan penalaran model bahasa besar. Pendekatan ini menjadi efektif untuk pascapelatihan LLM. Supervised fine-tuning berikutnya menyesuaikan model pasca-RL dengan bidang pengetahuan atau jenis perilaku baru. SFT semacam ini dapat secara serius menimpa kemampuan yang telah diperoleh.

Pendekatan yang ada untuk mengurangi lupa:

  • pemutaran ulang pengalaman;
  • inisialisasi khusus;
  • optimisasi terbatas dengan proyeksi gradien.

Metode-metode tersebut hanya mempertahankan kemampuan secara terbatas atau memerlukan komputasi tambahan yang signifikan. Kriteria praktis: untuk model pasca-RL, pilih metode yang mengurangi risiko kemampuan bernalar tertimpa.

Ruang nol sebagai sumber daya untuk adaptasi

Para penulis menunjukkan bahwa aktivasi yang terkait dengan penalaran terkonsentrasi pada subruang berdimensi rendah. Kapasitas yang cukup besar masih tersedia di ruang nol untuk adaptasi. Ruang nol hampiran dapat diperkirakan secara andal dari sejumlah kecil contoh.

Kriteria praktis: metode yang membangun pembaruan di ruang nol memanfaatkan kapasitas yang tersedia. Metode ini tidak mengubah aktivasi penalaran.

Cara kerja NB-LoRA

Null-Basis Low-Rank Adaptation (NB-LoRA) adalah metode adaptasi hemat parameter untuk LLM pasca-RL. Metode ini diperkenalkan dalam makalah arXiv:2609.25618. Metode ini mempertahankan kemampuan penalaran yang telah diperoleh.

Tahapan kerja:

  1. Membangun basis nol hampiran tetap dari aktivasi penalaran.
  2. Memparametrisasi ulang pembaruan LoRA melalui basis ini.
  3. Merumuskan pelestarian kemampuan bernalar sebagai kendala untuk mempertahankan keadaan tersembunyi di tingkat lapisan.
  4. Mematuhi kendala tersebut selama proses fine-tuning.

Kriteria praktis: pilih NB-LoRA ketika perlu melakukan SFT sambil mempertahankan kemampuan bernalar model pasca-RL.

Perbandingan dengan pendekatan yang ada

Tabel ini membandingkan mekanisme dan karakteristik.

KriteriaPendekatan yang adaNB-LoRA
MekanismePemutaran ulang pengalaman, inisialisasi khusus, optimisasi terbatas dengan proyeksi gradienParametrisasi ulang LoRA melalui basis nol hampiran tetap
KarakteristikPelestarian kemampuan yang terbatas atau komputasi tambahan yang signifikanKendala untuk mempertahankan keadaan tersembunyi di tingkat lapisan

Kriteria praktis: jika prioritasnya adalah mempertahankan kemampuan bernalar, NB-LoRA menyediakan kendala bawaan. Pendekatan yang ada mengharuskan pemilihan antara pelestarian terbatas dan komputasi tambahan.

Hasil eksperimen dan kriteria pemilihan

Eksperimen mencakup beberapa LLM yang dilatih dengan RL dan beragam tugas lanjutan. NB-LoRA menunjukkan kualitas adaptasi setara dengan LoRA standar. Metode ini mempertahankan akurasi penalaran mendekati tingkat sebelum fine-tuning. Pelestarian ini juga berlaku pada benchmark penalaran yang ditahan.

Kriteria praktis: untuk LLM pasca-RL yang kemampuan bernalarnya terancam oleh SFT, NB-LoRA layak dipertimbangkan untuk diuji pada tugas Anda sendiri.

Pertanyaan yang sering ditanyakan

Bahan terkait

Semua bahan
LoRA berbasis nol untuk menyempurnakan LLM setelah RL tanpa kehilangan kemampuan bernalar