- LLM은 복잡한 과제에서 정확도가 흔들리는데, 이 연구는 추가 구조보다 샘플링-투표만으로 성능을 끌어올릴 수 있는지 검증함
- Agent Forest는 같은 질의를 여러 번 실행해 답을 모은 뒤 다수결로 최종 답을 고르는 단순한 앙상블 방식임
- GSM8K에서는 앙상블 크기 15에서 Llama2-13B가 Llama2-70B와 비슷한 정확도에 도달했고, Llama2-70B와 GPT-3.5-Turbo도 더 강한 대응 모델에 가까워짐
- 이 방식은 CoT 기반 방법이나 다중 에이전트 협업 프레임워크와 독립적으로 결합될 수 있어 기존 기법 위에 추가 성능 향상을 만들 수 있음
- 개선 폭은 특히 어려운 과제와 약한 모델에서 크게 나타나며, 복잡한 프롬프트 설계 없이도 비용 대비 성능을 높일 여지가 있음
복잡한 과제에서 흔들리는 LLM 정확도
- LLM은 언어 생성, 이해, 추론 등 다양한 응용에서 강한 능력을 보이지만 복잡한 과제에서는 정확한 답을 내기 어려움
- 기존 성능 개선 연구는 크게 앙상블 방법과 다중 LLM 에이전트 협업 프레임워크를 활용해 왔음
- LLM-Debate는 여러 LLM 에이전트가 산술 과제의 최종 답을 토론하도록 구성해 단일 에이전트보다 추론 성능을 높임
- CoT-SC는 여러 사고 체인(thought chain)을 생성하고 가장 자기일관적인 답을 고르는 방식으로, 단일 사고 체인을 쓰는 CoT보다 추론 성능을 개선함
- 선행 결과에서도 에이전트 수나 사고 체인 수가 늘 때 성능이 오르는 현상이 관찰됐지만, 기본 LLM 에이전트 수 자체의 스케일링 특성은 별도 연구 대상으로 충분히 다뤄지지 않았음
Agent Forest의 동작 방식
- Agent Forest는 LLM 에이전트 수 증가가 성능에 미치는 영향을 보기 위해 단순한 샘플링-투표 절차를 사용함
- 동작은 두 단계로 나뉨
- 과제 질의를 단일 LLM 또는 다중 LLM 에이전트 협업 프레임워크에 반복 입력해 여러 출력을 생성함
- 생성된 출력에 다수결 투표를 적용해 최종 결과를 결정함
- 절차는 CoT-SC에서 영감을 받았지만, 복잡한 CoT 경로 설계에 의존하지 않음
- 이름은 고전적인 Random Forest에 대한 오마주임
GSM8K와 여러 과제에서의 결과
- 실험은 서로 다른 크기의 다양한 LLM과 추론·생성 과제를 포함하는 여러 데이터셋에서 수행됨
- 전반적으로 앙상블 크기, 즉 에이전트 수가 커질수록 LLM 성능이 향상될 수 있음
- Figure 1의 GSM8K 결과에서는 Llama2-13B, Llama2-70B, GPT-3.5-Turbo 모두 앙상블 크기가 커질수록 정확도가 증가함
- 앙상블 크기 15에서 Llama2-13B는 Llama2-70B와 비교 가능한 정확도에 도달함
- 앙상블 크기 15와 20에서 Llama2-70B와 GPT-3.5-Turbo는 각각 더 강한 대응 모델과 비교 가능한 정확도를 보임
- 그림의 오차 막대는 표준오차를 나타냄
- 작은 LLM도 단순 앙상블을 적용하면 더 큰 LLM과 비교 가능하거나 더 나은 성능을 낼 수 있음
기존 기법 위에 얹을 수 있는 성능 개선
- Agent Forest는 기존의 복잡한 LLM 성능 개선 방법과 독립적으로 결합 가능한 접근임
- CoT 기반 방법에는 플러그인처럼 붙여 추가 성능 향상을 만들 수 있음
- 복잡한 방법과 비교해도 Agent Forest만으로 대부분의 경우 비교 가능한 성능을 달성할 수 있음
- 추가 수작업 프롬프트 설계나 복잡한 협업 프레임워크 없이도 경쟁력 있는 결과를 얻을 수 있음
난이도에 따른 효과와 최적화
- 성능 향상은 어려운 과제와 약한 모델에서 더 크게 나타남
- 문제 난이도가 Agent Forest 효과에 미치는 영향은 세 가지 차원으로 나눠 분석됨
- 문제의 내재적 난이도
- 추론 단계의 길이
- 정답의 사전 확률
- 각 차원을 조정한 실험을 통해 Agent Forest 효과에 영향을 주는 속성을 확인함
- 확인된 속성을 바탕으로 “More Agents”의 효과가 더 잘 나타나도록 하는 최적화 전략을 추가로 개발함
- 공개 코드는
https://github.com/MoreAgentsIsAllYouNeed/AgentForest에서 제공됨