kangnlp

논문 리뷰

논문 리뷰: Zero-Shot Contextual Embeddings via Offline Synthetic Corpus Generation

17분 읽기

ZEST: 합성 코퍼스로 컨텍스트 임베딩을 제로샷 적응시키기

실제 코퍼스에 접근하지 않고도, LLM이 생성한 합성 프록시 코퍼스만으로 컨텍스트 기반 임베딩 모델을 도메인에 적응시키는 프레임워크

논문 정보

항목내용
제목Zero-Shot Contextual Embeddings via Offline Synthetic Corpus Generation
저자Philip Lippmann, Jie Yang (Delft University of Technology)
학회/저널arXiv Preprint, 2025
논문 링크arXiv:2506.23662

1. 들어가며

Dense retrieval의 성능은 임베딩 품질에 달려 있다. BERT 기반 bi-encoder들이 효과적인 벡터 표현을 학습하지만, 이들에게는 구조적 한계가 하나 있다. 임베딩을 만들 때 검색 대상 코퍼스의 통계적 특성을 전혀 반영하지 않는다는 것이다. 의료 문서 검색을 하든 법률 문서 검색을 하든, 같은 가중치로 같은 방식으로 임베딩을 뽑는 셈이다.

이 문제를 해결하기 위해 Contextual Document Embeddings(CDE) 같은 컨텍스트 인식(context-aware) 아키텍처가 등장했다. 타겟 코퍼스에서 이웃 문서를 샘플링해 임베딩 생성 시 함께 참조함으로써, 도메인별 용어 빈도나 토픽 분포 같은 코퍼스 통계를 반영하는 방식이다. 실제로 이 접근은 도메인 간 전이 성능을 눈에 띄게 개선했다.

그런데 여기엔 치명적인 전제 조건이 붙는다. 추론 시점에 타겟 코퍼스에 직접 접근할 수 있어야 한다는 거다. 프라이버시 규제가 엄격한 의료 데이터, 규모가 너무 큰 대형 코퍼스, 또는 아직 구축되지 않은 신규 도메인에서는 이 조건 자체가 성립하지 않는다. 결국 컨텍스트 인식 임베딩의 우수한 성능을 가장 필요한 곳에서 쓸 수 없는 역설이 생긴다.

이 논문이 제안하는 ZEST(Zero-Shot Embeddings via Synthetic Context)는 이 역설에 대한 실용적 해법이다. 타겟 도메인을 대표하는 예시 문서 5개만 있으면, LLM을 활용해 합성 프록시 코퍼스를 오프라인으로 한 번 생성하고, 이것을 실제 코퍼스 대신 컨텍스트 인식 인코더에 넣는다. 모델 파라미터를 전혀 건드리지 않고, 타겟 코퍼스에도 접근하지 않으면서, 전체 코퍼스를 활용한 CDE 대비 NDCG@10 기준 0.29포인트(0.45%) 차이밖에 나지 않는 성능을 달성한다.


2. 기존 연구의 한계

2.1 컨텍스트 비인식(Context-Agnostic) 임베딩

Sentence-BERT, DPR, GTE, BGE 같은 표준 bi-encoder 모델들은 쿼리와 문서를 독립적으로 인코딩한다. 학습 데이터의 분포 안에서는 잘 동작하지만, 검색 대상 코퍼스가 학습 분포와 다른 도메인일 때 성능이 급락하는 코퍼스 시프트(corpus shift) 문제가 있다. BM25 같은 전통적 어휘 기반 방법은 코퍼스 통계(TF-IDF)를 자연스럽게 반영하지만, 의미적 매칭 능력이 부족하다.

2.2 컨텍스트 인식(Context-Aware) 임베딩

CDE(Morris and Rush, 2024)로 대표되는 이 계열은 타겟 코퍼스에서 이웃 문서 JJ개를 샘플링해, 1단계 인코더 M1M_1으로 컨텍스트 벡터를 만든 뒤, 2단계 인코더 M2M_2가 이를 참조하여 최종 임베딩을 생성한다. 코퍼스의 통계적 패턴을 임베딩에 녹여내는 우아한 방법이지만, 추론 시 타겟 코퍼스 접근이 필수라는 강한 가정이 실용성을 제한한다.

ColBERT 같은 late-interaction 모델도 도메인 강건성이 뛰어나지만, 토큰 수준 상호작용으로 인한 온라인 연산 비용이 크다.

2.3 비지도/테스트 타임 적응

