TRACE: LLM 에이전트를 위한 오픈 소스 계층적 메모리, gpt-oss-20B를 사용하여 MemoryAgentBench의
요약
LLM 에이전트의 대화 기록을 토픽 트리 형태로 구성하는 오픈 소스 계층적 메모리 시스템 TRACE를 소개합니다. MemoryAgentBench의 EventQA 태스크에서 기존 Mem0나 MemGPT보다 높은 검색 정확도를 기록했습니다.
핵심 포인트
- 대화 기록을 평면적 RAG 대신 주제 트리 구조로 관리
- MemoryAgentBench의 EventQA 태스크에서 높은 F1 점수 달성
- gpt-oss 모델을 활용한 오픈 웨이트 기반의 메모리 시스템
- PyPI를 통해 pip install trace-memory로 간편하게 설치 가능
에이전트의 대화 기록을 평면적인 RAG 청크 (RAG chunks) 대신 토픽 트리 (주제 트리: 가지 + 요약) 형태로 구성하는 TRACE라는 메모리 시스템을 구축하였으며, 이를 MemoryAgentBench (ICLR 2026)의 특히 EventQA 정확한 검색 (accurate-retrieval) 태스크에서 벤치마킹했습니다.
이는 PyPI 패키지입니다:
pip install trace-memory
결과 (F1):
• TRACE (gpt-oss-20B): 82.5%
• TRACE (gpt-oss-120B): 83.8%
• Mem0 (GPT-4o-mini, 논문의 공식 수치): 37.5%
• MemGPT/Letta (GPT-4o-mini, 논문의 공식 수치): 26.2%
gpt-oss를 로컬에서 실행했으므로, 이는 GPT-4o-mini를 사용하는 MemGPT/Mem0와 비교했을 때 오픈 웨이트 (open-weights) 모델을 대상으로 한 테스트이며, 동일한 백본 (backbone)을 사용한 직접적인 비교 테스트(apples-to-apples)는 아닙니다 (OpenAI 토큰을 구매할 비용이 없습니다).
공정성을 위해 Mem0를 gpt-oss-20B에서 직접 실행하려고 시도했으나, Mem0의 사실 추출 (fact-extraction) 단계는 엄격한 JSON 출력을 필요로 하며 gpt-oss의 응답이 깔끔하게 파싱되지 않았습니다 (이는 gpt-oss에 국한된 것이 아닌 알려진 문제이며, Gemini/Mistral에서도 동일한 버그가 나타납니다). Letta는 전체 서버 설정이 필요하여 생략했습니다.
방법론을 직접 자세히 살펴보고 싶다면 두 실행 모두에 대한 전체 JSON 로그가 리포지토리(repo)에 있습니다. GitHub: https://github.com/husain34/TRACE
/u/PsychologicalDot7749 님이 r/MachineLearning 에 게시함
[link] [comments]
AI 자동 생성 콘텐츠
본 콘텐츠는 r/OpenAI Codex (search)의 원문을 AI가 자동으로 요약·번역·분석한 것입니다. 원 저작권은 원저작자에게 있으며, 정확한 내용은 반드시 원문을 확인해 주세요.
원문 바로가기