본문으로 건너뛰기

트렌딩 - GitHub 인기 레포 & HuggingFace 모델

PaddlePaddle/HPD-Parsing

문서 이미지를 레이아웃과 본문으로 동시에 파싱해 마크다운으로 변환하는 문서 파싱(OCR·레이아웃 이해)1Bapache-2.0

1B 모델로 문서 파싱 SOTA와 자기회귀 베이스라인 대비 3배 처리량을 함께 달성한 계층적 병렬 디코딩 파서

학습 방식 · InternVL3.5-1B 백본에 계층적 병렬 디코딩(HPD)과 P-MTP 멀티토큰 예측 헤드를 결합해 difficulty-aware 데이터로 단계적 적응 학습했다.

TL;DR

HPD-Parsing은 InternVL3.5-1B를 백본으로 쓰는 1B급 문서 파싱 모델로, 페이지 구조를 조율하는 레이아웃 branch와 영역별로 동시에 디코딩하는 콘텐츠 branch를 분리해 순차적 생성의 병목을 없앤 계층적 병렬 디코딩(HPD) 구조를 쓴다. 각 branch 안에서는 Progressive Multi-Token Prediction으로 한 번에 여러 토큰을 예측해 디코딩 단계를 추가로 줄이며, difficulty-aware 데이터 큐레이션으로 기존 자기회귀 파싱 능력을 이 새 디코딩 방식으로 단계적으로 이전했다. 그 결과 OmniDocBench v1.6에서 통합 파서 중 최고 점수인 94.91%를 내면서도 A800 배치 512 기준 초당 4,752토큰으로 자체 자기회귀 베이스라인의 3.06배, 기존 최속 파서 DeepSeek-OCR-2의 1.62배 처리량을 달성했고 문서가 길어질수록 이 격차는 더 커진다. 다만 이 처리량을 그대로 얻으려면 커스텀 vLLM 빌드가 필요하며, 표준 transformers 경로는 참조 구현으로 단일 이미지·배치 1만 지원한다.

핵심 포인트

  • 레이아웃 branch가 페이지 구조를 전역 조율하고 콘텐츠 영역별로 로컬 branch를 동시에 디코딩해 순차적 병목을 없앤다.
  • 각 branch 안에서 P-MTP가 한 번에 여러 토큰을 예측해 디코딩 단계 수를 추가로 줄인다.
  • 1B급 소형 모델로 OmniDocBench v1.6에서 통합 파서 SOTA를 기록하면서도 초당 4752토큰의 처리량을 낸다.
  • 문서가 길어질수록 디코딩 단계 감소·요청 처리량 증가 효과가 커져 대량 문서 처리에 유리하다.

제한사항

  • 커스텀 vLLM 빌드(v0.17.1 기반)가 필요해 표준 vLLM으로는 병렬 디코딩 성능을 낼 수 없다.
  • transformers 경로의 참조 구현은 단일 이미지·배치 크기 1만 지원해 실제 처리량은 vLLM 서빙에 못 미친다.

벤치마크

벤치마크지표비교
OmniDocBench v1.6overall accuracy94.91%엔드투엔드 통합 파서 중 SOTA
Peak throughputTPS (A800 80GB, batch 512)4752자체 자기회귀 베이스라인 1,554.8 대비 3.06배, 기존 최속 파서 DeepSeek-OCR-2 대비 1.62배
Batch 512 처리량PPS2.68자체 자기회귀 베이스라인 1.02 대비 2.62배, DeepSeek-OCR-2 대비 1.31배

74

LIKES

1.3k

DOWNLOADS

0 / 0

조회수

관련 토론

아직 관련 토론이 없습니다.

댓글

댓글을 작성하려면 로그인이 필요합니다.