Sign In

MA-RAG: Multi-Agent Retrieval-Augmented Generation via Collaborative Chain-of-Thought Reasoning

Created by
  • Haebom
Category
Empty

저자

Thang Nguyen, Peter Chin, Yu-Wing Tai

MA-RAG: Multi-Agent Retrieval-Augmented Generation

개요

MA-RAG는 복잡한 정보 검색 작업의 모호성과 추론 문제를 해결하기 위해 설계된 검색 증강 생성(RAG)을 위한 멀티 에이전트 프레임워크입니다. 이 프레임워크는 Planner, Step Definer, Extractor, QA Agents와 같이 RAG 파이프라인의 각 단계를 담당하는 전문 AI 에이전트의 협업을 통해 작동합니다. 작업은 쿼리 명확화, 증거 추출, 답변 종합과 같은 하위 작업으로 세분화되며, 에이전트는 사고 과정 프롬프트를 통해 중간 추론을 통신합니다. 실험 결과, MA-RAG는 단독 LLM과 기존 RAG 방식보다 우수한 성능을 보이며, 특히 LLaMA3-8B 모델은 더 큰 단독 LLM보다 뛰어난 결과를 보였습니다. 또한, 의료 QA와 같은 전문 분야에서도 경쟁력 있는 성능을 보였습니다.

시사점, 한계점

시사점:
멀티 에이전트 방식은 복잡한 정보 검색 작업에서 정확성과 견고성을 향상시킵니다.
해석 가능한 중간 추론 단계를 제공하여 RAG 시스템의 효율성과 신뢰성을 높입니다.
특정 도메인에 대한 추가적인 미세 조정 없이 전문 분야에서도 적용 가능합니다.
소규모 모델에서도 우수한 성능을 보여 효율성을 입증했습니다.
한계점:
플래너와 추출 에이전트의 중요성을 고려할 때, 이러한 에이전트의 설계 및 최적화가 성능에 중요한 영향을 미칠 수 있습니다.
QA 에이전트의 효과적인 답변 종합을 위해 고용량 모델이 필요할 수 있습니다.
본 논문에서 구체적으로 다루지 않은 잠재적인 복잡성 및 계산 비용 문제 발생 가능성이 있습니다.
👍