표가 포함된 PDF에서 RAG가 정답을 놓치지 않도록
Docling으로 파싱한 PDF의 표·캡션·설명 단락을 하나의 검색 단위(Evidence Unit)로 묶어 RAG 정답률을 높이는 파이썬 라이브러리
RAG 파이프라인에서 표가 포함된 PDF는 정답률이 유독 낮다. 원인은 파싱이 아니라 청킹이다.
표+설명 문단이 결합돼야만 풀리는 질문 536개 중, Docling HybridChunker 단독으로 정답을 맞춘 건 단 5개(EM 0.9%)였다.
Docling은 표와 캡션을 정확히 인식하고 연결까지 하지만(table.captions), 청킹 단계에서 이 연결이 다시 끊어진다.
flowchart LR
subgraph HC["Docling HybridChunker"]
direction TB
C1["캡션<br/>'Table 3: 지역별 매출'"]
C2["표 데이터<br/>지역 · Q1 · Q2 ..."]
C3["각주<br/>'(단위: 백만 달러)'"]
C4["설명 단락<br/>'서울은 신규 지점 2곳 개점 영향으로...'"]
end
EU["Evidence Chunker<br/>캡션 + 표 데이터 + 각주 + 설명<br/><b>하나의 검색 단위로 통합</b>"]
C1 -.-> EU
C2 -.-> EU
C3 -.-> EU
C4 -.-> EU
style HC fill:transparent,stroke:transparent,stroke-width:1.5px,color:#2E3440
style C1 fill:#ECEFF4,stroke:#D08770,stroke-width:1px,color:#2E3440
style C2 fill:#ECEFF4,stroke:#D08770,stroke-width:1px,color:#2E3440
style C3 fill:#ECEFF4,stroke:#D08770,stroke-width:1px,color:#2E3440
style C4 fill:#ECEFF4,stroke:#D08770,stroke-width:1px,color:#2E3440
style EU fill:#88C0D0,stroke:#5E81AC,stroke-width:2.5px,color:#2E3440
Docling HybridChunker는, 표 숫자와 설명 단락이 함께 있어야 풀리는 질문이 들어오면, 캡션·표·설명이 서로 다른 청크에 흩어져 있어 하나의 청크만으로는 답을 구성할 수 없다는 구조적 문제가 있다.
이에, Evidence Chunker는 같은 표의 캡션+숫자+각주+설명을 한 청크(EU)로 묶어 반환한다. Docling의 강력한 기능(파싱, 레이아웃 분석, 캡션-표 연결)은 사용하되, Docling만으로는 해결되지 않는 문제(청킹 단계에서 캡션·표·문맥이 다시 갈라지는 문제)를 해결한다.
Evidence Chunker는 같은 536문항에서 baseline 대비 정답을 188개까지 끌어올린다(EM 35.1%, +34.2pp). 자세한 유형별·대조군 수치는 Benchmark 참고.
"Evidence Unit"이라는 용어와 "파싱 요소를 개별 청크가 아닌 의미적으로 완결된 단위로 묶는다"는 상위 아이디어는 Han (2026), Evidence Units: Ontology-Grounded Document Organization for Parser-Independent Retrieval에서 가져왔다. 논문의 파이프라인 전체를 구현한 것이 아니라, 그 상위 개념을 Docling 한 파서에 한정하고 간단하게 재구성하였다.
Docling의 DoclingDocument를 입력받아, 이미 연결된 captions 참조와 prov[0].bbox를 활용해 표 하나당 Evidence Unit 하나를 구성한다.
- 캡션↔표 연결:
captions참조 우선, 실패하면 bbox 거리 → 인접 페이지 → 병합 헤더 순으로 fallback - 인접 설명 단락 부착: bbox 거리 기준으로 표 위/아래 단락을 EU에 포함
- 큰 표 분할: 토큰 한도(기본 512)를 넘는 표는 헤더+캡션을 반복 삽입하며 행 단위로 분할
| 기능 | 설명 |
|---|---|
| 캡션↔표 자동 연결 | 4단계 fallback(direct → bbox → 인접 페이지 → 병합 헤더)으로 캡션 없는 표까지 최대한 복구 |
| 인접 문맥 자동 부착 | bbox 거리 기반으로 표 위/아래 설명 단락을 탐지해 EU에 포함 |
| 큰 표 자동 분할 | 512토큰 초과 시 헤더+캡션을 반복 삽입하며 행 단위 분할, 모든 조각에 문맥 정보 동일 전파 |
| LangChain / LlamaIndex 래퍼 | to_langchain(), to_langchain_units()(small-to-big), EvidenceRetriever(max-pool dedupe 내장) |
| 가벼운 기본 의존성 | 기본 설정(sim_threshold=0.0)에서는 sentence-transformers 없이도 동작 |
flowchart LR
PDF([PDF]) --> Docling[Docling]
Docling --> Parser[Parser]
Parser --> Chunker["EvidenceChunker"]
Chunker --> Split[Split]
Split --> Export[Export]
Export --> RAG([RAG])
%% GitHub 호환용 Nord Deep 스타일 지정
style PDF fill:#EBCB8B,stroke:#D08770,stroke-width:1.5px,color:#2E3440
style Docling fill:#E5E9F0,stroke:#8FBCBB,stroke-width:1.5px,color:#3B4252
style Parser fill:#D8DEE9,stroke:#4C566A,stroke-width:1.5px,color:#2E3440
style Chunker fill:#88C0D0,stroke:#5E81AC,stroke-width:2.5px,color:#2E3440
style Split fill:#A3BE8C,stroke:#4C566A,stroke-width:1.5px,color:#2E3440
style Export fill:#B48EAD,stroke:#4C566A,stroke-width:1.5px,color:#2E3440
style RAG fill:#EBCB8B,stroke:#D08770,stroke-width:1.5px,color:#2E3440
모듈별 역할과 데이터 흐름은 Architecture 문서에서 더 자세히 볼 수 있다.
git clone https://github.com/EvidenceChunker/Evidence-Chunker cd Evidence-Chunker pip install -e ".[langchain]"
| extra | 설치 명령 | 필요할 때 |
|---|---|---|
langchain |
pip install -e ".[langchain]" |
LangChain 연동 |
llamaindex |
pip install -e ".[llamaindex]" |
LlamaIndex 연동 |
similarity |
pip install -e ".[similarity]" |
sim_threshold > 0으로 코사인 필터 사용 시 |
Evidence Unit 생성
from evidence_chunker import EvidenceChunker chunker = EvidenceChunker() eus = chunker.chunk("paper.pdf") # List[EvidenceUnit] (표만) for eu in eus: print(eu.caption_text, "->", len(eu.text), "chars")
LangChain 벡터스토어에 연결
from evidence_chunker.export.langchain import EvidenceRetriever from langchain_core.documents import Document from langchain_core.vectorstores import InMemoryVectorStore from langchain_huggingface import HuggingFaceEmbeddings # retrieval_text(캡션+문맥+표 문장, table_html 제외)로 임베딩 docs = [Document(page_content=eu.retrieval_text, metadata=eu.metadata) for eu in eus] vectorstore = InMemoryVectorStore.from_documents(docs, HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")) retriever = EvidenceRetriever(vectorstore, k=5) # max-pool dedupe 기본 적용
export.langchain.to_langchain()을 쓰면 편의상page_content=eu.text(HTML 포함)가 기본값으로 들어간다. 검색 정확도를 최대화하려면 위 예제처럼eu.retrieval_text를 직접 쓰는 걸 권장. 자세한 이유는 API Reference 참고.
표+본문을 합친 전체 코퍼스가 필요한 경우
chunks = chunker.build_corpus("paper.pdf") # EU(표) + TextChunk(일반 본문)
더 많은 예제는 API Reference 참고.
90개 PDF, 2725문항 기준 (baseline: Docling HybridChunker 단독):
| 지표 | baseline | Evidence Chunker | 갭 |
|---|---|---|---|
| Recall | 0.574 | 0.625 | +5.1pp |
| EM | 0.314 | 0.569 | +25.5pp (95% CI ±1.88pp) |
context_dependent(표+설명 문단이 결합돼야만 풀리는 질문 유형, 이 프로젝트가 해결하려는 핵심 케이스)는 baseline이 사실상 전혀 풀지 못하지만(EM 0.009) Evidence Chunker는 0.351까지 향상(+34.1pp).
유형별 성능 및 대조군(청크 크기 확대·행분할·semantic chunker) 비교 등은 Benchmark, 파라미터 스윕(bbox/sim_threshold) 등 전체 실험 과정은 Experiments 참고.
| 파라미터 | 기본값 | 설명 |
|---|---|---|
bbox_threshold |
300.0 (pt) |
표 위/아래 설명 단락을 수집할 거리 범위. 100~1000pt 스윕으로 검증 |
sim_threshold |
0.0 |
문맥 단락 채택 코사인 유사도 임계값. 0.0이면 임베딩 모델을 아예 로드하지 않음(의존성 절약) |
chunker = EvidenceChunker(bbox_threshold=300.0, sim_threshold=0.0) # 기본값
자세한 근거와 스윕 실험 결과는 Configuration 참고.
- Recall@10이 max-pool dedupe에도 불구하고 baseline 대비 근소하게 낮음
- 분할 조각 간 dedup 미적용이 원인
- 문서 내 구조적으로 유사한 표가 여러 개 있을 때 검색 단계에서 혼동 발생
bbox_threshold파라미터를 적응형 임계값으로 확장 검토
전체 목록과 각 항목의 원인 및 실측 근거, 진행 상황은 GitHub Issues에서 공개 추적한다.
Apache License 2.0.
더 자세한 문서는 Wiki Home에서 볼 수 있다.
| 문서 | 이런 게 궁금할 때 |
|---|---|
| Architecture | 파이프라인 동작 순서, 모듈별 책임 |
| API Reference | EvidenceChunker, EvidenceUnit, export 함수 시그니처 |
| Configuration | 파라미터 기본값 설정 근거 |
| Benchmark | baseline 대비 최종 결과 |
| Experiments | 최종 결과에 이르기까지의 전체 실험 로그 |
| QA Generation | QA 자동생성기 설계 |
| Examples | README보다 더 긴 실전 예제, 커스텀 설정·엣지케이스 처리 |