Daily Arxiv

전 세계에서 발간되는 인공지능 관련 논문을 정리하는 페이지 입니다.
본 페이지는 Google Gemini를 활용해 요약 정리하며, 비영리로 운영 됩니다.
논문에 대한 저작권은 저자 및 해당 기관에 있으며, 공유 시 출처만 명기하면 됩니다.

Evaluating Embedding Models and Pipeline Optimization for AI Search Quality

Created by
  • Haebom
Category
Empty

저자

Philip Zhong, Kent Chen, Don Wang

개요

AI 기반 검색 시스템을 위해 다양한 텍스트 임베딩 모델과 파이프라인 구성을 평가합니다. sentence-transformer 및 생성적 임베딩 모델(예: All-MPNet, BGE, GTE, Qwen)을 다양한 차원, 인덱싱 방법(Milvus HNSW/IVF), 청크 전략으로 비교합니다. 미국 시의회 회의록에서 LLM을 사용하여 11,975개의 질의-청크 쌍으로 구성된 맞춤형 평가 데이터 세트를 합성했습니다. 데이터 파이프라인에는 전처리, 청크당 자동 질문 생성, 수동 검증 및 CI/CD 통합이 포함됩니다. Top-K 정확도와 정규화된 할인 누적 이득(NDCG)을 사용하여 검색 정확도를 측정합니다. 더 높은 차원의 임베딩이 검색 품질을 향상시키고, 신경 재순위 모델이 순위 정확도를 더욱 개선하며, 더 세분화된 청크 분할이 정확도를 향상시킵니다.

시사점, 한계점

더 높은 차원의 임베딩(Qwen3-Embedding-8B/4096)이 검색 정확도를 크게 향상시킴.
신경 재순위 모델(BGE cross-encoder)이 순위 정확도를 추가로 향상시킴.
더 세분화된 청크 분할(512자)이 정확도를 향상시킴.
파이프라인 자동화 및 평가에 대한 추가 연구 필요.
👍