Hierarchical Reranking for Scalable Financial RAG System
Este artigo apresenta o Hierarchical Reranker, um framework de RAG escalável que integra otimização de pré-recuperação, um mecanismo de classificação em dois estágios e gerenciamento de contexto longo para aumentar significativamente a precisão da recuperação e a consistência factual para análise de documentos financeiros complexos, conforme evidenciado por seu desempenho superior no ACM-ICAIF '24 FinanceRAG Challenge.
Artigo original sob licença CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). Esta é uma explicação gerada por IA do artigo abaixo. Não foi escrita nem endossada pelos autores. Para precisão técnica, consulte o artigo original. Ler aviso legal completo
Imagine que você é um detetive tentando resolver um mistério, mas em vez de uma única cena de crime, lhe entregam uma biblioteca contendo milhões de livros, cada um repleto de milhares de páginas de texto minúsculo e confuso e gráficos complexos. Esta é a realidade diária para analistas financeiros que precisam ler documentos como os relatórios 10-K (grandes relatórios que as empresas enviam ao governo) para encontrar números ou fatos específicos. No mundo da Inteligência Artificial, existe uma ferramenta chamada RAG (Geração Aumentada por Recuperação) que atua como um bibliotecário superinteligente. Ele não apenas lê os livros; ele os pesquisa para encontrar as páginas exatas de que você precisa e então escreve um resumo baseado apenas no que encontrou. No entanto, os bibliotecários padrão costem ficar sobrecarregados pelo tamanho imenso dessas bibliotecas financeiras, se confundem com a mistura de palavras e números, ou inventam fatos quando os documentos são longos demais. Este artigo aborda o problema de como construir um bibliotecário que seja rápido, preciso e que não se perca nos detalhes quando o trabalho se torna enorme.
Os autores deste artigo, Joohyun Lee e Sungwoo Hong, propõem um novo sistema chamado Hierarchical Reranker (Reclassificador Hierárquico). Pense neste sistema como um processo de contratação de duas etapas para uma equipe de detetives. Em vez de contratar um único detetive gigante, caro e lento para ler cada página de um livro de 100.000 páginas, este sistema utiliza uma estratégia de "equipe pequena e grande". Primeiro, um detetive rápido e leve (um modelo de IA pequeno) faz uma varredura rápida em toda a biblioteca e descarta as páginas que são obviamente irrelevantes, reduzindo a pilha para as 100 páginas mais promissoras. Depois, um detetive altamente qualificado e especialista (um modelo de IA grande e poderoso) faz uma análise profunda e cuidadosa de apenas essas 100 páginas para escolher as 20 finais que realmente respondem à pergunta. Essa abordagem economiza tempo e dinheiro porque o especialista caro realiza o trabalho difícil apenas nos candidatos mais prováveis, em vez de desperdiçar energia com toda a biblioteca.
Mas encontrar as páginas certas é apenas metade da batalha. O artigo também apresenta uma maneira inteligente de lidar com os documentos antes mesmo de os detetives começarem a procurar. Documentos financeiros são bagunçados; eles usam abreviações como "YoY" (Ano sobre Ano) ou "EPS" (Lucro por Ação), e misturam parágrafos de texto com tabelas gigantes de números. O primeiro passo do sistema é "limpar" a consulta e os documentos. Ele transforma essas abreviações confusas em palavras completas, padroniza unidades (para que "1M" e "1.000.000" pareçam iguais) e, crucialmente, converte essas tabelas bagunçadas em um formato estruturado chamado JSON. Isso é como pegar uma planilha escrita à mão e transformá-la em um banco de dados digital para que a IA não se confunda sobre qual número pertence a qual categoria. Ao fazer essa "pré-limpeza", o sistema garante que a IA entenda a pergunta e as evidências perfeitamente antes mesmo de começar a busca.
O terceiro grande truque que os autores utilizam é uma estratégia para lidar com documentos que são simplesmente longos demais para caber na memória da IA de uma só vez. Embora os modelos de IA modernos afirmem que podem ler quantidades enormes de texto, o artigo descobriu que, quando o texto ultrapassa um certo ponto (especificamente 64.000 tokens, que é uma medida de comprimento de texto), a IA começa a cometer erros, especialmente com números. Para corrigir isso, o sistema age como um chef preparando um banquete massivo. Em vez de tentar comer a refeição inteira de uma vez, ele fatia o documento longo em dois pedaços menores e gerenciáveis. Ele pede à IA para analisar o primeiro pedaço e escrever uma resposta preliminar, depois analisa o segundo pedaço e escreve outra resposta preliminar. Finalmente, uma etapa de "fusão" combina esses dois rascunhos em uma única resposta final, verificando se eles concordam e resolvendo quaisquer conflitos. Isso garante que, mesmo que o documento seja um relatório de 100.000 páginas, a IA não perca o rastro dos fatos.
Quando os pesquisadores testaram seu sistema em vários benchmarks financeiros famosos (como FinQA e FinanceBench), os resultados foram impressionantes. O sistema alcançou uma pontuação de 0,7918 em uma métrica chamada NDCG@20, que mede o quão bem o sistema classifica as informações corretas no topo de sua lista. Esta pontuação foi significativamente melhor do que sistemas que não utilizaram seus métodos especiais de limpeza e classificação em duas etapas. Na verdade, essa abordagem foi tão eficaz que ajudou a equipe a conquistar o segundo lugar no desafio ACM-ICAIF '24 FinanceRAG, uma competição projetada para testar o quão bem a IA pode lidar com dados financeiros. O artigo sugere que, ao combinar essas três inovações — limpar os dados, usar uma equipe de detetives em duas etapas e fatiar inteligentemente documentos longos — as instituições financeiras podem agora construir sistemas de IA que não são apenas inteligentes, mas também confiáveis o suficiente para serem usados em tarefas do mundo real, como auditoria e análise de investimentos, sem que a IA invente fatos ou se perca no ruído.
Afogado em artigos na sua área?
Receba digests diários dos artigos mais recentes que correspondam às suas palavras-chave de pesquisa — com resumos técnicos, no seu idioma.