Correzione auto-premiante per il ragionamento matematicoSelf-rewarding correction for mathematical reasoning
Studiamo i grandi modelli linguistici (LLM) con ragionamento auto-premiante, che possono generare simultaneamente ragionamenti passo-passo e valutare la correttezza dei loro output durante il tempo di inferenza, senza feedback esterno. Questo approccio integrato consente a un singolo modello di guidare in modo indipendente il proprio processo di ragionamento, offrendo vantaggi computazionali per il deployment del modello. Ci concentriamo in particolare sul compito rappresentativo dell'auto-correzione, in cui i modelli rilevano autonomamente errori nelle loro risposte, revisionano gli output e decidono quando terminare i cicli iterativi di affinamento. Per abilitare ciò, proponiamo un framework algoritmico a due fasi per costruire modelli di ragionamento auto-premianti utilizzando esclusivamente dati auto-generati. Nella prima fase, utilizziamo il campionamento sequenziale per rifiuto per sintetizzare lunghe traiettorie di catena di pensiero che incorporano meccanismi di auto-premiazione e auto-correzione. Il fine-tuning dei modelli su questi dati curati consente loro di apprendere i pattern di auto-premiazione e auto-correzione. Nella seconda fase, miglioriamo ulteriormente la capacità dei modelli di valutare l'accuratezza delle risposte e affinare gli output attraverso l'apprendimento per rinforzo con segnali basati su regole. Esperimenti condotti con Llama-3 e Qwen-2.5 dimostrano che il nostro approccio supera le capacità intrinseche di auto-correzione e raggiunge prestazioni comparabili a sistemi che si basano su modelli di ricompensa esterni.