AI 요약 · 영어에서 AI 번역

PolyMemDB: A Polyglot Database System for AI Memory Management

Yu Wang, Jiaheng Lu

arXiv:2608.25577v1cs.DBcs.AI

TL;DR

기존 에이전트 메모리 시스템은 이종 데이터 관리와 시간에 따라 상충하는 사실 추적에 어려움을 겪는다. PolyMemDB는 이종 저장소와 시간 감쇠, 출처 정보, 세미링 추론을 활용한 확률적 그래프 관리를 결합한다. 사실 충돌 시연에서는 이분법적 답변 대신 상충하는 기억을 인정하고 출처 정보에 근거한 응답을 생성한다.

  • 문제

    기존 에이전트 메모리 시스템은 이종 데이터의 통합 저장과 다차원 연계 기능뿐 아니라, 시간에 따라 상충하는 사실을 관리할 출처 추적 체계도 부족하다.

    출처: A. Motivation and Contributions · p. 1
  • 방법

    PolyMemDB는 이종 데이터베이스와 시간 감쇠, 데이터 출처 정보, 세미링 추론을 통합한 확률적 메모리 그래프 갱신을 결합한다.

    출처: A. Motivation and Contributions · p. 1
  • 결과

    사실 충돌 시연에서 PolyMemDB는 에이전트가 이분법적 답변 대신 상충하는 기억을 인정하고 출처 정보에 근거한 응답을 생성하도록 유도했다.

    출처: §IV · p. 4
  • 시사점 및 한계

    사용자는 시연 인터페이스에서 답변의 근거 삼중항, 관측 사실, 시간 감쇠 확률, 원본 코퍼스를 확인할 수 있다.

    출처: §IV · p. 4

Abstract

from arXiv · show

With the widespread adoption of personal intelligent agents, users generate massive, heterogeneous data during long-term interactions. Leveraging this data as long-term memory helps reduce token overhead and deliver personalized experiences. However, existing memory systems face two primary limitations: they rely on single-storage paradigms that fragment multi-dimensional data, and they lack fine-grained data provenance to resolve long-term factual conflicts, thereby worsening LLM hallucinations. In this demonstration, we introduce PolyMemDB, a novel system tailored for managing agent memory. PolyMemDB has a polyglot storage architecture designed to track and manage various memory types, including graph, vector, probability and spatial-temporal data. To ensure factual consistency and reduce hallucinations, it features a probabilistic inference engine that integrates temporal decay with semiring aggregation, resolving long-term factual conflicts, providing detailed data provenance, and enabling users to trace reasoning chains transparently.

I. 서론 · A. 연구 동기 및 기여

PolyMemDB는 폴리글롯 저장 계층과 확률적 메모리 그래프 유지 관리를 결합해 이질적인 메모리 저장과 장기적인 사실 충돌 문제를 해결한다.

A. 연구 동기 및 기여

  • 영속적 에이전트 메모리는 추론 지연 시간과 토큰 비용을 줄이고, 사용자 선호를 포착하며, 고차 지식 추상화를 지원한다.의미적으로 유사한 응답을 재사용함으로써 이러한 지연 시간 및 토큰 비용 절감 효과를 얻는다.출처: A. Motivation and Contributions · p. 1
  • 단일 저장 패러다임은 개체 네트워크, 논리적 관계, 시공간 제약을 포함하는 이질적인 상호작용 이력을 저장·검색·연계하는 데 한계가 있다.이러한 한계는 장기 메모리의 통합 저장과 다차원 연계가 필요한 이유다.출처: A. Motivation and Contributions · p. 1
  • 거친 단위의 상태 덮어쓰기와 시간·확률 기반 사실 정보 유지 관리의 부재로 인해 기존 시스템은 출처 추적과 환각 대응을 강화할 필요가 있다.장기 메모리를 유지 관리하는 과정에서 사실이 변화하고 충돌하면서 이 문제가 발생한다.출처: A. Motivation and Contributions · p. 1
  • PolyMemDB는 멀티모달 객체 저장소, 그래프 데이터베이스, 시공간 데이터베이스, 확률 데이터베이스를 통합해 텍스트 전용 검색과 벡터 기반 검색의 한계를 해결한다.구현에서는 그래프 데이터베이스와 시공간 인덱싱도 결합한다.출처: A. Motivation and Contributions · p. 1
  • 확률 데이터베이스의 세미링 추론을 증거 사슬로 확장하면 충돌 해결과 개체 정렬이 가능해지며, LLM의 환각을 완화하면서 해석 가능성을 유지할 수 있다.확률적 메모리 그래프 유지 관리 방식은 장기 상호작용 중 발생하는 사건 통합과 사실 충돌 문제를 다룬다.출처: A. Motivation and Contributions · p. 1
  • 동적 그래프 갱신은 시간 감쇠와 데이터 계보를 결합해 여러 출처에서 유래한 사실 간의 충돌 해결을 지원한다.이 메커니즘은 PolyMemDB의 확률적 메모리 그래프 유지 관리의 일부다.출처: A. Motivation and Contributions · p. 1