GPL, Boot&Switch, LaPraDoR 같은 비지도 적응 방법들은 타겟 코퍼스에서 파인튜닝을 수행하므로 코퍼스 접근이 필요하다. Pseudo-relevance feedback 같은 테스트 타임 기법도 마찬가지로 온라인 최적화나 코퍼스 접근을 요구한다. ZEST는 이들과 달리 모델 가중치를 동결한 채 합성 컨텍스트만으로 적응하므로, 학습이 전혀 필요 없다.

2.4 합성 데이터를 활용한 검색

LLM으로 합성 쿼리-문서 쌍을 만들어 검색 모델을 학습시키는 연구(InPars, Promptagator, CRAFT)는 이미 활발하다. 하지만 이들은 합성 데이터를 훈련 단계에서 사용하는 반면, ZEST는 추론 단계에서 합성 코퍼스를 컨텍스트로 직접 주입한다는 점에서 근본적으로 다른 접근이다.


3. 핵심 아이디어

ZEST의 출발점은 간단한 가설이다: LLM은 소수의 예시 문서만 보고도 해당 도메인의 통계적 특성(용어 빈도, 토픽 분포, 문체)을 근사하는 합성 문서들을 생성할 수 있고, 이 합성 코퍼스가 컨텍스트 인식 모델에게 실제 코퍼스와 유사한 도메인 신호를 제공할 수 있다는 것이다.

비유하자면 LLM을 거대한 데이터베이스처럼 취급하는 거다. 도메인 키워드 몇 개로 검색하듯, 예시 문서 몇 개를 넣으면 해당 도메인에 맞는 합성 문서들을 "검색"해 오는 셈이다. 이렇게 만든 합성 코퍼스를 기존 컨텍스트 인식 인코더의 컨텍스트 입력으로 그대로 넣으면, 모델을 전혀 수정하지 않고도 도메인 적응 효과를 얻을 수 있다.

기존 방법들이 "모델을 어떻게 바꿀까"에 집중했다면, ZEST는 "입력 데이터를 어떻게 바꿀까"라는 데이터 중심(data-centric) 관점에서 문제를 풀었다는 점이 독특하다.


4. 제안 방법 (Method)

4.1 전체 구조

ZEST는 오프라인 합성(Offline Synthesis)과 온라인 추론(Online Inference) 두 단계로 구성된다. 오프라인 단계에서는 소수의 예시 문서로부터 LLM을 통해 합성 프록시 코퍼스를 생성하고, 온라인 단계에서는 기존 컨텍스트 인식 모델의 가중치를 동결한 채 이 합성 코퍼스를 컨텍스트로 활용한다.

Figure 1: Overview of the ZEST framework
Figure 1: Overview of the ZEST framework

Figure 1: ZEST 프레임워크의 전체 개요. 상단은 오프라인 합성 과정으로, 소수의 예시 문서(DexD_{ex})에서 도메인 앵커(DanchorD_{anchor})를 순차 생성한 뒤 합성 프록시 코퍼스(DsynthD_{synth})를 병렬 확장한다. 하단은 온라인 추론 과정으로, 동결된 컨텍스트 인식 모델(M1M_1, M2M_2)이 DsynthD_{synth}를 컨텍스트로 활용해 새 문서/쿼리의 도메인 적응 임베딩을 생성한다. (원논문)

Figure 1의 상단을 보면 계층적 합성 과정이 명확하게 드러난다. 예시 문서 DexD_{ex}에서 도메인 앵커 DanchorD_{anchor}를 순차적으로 만들고, 각 앵커에서 합성 문서 DsynthD_{synth}를 병렬로 확장하는 2단계 구조다. 하단의 추론 과정은 기존 CDE 아키텍처와 동일한데, 실제 코퍼스 대신 합성 코퍼스가 들어가는 것만 다르다. 이 "드롭인 교체(drop-in replacement)" 설계 덕분에 기존 CDE 모델을 아무런 수정 없이 그대로 사용할 수 있다.

4.2 배경: 컨텍스트 인식 임베딩 아키텍처

ZEST가 기반으로 삼는 CDE 아키텍처를 먼저 이해해야 한다. 일반적인 dense retrieval에서는 쿼리 qq와 문서 dd를 인코더 ϕ\phi, ψ\psi로 각각 벡터 공간에 매핑하고, 내적(dot product)으로 관련성을 평가한다:

