$$\rightleftharpoonup{xx}$$
$$\longleftharp{xx}$$,
$$\longrightharp{xx}$$,
O objetivo deste estudo é afinar um modelo de linguagem (LM) para gerar resumos com alucinações de entidade reduzida. Alucinações ocorrem quando um modelo "produz com confiança resultados incorretos ou irrelevantes" porque gera texto baseado em probabilidade estatística, e não em verificação factual. Isso é alcançado projetando uma função de recompensa baseada no Índice de Alucinação da Entidade (EHI) e aplicando aprendizado por reforço para maximizá-la.
Formulação do problema
Dado um documento de entrada Xi, o objetivo é gerar um resumo Yi de modo que as entidades mencionadas em Yi estejam fundamentadas em Xi. O treinamento tradicional de máxima verosimilhança não otimiza explicitamente a consistência factual. Esses métodos de treinamento não penalizam a inclusão de entidades fabricadas. Portanto, é introduzida uma estratégia de ajuste fino orientada por recompensa, onde a recompensa é proporcional à fidelidade da entidade, medida usando EHI.
Conjunto de dados e método
São usados dois corpora: primeiro, um conjunto de dados de transcrição de diálogos com transcrições de reuniões em múltiplos turnos e resumos abstratos ouro, e segundo, o benchmark de resumo de notíciasXSUM 22,23. Cada conjunto de dados foi limpo, achatado e dividido em 80% de treinamento, 10% de validação e 10% de partições de teste. Os diálogos são ricos em linguagem informal e pronomes, enquanto o XSUM contém artigos de notícias concisos; A combinação nos permite avaliar tanto a generalização dentro quanto fora do domínio.
A extração de entidades é realizada tanto em documentos fonte quanto em resumos para calcular o Índice de Alucinações de Entidades (EHI). Durante o ajuste fino, este estudo utiliza exclusivamente a divisão de treinamento, enquanto as pontuações EHI de validação são monitoradas para selecionar o melhor ponto de verificação. As avaliações finais são reportadas no conjunto de teste que está em aberto, que permanece zero-shot antes e depois do ajuste fino dos modelos.
Em resumo, primeiramente, as transcrições são organizadas em um formato achatado e dividem os dados. O modelo pré-treinado gera resumos de linha base. Em segundo lugar, o EHI é calculado usando NER baseado em transformador e os cinco fatores de alucinação. Durante o ajuste fino, o modelo é atualizado com a recompensa EHI via adaptadores LoRa. Por fim, os resumos são gerados usando o modelo ajustado e avaliados usando EHI, Entidade F1 e outras métricas (Figura 2). Todos os experimentos são reproduzíveis através do código fornecido e dos arquivos de configuração. A Figura 3 mostra o fluxo de processo passo a passo para preparação de conjuntos de dados, sumarização de linha de base, cálculo de EHI, ajuste fino e avaliação.
Resumo da linha de base
Três LLMs pré-treinados são escolhidos, e suas sumarizações básicas são capturadas: Flan-T5, Mistral (Nous-Hermes-2-Mistral-7B-DPO) e DistilBART 24,25,26. Cada modelo era executado em modo zero-shot para produzir um resumo inicial (Yi) para cada documento de entrada (xi). A busca por feixe com largura de feixe de 4 e penalidade de comprimento de 1,0 foi usada para incentivar resumos fluentes, porém concisos. Esses resumos de base serviram para estimar a prevalência de alucinações e forneceram pontos de partida para ajustes finos.
Extração de entidades e computação EHI
Calcular com precisão o Índice de Alucinação da Entidade (EHI) requer um NER robusto. Inicialmente, o Spacy foi usado para operações da NER. Entende-se que modelos NER baseados em transformadores são mais precisos, mas Spacy foi a escolha inicial pelos seguintes motivos: (1) Modelos NER baseados em transformadores provaram alucinar mais do que Spacy. (2) O espaçamento sendo estatístico traz eficiência computacional em termos de custo de implementação e tempo de execução. (3) Também nos ajuda a provar que o EHI é robusto na redução de alucinações mesmo com um modelo NER mais fraco. No entanto, dado que os experimentos são feitos com conjuntos de dados estabelecidos e o modelo en_core_web_sm do spaCy é frágil nas transcriçõescoloquiais 13. O espaçamento é substituído por um modelo NER baseado em transformador (dslim/bert-base-NER), que é um modelo BERT ajustado com base no conjunto de dados CoNLL-2003. Sistemas NER baseados em BERT demonstraram superar modelos spaCy em tarefas específicas de domínio — por exemplo, um modelo Aviation-BERT-NER atingiu um F1 de 94,78% ao identificar 17 entidades, comparado a 93,73% para o spaCy NER, que reconheceusete entidades 27. O modelo NER do BERT foi configurado via Hugging Face Transformers28 e realizou correspondência insensível a maiúsculas minúsculas. Para capturar pronomes e variações na forma superficial, esse método aplicava ainda regras simples que mapeiam "Sr. Smith" e "Smith" para a mesma forma canônica; no entanto, a resolução total de copreferência ainda é trabalho futuro. A análise comparativa dos modelos NER em uma amostra de 200 registros do conjunto de dados XSUM pode ser inferida a partir da Tabela 1.
O Índice de Alucinação da Entidade (EHI) é calculado como:

