피드로 돌아가기
Query Rewriting Before Retrieval: The Cheap Recall Win Most Skip
Dev.toDev.to
AI/ML

LLM 기반 Query Rewriting을 통한 RAG Recall 최적화 및 비용 효율적 성능 개선

Query Rewriting Before Retrieval: The Cheap Recall Win Most Skip

Gabriel Anhaia2026년 6월 13일9intermediate

Context

사용자의 짧고 모호한 쿼리와 문서의 풍부한 문맥 간 Lexical Mismatch로 인한 Retrieval 성능 저하 발생. 단순 Embedding 모델 교체나 Reranker 도입은 비용 및 연산 오버헤드가 높아 효율적인 해결책이 되지 못하는 한계 존재.

Technical Solution

  • Index 진입 전 단계에 Query Rewriting 레이어를 배치하여 Search Key를 최적화하는 구조 설계
  • Multi-query Expansion을 통한 쿼리 변형 생성 및 Reciprocal Rank Fusion(RRF) 적용으로 검색 표면적 확대 및 정밀도 향상
  • Step-back Rewriting을 통해 구체적 질문을 상위 추상화 질문으로 변환하여 광범위한 컨텍스트 문서 확보
  • Parallel Search 구조를 채택하여 N개의 검색 요청 시 전체 Latency를 가장 느린 단일 요청 수준으로 유지
  • 쿼리 길이 기반 게이팅(Gating) 및 결과 캐싱(Caching)을 적용하여 불필요한 LLM 호출 비용 최소화
  • 소형 모델(Mini Model) 활용으로 고성능 모델 대비 Latency는 낮추고 리라이팅 품질은 유지하는 최적점 도달

1. 쿼리 길이가 짧은 경우에만 Rewriting을 수행하는 Gating 로직 검토

2. 단순 Union-and-Dedupe 대신 RRF 알고리즘을 통한 랭킹 퓨전 적용

3. 반복 쿼리 감소를 위한 리라이팅 결과물 전용 Cache 레이어 구축

4. 200개 이상의 Evaluation Set을 구축하여 리라이터 변경 시 Recall 지표 정량 측정

원문 읽기