B. 기존 시스템과의 비교

PolyMemDB는 이질적인 메모리를 위한 폴리글랏 저장소와 확률적 추론에 시간 감쇠를 결합한 데이터베이스 수준의 충돌 해결로 차별화된다.

B. 기존 시스템과의 비교

  • PolyMemDB는 MemForest 의 텍스트·시간 인덱싱 및 HyperMem 의 동종 하이퍼그래프와 달리, 단일 저장소 기반 메모리 구성을 이질적인 다차원 데이터를 위한 폴리글랏 저장소로 대체한다.MemForest 는 시간 관련 정보를 트리로 구성하고 전체 메모리를 다시 쓰지 않고도 특정 경로를 갱신한다. HyperMem 은 주제, 조각, 사실을 3계층 하이퍼그래프로 모델링하고, 하이퍼간선으로 메모리 조각 간 의존성을 포착한다.출처: B. Comparison to Existing Systems · p. 1
  • PolyMemDB는 Neo4j에서 개체 간 연결 구조를 유지하고 시공간 제약을 MobilityDB로 내려보내 동적 시간 구간과 지리적 다각형을 함께 필터링한다.이는 비교 대상인 기존 시스템이 지원하는 텍스트 관계만을 다루는 범위를 넘어 처리를 확장한다.출처: B. Comparison to Existing Systems · pp. 1–2
  • PolyMemDB는 데이터베이스 수준의 세미링 추론과 시간 감쇠로 메모리 충돌을 해결하고 세밀한 데이터 계보를 제공하는 반면, MRAgent 는 검색 시점의 능동적 재구성을 사용한다.PolyMemDB는 메모리 삼중항 간선에 과거 관측 시퀀스를 부착하고 양의 촉진, 음의 억제, 증거 충돌에 대한 분포 벡터를 계산한다. MRAgent 는 LLM이 중간 검색 상태에 따라 경로를 반복적으로 탐색하고 가지치기하는 “Cue-Tag-Content” 그래프를 사용한다.출처: B. Comparison to Existing Systems · p. 2

II. 시스템 개요

PolyMemDB의 3단계 워크플로는 입력을 원자적 사실로 변환하고, 특화된 저장소에 분배한 뒤, 단계적 검색으로 질의에 답한다.

메모리 수집 및 색인

  • 메모리 수집은 LLM과 시각 언어 모델로 입력을 원자적 사실로 분해하며 3단계 워크플로를 시작한다.처리 과정에는 개체 정렬, 공지시 해소, 생략 복원, 개체명 추출, 사실 정규화, 신뢰도 추정이 포함된다.출처: §II · p. 2

폴리글롯 저장 계층

  • 폴리글롯 저장 계층은 정규화된 사실을 다차원 특성에 따라 분배하여, 관계, 신뢰도, 시공간 제약, 임베딩, 원본 맥락을 특화된 저장소에 보존한다.Neo4j는 개체, 속성, 장기 변화 궤적을 연결된 그래프로 표현하여 복잡한 연쇄 추론을 지원한다. ProvSQL은 사건의 신뢰도를 저장하고, MobilityDB는 시공간 제약을 관리한다. ChromaDB는 임베딩 벡터를 저장하며, 객체 저장소는 원본 맥락을 보존한다.출처: §II · p. 2

추론 및 단계적 질의응답

  • PolyMemDB는 3단계 연합 검색을 사용하며, 앞선 단계에서 답이나 충분한 근거를 찾지 못하면 그래프 기반 추론에서 밀집 검색으로, 마지막으로 원본 멀티모달 조각 검색으로 진행한다.1단계는 저지연 부분 그래프 탐색에 확률·시공간 필터링과 논리적 추론을 결합한다. 후속 단계에서는 질의 임베딩으로 Top-k 구조화된 사실을 검색한 뒤 원본 조각을 검색하여 최종 응답을 생성하며, 추론 오버헤드와 검색 재현 정확도 사이의 균형을 맞춘다.출처: §II · p. 2

III. 확률적 메모리 그래프 유지 관리

PolyMemDB는 시간 감쇠와 확률적 추론을 결합해 에이전트의 메모리 그래프를 동적으로 갱신하고 상충하는 증거를 조정한다. 과거 관측을 보존하고 신뢰도 점수를 구성 요소별로 제공해 데이터 계보에 근거한 답변을 지원한다.