f(d,q)=ϕ(d)ψ(q)f(d, q) = \phi(d) \cdot \psi(q)

CDE는 여기에 코퍼스 컨텍스트를 추가한다. 1단계 인코더 M1M_1이 타겟 코퍼스 DD에서 샘플링한 JJ개의 컨텍스트 문서를 처리하고, 2단계 인코더 M2M_2가 이 컨텍스트 표현과 입력 텍스트의 토큰 임베딩을 함께 받아 최종 임베딩을 생성한다:

ϕ(d;D)=M2({M1(d1),,M1(dJ)},E(d))\phi(d; D) = M_2(\{M_1(d_1), \ldots, M_1(d_J)\}, E(d))

여기서 E(d)={E(w1),,E(wT)}E(d) = \{E(w_1), \ldots, E(w_T)\}는 문서 dd의 토큰 임베딩이다. 이 구조 덕분에 최종 임베딩에 코퍼스의 도메인 특성이 반영되는데, 문제는 DD에 접근할 수 없으면 M1M_1에 넣을 컨텍스트 문서가 없다는 것이다.

4.3 오프라인 합성: 계층적 합성 코퍼스 생성

ZEST의 오프라인 단계는 도메인 앵커 생성과 코퍼스 확장 두 단계로 나뉜다.

입력: 도메인 예시 문서. 타겟 도메인을 대표하는 kk개의 예시 문서 Dex={d1ex,,dkex}D_{ex} = \{d_1^{ex}, \ldots, d_k^{ex}\}를 준비한다. 이 문서들은 타겟 코퍼스 DD와는 별개의 소스(논문에서는 BEIR 벤치마크)에서 가져온다. 정밀한 선택보다는 도메인의 일반적 특성을 전달하는 것이 목적이므로, 선택에 대한 민감도는 낮다.

Step 1: 도메인 앵커 생성. LLM에게 DexD_{ex}를 보여주고, AA개의 도메인 앵커 문서 Danchor={a1,,aA}D_{anchor} = \{a_1, \ldots, a_A\}를 순차적으로 생성시킨다. 각 앵커는 예시 문서에서 관찰되는 서로 다른 토픽이나 문체적 측면을 포착하는 간결한 텍스트다.

순차 생성이 중요한 이유가 있다. 각 앵커를 만들 때 이전에 생성된 앵커 목록을 함께 제공함으로써, LLM이 이미 다룬 테마를 피하고 새로운 측면을 탐색하도록 유도한다. 이를 통해 (i) 토픽 다양성을 명시적으로 보장하고, (ii) LLM이 특정 테마에 편향되는 모드 붕괴(mode collapse)를 방지하며, (iii) 이후 확장 단계에서 더 세밀한 의미적 제어가 가능해진다.

Step 2: 합성 코퍼스 확장. 각 앵커 aia_i에 대해 LLM이 J/AJ'/A개의 합성 문서를 생성한다. 이 단계는 앵커별로 병렬 처리가 가능하므로 속도 측면에서도 유리하다. 최종 합성 코퍼스는 모든 앵커에서 생성된 문서의 합집합이다:

Dsynth=i=1AGenerate(ai,J/A)D_{synth} = \bigcup_{i=1}^{A} \text{Generate}(a_i, J'/A)

기본 설정에서 k=5k=5개 예시 문서, A=20A=20개 앵커, 총 J=512J'=512개 합성 문서를 생성한다. 합성 문서의 평균 길이는 255.6 토큰으로, 벤치마크 문서의 평균 225.1 토큰과 비슷한 수준이다.

왜 예시 문서에서 바로 합성 코퍼스를 만들지 않고 앵커를 거치는 걸까? 예시 5개만으로 512개를 직접 뽑으면 LLM이 예시의 주요 주제 1-2개에 과도하게 집중할 위험이 크다. 앵커를 중간 단계로 두면, 먼저 도메인의 다양한 측면을 20개 테마로 분산시킨 뒤, 각 테마에서 균등하게 문서를 확장할 수 있다. 실험에서도 이 계층적 접근이 단순 생성(GSC 베이스라인)보다 일관되게 우수한 성능을 보였다.

4.4 온라인 추론: 합성 컨텍스트 활용

온라인 단계에서는 기존 CDE 모델의 가중치를 전혀 수정하지 않고, 합성 코퍼스를 실제 코퍼스 자리에 넣기만 한다.

컨텍스트 임베딩 사전 계산. DsynthD_{synth}는 고정이므로, 1단계 인코더 M1M_1의 출력을 미리 계산해 캐싱한다:

Cj=M1(djsynth),j=1,,JC_j = M_1(d_j^{synth}), \quad j = 1, \ldots, J'

이 사전 계산은 도메인당 한 번만 수행하면 된다.

최종 임베딩 생성. 새 문서 dd나 쿼리 qq가 들어오면, 토큰 임베딩과 캐싱된 합성 컨텍스트를 함께 2단계 인코더에 넣어 최종 임베딩을 생성한다:

ϕ(d;Dsynth)=M2({C1,,CJ},E(d))\phi(d; D_{synth}) = M_2(\{C_1, \ldots, C_{J'}\}, E(d))

ψ(q;Dsynth)=M2({C1,,CJ},E(q))\psi(q; D_{synth}) = M_2(\{C_1, \ldots, C_{J'}\}, E(q))

결국 온라인 추론 시 추가 비용은 기존 CDE와 동일한 M2M_2 포워드 패스 한 번뿐이다. 합성 코퍼스 생성 비용은 오프라인에서 이미 처리되었기 때문이다.


5. 실험 결과

5.1 실험 설정

  • 데이터셋: MTEB(Massive Text Embedding Benchmark) 벤치마크. Classification, Clustering, Pair Classification, Reranking, Retrieval, STS, Summarization 등 7개 태스크 카테고리를 포함한다.
  • 예시 문서 소스: BEIR 벤치마크에서 타겟 MTEB 태스크와 가장 유사한 도메인의 문서를 샘플링. 동일 태스크가 양쪽에 존재하는 경우 차순위 유사 태스크를 선택하며, 100토큰 이상 문서만 사용. 20-토큰 스팬 오버랩이 있는 문서는 교체하여 데이터 누출을 방지했다.
  • 평가 지표: NDCG@10
  • 베이스라인:
    • GTE v1.5 / BGE v1.5: 컨텍스트 비인식 bi-encoder 모델
    • CDE (J=512J=512): 실제 타겟 코퍼스에서 512개 문서를 컨텍스트로 사용 (실질적 상한)
    • Random: C4 데이터셋에서 무작위 512개 문서를 컨텍스트로 사용 (하한)
    • GSC (Generic Synthetic Context): 동일 LLM으로 합성하되, 도메인 앵커 없이 단순 프롬프팅
  • 합성 생성 LLM: GPT-4o (gpt-4o-2024-11-20), k=5k=5 예시, A=20A=20 앵커, J=512J'=512 합성 문서
  • 하드웨어: NVIDIA A100 GPU

5.2 주요 결과 (Main Results)

Task CategoryGTE v1.5BGE v1.5CDEGSCRandomZEST (ours)
Classification77.274.782.581.880.482.2
Clustering46.845.349.348.746.949.1
Pair Classification85.285.787.587.085.687.2
Reranking57.758.360.059.457.359.7
Retrieval54.152.855.254.652.855.0
STS82.081.683.382.781.483.0
Summarization31.230.832.732.130.432.3
Average62.0361.3164.3663.7662.1164.07

Table 1 (원논문): MTEB 벤치마크의 태스크 카테고리별 검색 성능. 전체 최고 성능은 볼드, 코퍼스 접근 불가 조건의 최고 성능은 밑줄. ZEST는 k=5k=5 예시와 J=512J'=512 합성 문서를 사용.

Table 1의 결과가 상당히 인상적이다. ZEST와 실제 코퍼스를 사용한 CDE 간 평균 차이가 0.29 NDCG@10에 불과하다. 전체 코퍼스에 대한 접근 권한 없이 예시 문서 5개만으로 이 수준의 성능을 달성한 것은, 타겟 코퍼스 접근이 불가능한 환경에서의 실용적 가치가 크다.

카테고리별로 뜯어보면, Classification(82.2 vs 82.5)과 Clustering(49.1 vs 49.3)에서 CDE와의 격차가 가장 작다. 이 태스크들은 문서의 전반적 토픽 분포를 파악하는 것이 중요한데, 합성 코퍼스가 이런 거시적 도메인 특성을 효과적으로 포착하고 있다고 볼 수 있다. 반면 Reranking(59.7 vs 60.0)과 Retrieval(55.0 vs 55.2)에서는 격차가 상대적으로 크다. 이 태스크들은 미세한 관련성 판별이 필요한데, 합성 코퍼스가 실제 코퍼스의 세밀한 통계까지는 완벽히 재현하지 못하는 것으로 보인다.

Random 베이스라인(62.11)이 컨텍스트 비인식 모델(GTE 62.03)과 비슷한 수준인 점도 주목할 만하다. 도메인과 무관한 랜덤 문서를 컨텍스트로 넣으면 사실상 컨텍스트가 없는 것과 다를 바 없다는 뜻이고, 이는 ZEST의 도메인 특화 합성 코퍼스가 단순한 "아무 텍스트"가 아니라 실질적인 도메인 신호를 제공하고 있음을 방증한다.

GTE v1.5 대비 ZEST의 성능 향상은 2.04 NDCG@10이다. GTE v1.5와 CDE 간 격차(2.33)의 87.6%를 ZEST가 회복한 셈인데, 모델 파라미터 업데이트 없이 이 정도 회복률을 보인 건 데이터 중심 접근의 가능성을 잘 보여준다.

GSC 베이스라인과의 비교도 중요하다. GSC(63.76) 대비 ZEST(64.07)의 0.31 포인트 우위는 도메인 앵커 기반 계층적 생성의 효과를 입증한다. 같은 LLM으로 같은 양의 합성 문서를 만들더라도, 생성 전략에 따라 품질이 달라진다는 거다.

5.3 Ablation Studies

예시 문서 수(k)의 영향

Figure 3: 예시 문서 수에 따른 성능 변화
Figure 3: 예시 문서 수에 따른 성능 변화

Figure 3: 예시 문서 수(kk)를 변화시켰을 때의 MTEB 벤치마크 평균 성능. J=512J'=512 합성 문서 고정. GSC와 비교. (원논문)

k=1k=1에서 k=5k=5로 늘리면 ZEST의 성능이 약 1포인트 가까이 올라간다(63.2 → 64.1 근방). 예시가 하나뿐이면 LLM이 해당 도메인의 한 측면에만 과적합하는데, 5개로 늘리면 다양한 관점을 포착할 수 있기 때문이다. 그러나 k=10k=10으로 더 늘려도 개선이 거의 없다. A=20A=20개 도메인 앵커가 이미 충분한 다양성을 제공하므로, 예시 수를 늘리는 것의 한계 효용이 빠르게 체감되는 것이다.

GSC도 비슷한 경향을 보이지만, 모든 kk 값에서 ZEST에 못 미친다. k=1k=1일 때 두 방법의 격차(약 0.4 포인트)가 k=5k=5에서의 격차(약 0.3 포인트)보다 크다는 점은, 예시가 적을수록 앵커 기반 구조화의 가치가 더 크다는 것을 시사한다.

앵커 수 AA에 대해서도 실험했는데, 기본값 20 근처에서 변동을 줘도 전체 MTEB 스코어에 무시할 만한 차이만 있었다고 한다.

합성 컨텍스트 크기(JJ')의 영향

Figure 2: 컨텍스트 크기에 따른 성능 변화
Figure 2: 컨텍스트 크기에 따른 성능 변화

Figure 2: 컨텍스트 크기(JJ' 또는 JJ)에 따른 MTEB 벤치마크 평균 성능. k=5k=5 예시 고정. ZEST, GSC, CDE(실제 코퍼스) 비교. (원논문)

Figure 2는 세 가지 흥미로운 패턴을 보여준다. 첫째, 세 방법 모두 컨텍스트 크기가 커질수록 성능이 향상되지만, 수확 체감(diminishing returns)이 뚜렷하다. J=16J'=16 이후부터는 개선 폭이 급격히 줄어든다. 둘째, CDE(실제 코퍼스, 초록 선)가 모든 크기에서 가장 높은 성능을 보이지만, ZEST(파란 선)와의 격차가 크기가 커질수록 좁혀진다. 작은 컨텍스트에서는 합성 문서의 품질이 특히 중요한데, ZEST의 앵커 기반 생성이 소수 문서에서도 효과적인 다양성을 확보하기 때문으로 보인다. 셋째, 모든 컨텍스트 크기에서 ZEST가 GSC(주황 선)를 일관되게 앞선다. 이 격차가 크기에 관계없이 유지된다는 것은 계층적 생성 전략의 이점이 컨텍스트 양과 독립적임을 의미한다.

J=512J'=512에서 ZEST의 평균 NDCG@10은 약 64.1로, CDE의 약 64.8과 비교해 0.7 포인트 차이다. J=128J'=128 정도에서도 63.5 이상을 달성하므로, 생성 비용을 줄이고 싶다면 JJ'를 줄이는 것도 합리적 선택이다.

5.4 LLM 선택의 영향

LLMGSCZEST
GPT-4o63.7664.07
Llama-3.3-70B-Instruct62.7162.96

Table 2 (원논문): 합성 컨텍스트 생성에 사용한 LLM에 따른 검색 성능 비교. k=5k=5, J=512J'=512.

GPT-4o가 Llama 70B 대비 GSC에서 1.05, ZEST에서 1.11 포인트 높다. LLM의 지시 따르기(instruction following) 능력과 도메인 지식이 합성 코퍼스 품질에 직접 영향을 미치는 거다. 그런데 주목할 점은 Llama 70B를 사용해도 ZEST(62.96)가 GSC(62.71)보다 여전히 0.25 포인트 앞선다는 것이다. 앵커 기반 계층적 생성 전략의 이점이 LLM 품질과 독립적으로 존재한다는 증거다.

다만 Llama 70B 기반 ZEST(62.96)가 컨텍스트 비인식 모델(GTE 62.03, BGE 61.31)보다는 확실히 높으므로, 비용이나 프라이버시 이유로 오픈소스 모델을 써야 하는 경우에도 ZEST가 여전히 유효하다.

5.5 연산 비용

합성 코퍼스 생성은 도메인당 한 번만 수행하면 되는 오프라인 프로세스다. GPT-4o API로 512개 합성 문서(평균 255.6 토큰)를 생성하는 비용은 상대적으로 저렴하다. 온라인 추론 시에는 합성 컨텍스트 임베딩 {Cj}\{C_j\}가 이미 사전 계산되어 있으므로, 기존 CDE와 동일한 M2M_2 포워드 패스 한 번만 필요하다. 저자들에 따르면 계층적 생성 과정 자체의 시간은 추론 시간 대비 무시할 수 있는 수준이었다.

이는 파인튜닝 기반 적응 방법(GPL 등)이 요구하는 상당한 GPU 시간과 비교하면 확실한 이점이다.


6. 부록: 프롬프팅 전략, 예시 문서 매핑, 합성 문서 예시

6.1 프롬프팅 전략 (Appendix A)

ZEST의 합성 코퍼스 생성에는 두 가지 프롬프트가 사용된다.

도메인 앵커 생성 프롬프트 (Figure 4): LLM에게 kk개 예시 문서와 이전에 생성된 앵커 목록을 제공하고, (1) 예시 문서의 핵심 테마, 문체, 도메인 용어를 분석하여, (2) 이전 앵커와 다른 측면을 다루는 새로운 앵커 문서를 생성하도록 지시한다. 앵커는 예시 문서와 비슷한 길이의 일관된 텍스트여야 하며, 단순 키워드 나열이 아니어야 한다.

합성 코퍼스 확장 프롬프트 (Figure 5): 각 앵커 문서를 기반으로, 해당 테마를 정교화하고 다양화하는 합성 문서를 생성하도록 지시한다. 생성된 문서는 (1) 앵커와 토픽 일관성을 유지하고, (2) 비슷한 길이의 완전한 문서 형태이며, (3) 앵커 테마의 다양한 하위 주제를 탐색하고, (4) 도메인에 맞는 문체를 유지해야 한다. 각 문서 끝에 ---DOCUMENT END---를 넣어 구분한다.

6.2 예시 문서 매핑 (Appendix B)

MTEB 태스크에 적합한 예시 문서를 선택하기 위해, 18개 BEIR 태스크 각각에 고유 도메인 키워드를 할당했다.

BEIR TaskUnique Keyword
MS MARCOWebSearch
TREC-COVIDCOVIDResearch
NFCorpusNutrition
BioASQBiomedQA
HotpotQAMultiHopQA
FiQA-2018FinanceQA
Signal-1M (RT)SocialMedia
TREC-NEWSNewsSearch
Robust04NewsArchive
ArguAnaArgumentation
Touché-2020Debate
CQADupStackForumQA
QuoraDuplicateQA
DBPedia-EntityEntityRetrieval
SCIDOCSCitation
SciFactSciFactCheck
Climate-FEVERClimateClaims
FEVERFactCheck

Table 3 (원논문): BEIR 태스크와 고유 도메인 키워드 매핑. MTEB 태스크와의 도메인 유사도 기반 정렬에 활용.

이 매핑은 각 BEIR 태스크의 주요 도메인이나 검색 목표를 키워드 하나로 요약한 것이다. 예를 들어 BioASQ와 TREC-COVID가 모두 생물의학 분야이지만 'BiomedQA'와 'COVIDResearch'로 구분해, 같은 상위 도메인 내에서도 세부 특성 차이를 반영했다. MTEB 태스크에 대해서는 이 키워드와의 도메인 매칭을 통해 가장 적합한 BEIR 소스를 수동으로 선택했다.

6.3 태스크별 접두사 (Appendix C)

CDE의 원래 방법론을 따라, 모든 평가에서 MTEB 데이터셋별 표준 접두사를 사용했다. 접두사 카테고리는 Search query, Search document, Classification, Clustering 네 가지다. 이는 실제 컨텍스트와 합성 컨텍스트 시나리오 간 일관성을 보장하기 위한 것이다.

6.4 합성 문서 생성 파이프라인 예시 (Appendix D)

논문은 생물의학과 금융 두 도메인에서의 전체 생성 파이프라인을 예시로 보여준다.

생물의학 도메인 예시:

  • 예시 문서 (Figure 6): 고세균(Archaea)에서의 유전적 리코딩(recoding)과 프로그래밍된 -1 프레임시프팅에 관한 실험 논문 초록
  • 도메인 앵커 (Figure 7): 프로그래밍된 리보솜 프레임시프팅이라는 핵심 테마를 포착하되, 이를 유전자 발현 조절 메커니즘으로 일반화하고 생명공학 응용 가능성까지 언급하는 간결한 요약
  • 합성 문서 (Figure 8): 앵커에서 분기하여, 프레임시프트 사이트의 RNA 구조적 모티프, 리보솜 일시정지 메커니즘, 게놈 수준 매핑 연구, 합성생물학 도구 개발 등으로 확장된 상세 텍스트

예시 문서의 구체적 실험 내용에서 출발해, 앵커에서 테마를 추상화하고, 합성 문서에서 다시 구체적이면서도 원래와는 다른 관점의 텍스트를 만들어내는 과정이 잘 드러난다.

금융 도메인 예시:

  • 예시 문서 (Figure 9): 고정/변동 금리 대출, 금리 상승기의 차입자 위험에 관한 실무적 설명
  • 도메인 앵커 (Figure 10): 고정 vs 변동 금리의 영향, 중앙은행 통화정책, 거시경제 요인 등을 포괄하는 도메인 요약
  • 합성 문서 (Figure 11): 금융기관의 리스크 관리 전략, 금리 스왑 헤징, 리파이낸싱 수요 패턴, 신용평가 모델 조정 등으로 확장된 전문 텍스트

합성 문서의 문체와 전문 용어 사용이 도메인에 매우 적합한 것을 확인할 수 있다. 다만 저자들도 인정하듯, 가끔 너무 일반적이거나 도메인과 덜 관련된 합성 문서가 생성되기도 한다.


7. 강점과 한계

강점

  • 실용성과 효율성. 모델 파라미터를 전혀 수정하지 않는 제로샷 적응이라는 점에서, 파인튜닝 인프라가 없는 환경에서도 바로 적용 가능하다. 합성 코퍼스 생성은 도메인당 한 번이면 충분하고, 추론 비용은 기존 CDE와 동일하다.
  • 놀라운 성능 근접도. Table 1에서 확인할 수 있듯, 실제 코퍼스 512개를 사용한 CDE 대비 0.29 포인트(0.45%) 차이밖에 나지 않는다. 예시 문서 5개라는 극히 적은 정보로 이 수준을 달성한 것은 방법론의 효과를 강하게 뒷받침한다.
  • 깔끔한 실험 설계. GSC 베이스라인으로 앵커 기반 생성의 효과를 분리하고, Random 베이스라인으로 도메인 특화의 필요성을 검증하며, 두 가지 LLM으로 생성기 의존성을 확인하는 등 ablation 설계가 체계적이다.
  • 데이터 중심 관점의 신선함. 기존의 모델 중심 적응(파인튜닝, 어댑터, 프롬프트 튜닝) 대신 입력 데이터를 합성적으로 대체한다는 패러다임이 실용적이면서도 학술적으로 흥미롭다.

한계 및 아쉬운 점

  • CDE 아키텍처 의존성. ZEST는 CDE라는 특정 컨텍스트 인식 아키텍처에 묶여 있다. CDE 자체가 137M 파라미터의 비교적 작은 모델이고, 더 크고 강력한 컨텍스트 인식 모델에서도 동일한 효과를 보일지는 검증되지 않았다. 또한 CDE가 아닌 다른 컨텍스트 인식 아키텍처(예: late-interaction 계열)와의 호환성도 미확인이다.
  • 외부 LLM API 의존. GPT-4o API에 의존하면 비용, 지연시간, 프라이버시 문제가 발생할 수 있다. Llama 70B로도 작동하지만 1.11 포인트 성능 하락이 있었다. 프라이버시가 ZEST의 주요 동기 중 하나인데, 합성 생성 시 예시 문서를 외부 API에 보내야 한다는 것은 다소 모순적이다.
  • 도메인 키워드 매핑의 수동성. BEIR 태스크에서 MTEB 태스크로의 도메인 매핑이 수동 검사(manual inspection)에 기반한다. 자동화된 도메인 유사도 측정이나 예시 문서 선택 전략이 없어, 새 도메인에 적용 시 전문가 판단이 필요하다.
  • 합성 코퍼스의 한계. 합성 문서는 실제 코퍼스의 프록시일 뿐, 진정한 코퍼스 통계를 완벽히 재현하지 못한다. LLM이나 예시 문서의 편향이 합성 코퍼스에 증폭될 수 있으며, 이 점은 민감한 응용(의료, 법률)에서 특히 주의가 필요하다.
  • 대규모 모델 비교 부재. 비교 대상 모델이 base 크기(GTE-base, BGE-base, CDE-small 137M)에 한정되어 있다. 현재 MTEB 리더보드 상위권의 대형 모델들과의 비교가 없어, ZEST의 실질적 경쟁력을 가늠하기 어렵다.

8. 마치며

ZEST의 가장 큰 기여는 컨텍스트 인식 임베딩의 활용 범위를 프라이버시 제약 환경까지 확장했다는 점이다. "코퍼스에 접근해야 한다"는 이전까지의 필수 조건을 "도메인 예시 문서 5개면 충분하다"로 대폭 완화했다. 이는 의료, 법률, 금융 등 데이터 접근이 제한된 도메인에서 고성능 검색 시스템을 배포할 수 있는 실질적 경로를 열어준다.

방법론적으로는 LLM을 "거대한 도메인 지식 데이터베이스"로 활용한다는 관점이 인상적이다. 학습 데이터 생성이 아닌 추론 시점의 컨텍스트 대체라는 독특한 활용 방식은, 향후 다른 컨텍스트 의존적 모델에도 적용할 수 있는 일반적 전략으로 확장될 가능성이 있다.

후속 연구로는 (1) 합성 코퍼스 생성에 특화된 오픈소스 LLM 파인튜닝으로 외부 API 의존성을 줄이는 방향, (2) 생성된 합성 코퍼스의 품질을 자동 평가하고 필터링/정제하는 기법 개발이 제안되었다. 개인적으로는 합성 코퍼스의 다양성을 측정하는 정량적 지표를 도입하고, 이를 최적화 목표로 삼는 접근도 흥미로울 것 같다. 또한 CDE 이외의 컨텍스트 인식 아키텍처(예: Nomic Embed 등)로의 일반화 검증도 중요한 후속 과제가 될 것이다.


References

  • Morris, J. X. and Rush, A. M. (2024). Contextual Document Embeddings. arXiv:2402.02525.
  • Thakur, N. et al. (2021). BEIR: A Heterogenous Benchmark for Zero-Shot Evaluation of Information Retrieval Models. arXiv:2104.08663.
  • Muennighoff, N. et al. (2022). MTEB: Massive Text Embedding Benchmark. arXiv:2210.07316.
  • Wang, K. et al. (2022). GPL: Generative Pseudo Labeling for Unsupervised Domain Adaptation of Dense Retrieval. arXiv:2112.07577.
  • Shao, R. et al. (2025). ReasonIR: Training Retrievers for Reasoning Tasks. arXiv:2504.20595.
  • Dai, Z. et al. (2022). Promptagator: Few-shot Dense Retrieval From 8 Examples. arXiv:2209.11755.
  • Jiang, F. et al. (2023). Boot and Switch: Alternating Distillation for Zero-Shot Dense Retrieval. EMNLP Findings.
  • Karpukhin, V. et al. (2020). Dense Passage Retrieval for Open-Domain Question Answering. arXiv:2004.04906.