onde PH, EF, OF, NH, LF representam pontuações correspondentes a Alucinação Positiva (HP), Fator de Extratividade (FE), Alucinação Negativa (NH), Entidades Excessivamente Focadas (OF) e Foco Perdido (LF), respectivamente, para o artigo i.
Detalhes dos fatores de alucinação:
Alucinação Positiva (HP): Medidas de entidades recém-introduzidas que são factualmente corretas e benéficas.
Fator de Extração (FE): Mede as entidades extraídas corretamente do documento de entrada no resumo.
Alucinação Negativa (NH): Captura entidades alucinadas que estão incorretas ou não fundamentadas na entrada.
Focado demais (OF): Penaliza resumos que focam excessivamente em um subconjunto restrito de entidades, faltando diversidade.
Foco Perdido (LF): Penaliza resumos que omitem entidades importantes presentes na entrada.
O cálculo dos fatores acima é detalhado na Figura 4 com um cálculo de exemplo para fins ilustrativos. Considerando um exemplo de Entidades de Documento de Entrada (I):"John"," "IA"," conferência" Entidades de Resumo de Referência (R):"John"," "IA" Entidades de Resumo Geradas (G):"John"," "IA"," tecnologia". Valores mais altos de EHI indicam melhor fidelidade à entidade, recompensando resumos que são tanto extrativos quanto positivamente alucinados (introduzindo entidades úteis, porém fiéis), enquanto penalizam entidades20 sem fundamento, excessivas ou ausentes. Esses fatores foram normalizados pelo número total de entidades detectadas para produzir uma pontuação EHI entre 0 e 1. A validação EHI foi monitorada durante o treinamento para selecionar o melhor ponto de verificação. PH e EF recompensam novas entidades benéficas e a extração correta, enquanto OF, NH e LF penalizam repetição, fabricação e omissão. Uma pontuação perfeita de 1,0 significa que todas as entidades no resumo estão fundamentadas ou alucinavas benéficas, enquanto uma pontuação 0 indica que nenhuma entidade nomeada no resumo aparece na fonte.
Procedimento de ajuste fino com RL
A configuração detalhada dos hiperparâmetros para ajuste fino é fornecida separadamente nas Tabelas 2, Tabela 3 e Tabela 4, que mostram tipos de otimizadores, taxas de aprendizado, tamanhos de lote, especificações de hardware e outros detalhes de configuração para Mistral-7B, DistilBart e Flan-T5, respectivamente. O objetivo aqui é ajustar os parâmetros do modelo θ maximizando a recompensa esperada do Índice de Alucinação de Entidade (EHI) em resumos gerados. Formalmente, para um documento de entrada Xi, um resumo gerado Yi e parâmetros do modelo θ , o objetivo esperado de recompensa é definido como:
(2)
onde EHI (y, x) denota o Índice de Alucinação da Entidade calculado entre o resumo gerado Yi e a entrada Xi.
Seguindo o algoritmoREINFORCE 25, o gradiente desse objetivo é estimado como:
(3)
Na prática, resumos foram amostrados dos modelos, seus correspondentes EHI foram calculados e atualizações do gradiente de políticas foram aplicadas para incentivar resultados de maior recompensa. Para possibilitar ajustes finos eficientes em parametros, este estudo utilizou Adaptação de Baixo Nível (LoRA). Apenas os adaptadores LoRA foram atualizados, enquanto os pesos do modelo base permaneceram congelados. O ajuste fino foi realizado em uma GPU A100 com baixa taxa de aprendizado (por exemplo, 5 × 10-6), tamanho de lote de 4 e acumulação de gradiente em 8 passos com o otimizadorAdam 29. Resumos eram regenerados a cada 500 atualizações para atualizar as estimativas de recompensa, e o treinamento continuava até que o EHI de validação convergisse. A linha base do REINFORCE foi definida para a média contínua das recompensas recentes para reduzir a variância. Além do EHI, Entidade F12, as pontuações ROUGE-1/2/L foram registradas para monitorar a qualidade geral. Os resumos são regenerados periodicamente para refletir melhorias do modelo durante o ajuste fino.
Métricas de avaliação
Os resultados foram avaliados usando métricas de Informatividade como Rouge-1, Rouge-25. Métrica de fluência como a RougeLCS 5. Métrica de sobreposição de entidades Entidade F1, e métricas de alucinação como EHI, FactCC e QAGS 2,6,18,30. O FactCC utiliza um classificador treinado com dados sintéticos de contradição para rotular frases como implicadas ou contraditas pela fonte. O factCC dá duas pontuações; as partituras foram capturadas onde o rótulo é VÁLIDO. Por outro lado, o QAGS utiliza um método de perguntas e respostas que roda em um LLM. O modelo leve T5 pequeno foi usado para gerar perguntas no texto31. Roberta_base_Squad2 era usado para gerar respostas sobre o textode saída 32. Esses modelos foram escolhidos por seu custo conveniente de computação para execução.