메모리 그래프 정식화와 동적 갱신

  • PolyMemDB는 메모리를 이종 방향 그래프로 표현하고, 에이전트가 생성하거나 갱신할 노드와 관계를 결정하기 전에 의미적으로 유사한 Top-k 국소 부분 그래프를 검색한다.그래프는 이종 개체, 레이블이 부여된 방향성 관계, 속성 레이블링 함수로 구성된다. 각 사실에는 증거의 극성, LLM이 초기에 부여한 0과 1 사이의 신뢰도, 관측 타임스탬프가 기록된다.출처: §III · p. 3

데이터 계보와 증거 사슬 구성

  • PolyMemDB는 상태를 덮어쓰는 대신 각 관계의 관측 이력을 타임스탬프 순으로 간선 속성에 보존해 데이터 계보와 해석 가능성을 위한 증거를 유지한다.각 그래프 트리플은 과거 사실을 순차적으로 덧붙인 이력을 유지해 서로 다른 시점의 관측을 보존한다.출처: §III · p. 3

시간 감쇠 기반 확률적 추론

  • PolyMemDB는 시간 감쇠와 확률적 추론을 결합하고, ProvSQL 에서 사용하는 semiring 프레임워크를 확장해 긍정 증거와 부정 증거를 각각 집계한다.각 관측에는 가중치 w_i = c_i · λ^(T − i)가 부여되며, λ의 범위는 (0, 1]이다. 포함배제 원리로 긍정 증거와 부정 증거를 각각 누적 결합 신뢰도로 집계한다.출처: §III · pp. 2–3
  • PolyMemDB는 상충하는 증거 간 상호작용을 포착하기 위해 P_+와 P_−로부터 긍정적 추동, 부정적 억제, 증거 충돌, 증거 부재라는 네 가지 중간 인지 상태를 계산한다.긍정적 추동은 P_+(1 − P_−)이고, 부정적 억제는 P_−(1 − P_+)이다. 증거 충돌은 P_+P_−이며, 증거 부재는 (1 − P_+)(1 − P_−)이다.출처: §III · p. 3
  • PolyMemDB는 순 증거 신뢰도 R을 계산해 지지와 반박을 구분하고 상충하거나 불충분한 증거에 불이익을 부여함으로써 데이터 계보에 근거한 답변을 유도한다.R의 범위는 −1부터 1까지다. R ≈ 0일 때는 값이 가장 큰 중간 상태를 통해 증거의 모순과 부재를 구분한다. QA에는 R과 구성 요소별 상태 분포가 모두 전달된다.출처: §III · p. 3

IV. 시연

PolyMemDB의 대화형 대시보드는 데이터 계보를 추적할 수 있는 장기 세션 질의응답, 시공간 기억 탐색, 상충하는 기억의 확률적 처리를 시연한다. 백엔드는 FastAPI와 Pydantic-AI를 사용한다.

시나리오 1: 세밀한 데이터 계보를 제공하는 장기 세션 질의응답

  • 장기 세션 질의응답 시연을 위해 LongMemEval 의 48개 세션으로 엔터티 229개와 관계 221개로 구성된 기억 그래프를 구축한다.도서 할인율 질의에 20%를 검색 결과로 제시하며, 추론 경로, 근거 삼중항, 간선 관측값, 시간 감쇠로 계산한 긍정적 기여 확률을 확인할 수 있다. 인터페이스는 매칭된 엔터티의 부분 그래프와 원본 말뭉치도 제공한다.출처: §IV · pp. 3–4

시나리오 2: 시공간 기억의 변화

  • 동적 시간 창으로 졸업 여행 계획 기간을 필터링하면 이탈리아 도시 5곳만 강조되며, 구조화된 여행 일정 추출 결과와 근거에 연결된 사실도 함께 표시된다.LongMemEval 의 확장 사례를 사용해 전 세계 지도에 언급된 모든 지리적 위치와 관련 사실을 표시한다. 시간 창 필터링은 무관한 위치를 제외하고, 근거 출처를 추적할 수 있는 세밀한 사실 목록과 지도를 동기화한다.출처: §IV · p. 4, Figure 3 · p. 4

시나리오 3: 사실 충돌에 대한 동적 확률 추론

  • Sconflict의 52.3%가 Saction의 47.2%를 넘어, 앨리스의 10개월간 달리기 이력 전반에 대한 순 신뢰도가 R ≈0.22로 낮아진다.시간 감쇠(λ = 0.8)와 semiring 집계로 4차원 인지 분포를 생성한다. 근거 기반 응답은 단순히 “그렇다”라고 답하거나 상충하는 기억을 덮어쓰는 대신, 실천 의지와 함께 신체적·동기적 장애 요인도 인정한다.출처: §IV · p. 4, Figure 4 · p. 4
Loading 2608.25577v1…