본문으로 건너뛰기

Answer-Backtracked Credit Assignment(ABC) 기반 ABSeeker

정답에서 역추적한 검증 가능한 단서로 각 단계에 밀집 보상을 부여해 4B 검색 에이전트의 장기 탐색 성능을 개선

왜 중요한가

장기 탐색 과제에서는 최종 정답만으로 학습하면 중간 단계의 유효성과 오류를 구분할 수 없어 잘못된 행동이 강화되거나 유용한 행동이 약화된다. ABSeeker는 정답에서 역추적해 검증 가능한 단서를 만들고 이를 기준으로 각 단계에 밀집 보상을 부여해 실패 궤적에서도 유용한 탐색을 강화하고 성공 궤적의 오류를 억제한다. 이러한 절차 감독은 같은 모델 규모에서 검색 정확도와 탐색 깊이를 동시에 끌어올려 대규모 모델에 준하는 성능을 얻을 수 있게 한다.

핵심 기여

답변 역추적 기반 단계별 신용 할당(ABC)

정답을 출발점으로 삼아 해당 정답으로 이어지는 중간 엔티티·사실·관계를 LLM 도구 호출 체계로 역추적하고, 검증된 단서 집합을 고정된 평가 기준으로 사용해 모든 단계에 대해 독립적인 스코어를 산출하는 프레임워크를 도입했다. 스코어는 기본 1.0에 단서 발견(+0.8), 후보 배제(+0.4), 잘못된 배제(−0.8), 정답 제출(+1.0), 오답 제출(−1.0) 등 델타를 누적하고 [0,2.0]로 클리핑하는 룰을 따른다. 이 접근법은 실패한 궤적에서도 가치 있는 행동을 긍정적으로 보상하고 성공 궤적의 오류는 감소시켜 단계별로 세밀한 학습 신호를 제공한다.

단계 가중 SFT와 단계 보상 기반 GRPO(ABC-SFT, ABC-GRPO)

회수한 단계별 보상을 SFT 손실과 RL 보상으로 직접 연결하는 두 가지 학습 절차를 적용했다. SFT에서는 각 단계의 토큰 로그확률에 가중치 w(r_t)=σ(α(r_t−β))를 곱해 고득점 단계가 더 큰 그래디언트를 갖도록 하고, 이후 RL 단계에서는 단계별 정규화 보상에 할인계수 γ를 적용해 단계별 advantage Ai,t=k=tTiγktR^i,kA_{i,t}=\sum_{k=t}^{T_i}\gamma^{k-t}\widehat{R}_{i,k}를 계산해 GRPO로 정책을 갱신한다. 이 연속적 학습 파이프라인은 초기 SFT로 안정화를 돕고 단계별 RL로 탐색·정책을 개선하는 방식으로 작동한다.

소규모(4B) 모델로의 실용적 성능 개선

Qwen3.5-4B를 기반으로 ABSeeker를 8.5K 궤적으로 SFT한 뒤 ABC-GRPO로 추가 최적화하여 BrowseComp에서 문맥 관리 없는 상태 37.3%, 문맥 관리 적용 시 55.3%를 기록했고 BrowseComp-ZH는 39.1%→52.9%로 향상했다. xbench 및 GAIA-text에서도 77.0%와 81.6%를 달성해 동일 규모(4B) 내부 최고 성능을 보였고 일부 대형(∼30B) 시스템과도 경쟁할 수 있는 수준을 나타냈다. 이러한 결과는 단계별 절차 감독(stepwise process supervision)이 동일한 파라미터 예산에서 검색 능력과 장기 탐색을 동시에 개선함을 시사한다.

핵심 아이디어 이해하기

검색 문제는 최종 정답이 주어지면 그 정답과 질의를 잇는 중간 증거 경로가 존재한다는 성질을 가진다. 이 논문은 정답에서 출발해 필요한 중간 엔티티·사실·관계를 역추적해 검증 가능한 단서 집합을 만든 뒤, 각 전진 탐색 단계가 해당 단서를 발견·검증·배제하는지를 기준으로 점수를 부여하는 방법을 사용한다. 단계별 점수는 궤적 전체의 성공 여부와 무관하게 동작을 평가하므로 실패 궤적의 유효한 탐색도 강화되고 성공 궤적의 오류는 억제되며, 결과적으로 단계 단위의 세밀한 학습 신호로 정책을 효과적으로 조정할 수 있다.

방법론

전체 파이프라인은 두 단계로 구성된다. 첫째, Answer-Backtracked Clue Recovery 단계에서 LLM(DeepSeek-V4-Flash 백본)을 동일한 도구 호출 프로토콜로 웹을 재탐색하면서 질의와 검증된 정답을 연결하는 중간 단서를 찾아 실제 웹 근거로 검증해 단서 집합 C를 생성한다. 둘째, Clue-Anchored Step Scoring 단계에서 각 단계 s_t의 추론·도구 호출·응답과 단서 집합 C를 입력으로 받아 기본 점수 1.0에 표(Table 1)에 정의된 델타를 누적하고 다음 식으로 클리핑해 단계 보상 r_t를 산출한다: rt=clip(1.0+jAtΔj,0,2.0)r_t=\operatorname{clip}\big(1.0+\sum_{j\in\mathcal{A}_t}\Delta_j,\,0,\,2.0\big). 산출된 r_t는 ABC-SFT의 가중치 및 ABC-GRPO의 단계별 보상으로 직접 사용되며 SFT-가중치 매핑은 w(rt)=σ(α(rtβ))w(r_t)=\sigma(\alpha\cdot(r_t-\beta)) 형태로 처리되어 고득점 단계의 손실 기여도를 증가시킨다. 이후 RL 단계에서는 각 궤적 그룹 내 보상을 정규화하고 할인계수 γ를 적용해 단계별 advantage를 계산한 뒤(도구 응답은 옵티마이즈에서 마스킹) GRPO로 업데이트한다.

관련 Figure

ABSeeker 파이프라인 개요 다이어그램.
Diagram

왼쪽에서 오른쪽으로 Answer-Backtracked Clue Recovery와 Trajectory Generation, Clue-Anchored Step Scoring, 그리고 ABC-SFT/ABC-GRPO의 학습 단계가 흐름도로 표현되어 있다. 도식은 정답 역추적으로 고정된 단서 집합을 만들고 이를 각 단계의 보상으로 변환해 SFT와 GRPO에 연결하는 처리 흐름(input→process→output)을 명확히 보여준다. 이 그림은 논문의 핵심 메커니즘을 한눈에 파악하도록 해 후속 실험에서 어떤 신호가 학습에 사용되는지 연결 관계를 확인하는 근거로 사용하기에 적합하다.

ABSeeker 파이프라인 개요 다이어그램.

구체적 질의 예시와 그에 대한 역추적 단서 및 샘플 단계별 채점 사례.
Diagram

질의와 정답(CeraVe)을 출발점으로 L'Oréal, Ceramides, 창업자 등 6개의 단서를 회수한 사례를 제시하고, 특정 단계들이 어떤 단서를 발견하거나 잘못 배제해 어떤 보상을 받았는지 예시로 보여준다. 이 예시는 단계 점수 규칙(+0.8, −0.8 등)과 클리핑 동작이 실제 궤적에서 어떻게 적용되는지(예: 두 개의 +0.8가 누적되어 2.0으로 클리핑된 사례)를 구체적으로 연결해 이해를 돕는다. 또한 회수 모델이 동일한 도구 호출 프로토콜을 사용해 역추적을 수행함을 시각적으로 확인할 수 있다.

구체적 질의 예시와 그에 대한 역추적 단서 및 샘플 단계별 채점 사례.

주요 결과

ABSeeker는 Qwen3.5-4B 백본으로 8.5K 궤적 SFT와 ABC-GRPO를 거쳐 BrowseComp에서 문맥 관리 없이 37.3%를, 문맥 관리 적용 시 55.3%를 기록했고 BrowseComp-ZH는 39.1%에서 52.9%로 향상했다. xbench-2505에서 77.0%, xbench-2510에서 46.0%, GAIA-text에서 81.6%를 달성해 동일 규모(4B) 대비 최고 성능을 보였고 일부 ∼30B 에이전트와도 경쟁 가능한 결과를 얻었다. 분석 지표로는 단계 보상 분포에서 실패 궤적의 약 10% 단계가 r_t>1.0으로 유용한 탐색을 포함함이 드러났고, RL 학습에서는 ABC-GRPO가 표준 GRPO보다 학습 초반부터 점수와 평균 탐색 턴 수를 동시에 증가시켜 더 긴 탐색과 높은 정확도를 달성했다.

관련 Figure

SFT에 사용된 8.5K 궤적에서 단계별 보상 분포를 성공/실패 궤적으로 나눠 시각화한 막대 그래프.
Chart

성공 궤적에서도 약 4%의 단계가 r_t<1.0으로 낮게 평가되며, 실패 궤적에서는 약 10%의 단계가 r_t>1.0으로 유용한 단계를 포함하는 것으로 나타나 단계별 보상이 궤적 수준 신호의 한계를 보완함을 정량적으로 보여준다. 그래프는 시간축 상의 위치(궤적 초중후반)에 따른 보상 분포를 나타내며, 이는 ABC가 초기 탐색·후기 결정 등 단계별 역할을 다르게 보정할 수 있다는 근거가 된다. 이러한 분포 증거는 실패 궤적의 유용한 행동을 보강하고 성공 궤적의 오류를 억제한다는 논리적 연결을 제공한다.

SFT에 사용된 8.5K 궤적에서 단계별 보상 분포를 성공/실패 궤적으로 나눠 시각화한 막대 그래프.

학습 스텝별 BrowseComp 성능 비교(표준 GRPO vs ABC-GRPO)와 평균 상호작용 턴 수 변화 그래프.
Chart

그래프는 ABC-GRPO가 학습 진행 중 표준 GRPO보다 높은 BrowseComp 점수를 유지하며 동시에 평균 상호작용 턤 수가 증가함을 보여준다. 이 결과는 단계 보상이 정책을 더 길고 탐색적인 행동으로 유도하면서도 정확도를 개선한다는 작동 메커니즘을 지원하는 증거로 해석할 수 있다. 성능 곡선의 차이는 ABC가 제공하는 세밀한 단계 신호가 탐색-활용 균형을 바꾸는 실험적 근거가 된다.

학습 스텝별 BrowseComp 성능 비교(표준 GRPO vs ABC-GRPO)와 평균 상호작용 턴 수 변화 그래프.

학습 스텝별 평균 턴 수(탐색 길이) 곡선으로 ABC-GRPO가 표준 GRPO보다 더 긴 탐색을 학습함을 나타낸 그래프.
Chart

곡선은 ABC-GRPO가 훈련 초반부터 평균 턴 수를 점차 늘려 120회 이상까지 도달하는 반면 표준 GRPO는 상대적으로 완만한 증가를 보이는 것을 나타낸다. 이는 단계별 보상이 단기적인 정답 제출보다 증거 축적을 장려해 더 긴 탐색을 유도함을 의미하며, 긴 탐색이 오히려 정확도 향상으로 이어지는 실험적 근거와 연결된다. 그래프의 변동은 롤아웃 방식과 보상 정규화가 탐색 행동의 다양성에 영향을 줌을 시사한다.

학습 스텝별 평균 턴 수(탐색 길이) 곡선으로 ABC-GRPO가 표준 GRPO보다 더 긴 탐색을 학습함을 나타낸 그래프.

문맥 관리(context management) 예산별 BrowseComp/BrowseComp-ZH 성능 변화를 보여주는 선 그래프.
Chart

문맥 길이를 256K 토큰으로 확장하고 discard-all 전략을 적용했을 때 BrowseComp 성능이 37.3%에서 55.3%로, BrowseComp-ZH가 39.1%에서 52.9%로 크게 향상된 점을 시각적으로 보여준다. 이 결과는 ABSeeker의 단계 보상과 문맥 유지 정책이 상호 보완적으로 동작해 장기 정보 축적에서 큰 이득을 준다는 증거로 해석할 수 있다. 또한 문맥 관리의 효과가 작은 모델에서도 성능 개선의 핵심 변수임을 정량적으로 확인할 수 있다.

문맥 관리(context management) 예산별 BrowseComp/BrowseComp-ZH 성능 변화를 보여주는 선 그래프.

기술 상세

훈련 데이터와 구성은 논문 본문에 명시된 대로 Qwen3.5-4B를 백본으로 사용하고 SFT는 8.5K 궤적에 대해 3 epoch로 수행했으며, RL 단계에서는 검증 질문 1,000개에 대해 각 8개의 롤아웃을 샘플링했다. 단계 점수 산출은 표의 델타 집합을 합산해 기본 1.0에 더하고 rt=clip(1.0+Δ,0,2.0)r_t=\operatorname{clip}(1.0+\sum\Delta,0,2.0)로 클리핑하는 방식이며 예컨대 두 개의 +0.8 보상이 누적되면 1.0+1.6=2.6이 되어 2.0으로 클리핑된다. ABC-SFT의 손실은 단계별 토큰 로그확률에 가중치 w(rt)=σ(α(rtβ))w(r_t)=\sigma(\alpha(r_t-\beta))를 곱해 합산하는 형태로 주어지며 w의 sharpness α와 기준 β로 고득점 단계의 기여도를 조절한다. RL 단계에서는 롤아웃 그룹 내에서 보상 정규화를 거쳐 할인계수 γ로 Ai,t=k=tTiγktR^i,kA_{i,t}=\sum_{k=t}^{T_i}\gamma^{k-t}\widehat{R}_{i,k}를 계산하여 토큰 단위로 할당한 뒤 GRPO의 클리핑된 objective로 업데이트하며, 환경 제공 도구 응답 토큰은 최적화에서 마스크된다. 실험상의 운영 제한으로 궤적 최대 단계 수는 200이며 문맥 관리는 최대 256K 토큰과 최대 5회 discard-all 전략으로 수행되었다.

한계점

실험은 연산 제약으로 4B급 모델에서만 실시되어 대형 모델에 대한 확장성 검증은 남아 있다. 또한 이 접근법은 정답이 고유하고 검증 가능한 벤치마크에서 잘 작동하도록 설계되어, 정답이 모호하거나 검증 불가능한 문제에는 단서 역추적이 어려울 수 있다. 답변 역추적과 단계 점수 판정은 추가 LLM(DeepSeek-V4-Flash)에 의존하므로 그 자체의 오류가 전체 보상 품질에 영향을 줄 수 있으며 이러한 외부 스코어러의 편향·오류 분석이 추가로 필요하다.

실무 활용

검증된 정답을 이용해 중간 증거를 회수하고 이를 기준으로 단계별 보상을 만들면 장기 탐색 에이전트를 적은 데이터·작은 모델 크기에서도 효과적으로 훈련할 수 있다. 특히 웹 브라우징·문서 통합·다중 제약 검색 같은 작업에서 유용한 미세한 피드백을 제공한다.

  • 장기 웹 탐색 및 정보 수집 에이전트의 효율적 학습 — 단계별로 유용한 행동을 강화하고 오류를 억제하여 정확도를 높인다.
  • 제한된 계산 자원으로 동작해야 하는 소형 모델(4B급)에서 대형 모델에 버금가는 성능을 얻고자 할 때의 학습 전략으로 활용할 수 있다.
  • 데이터셋의 정답이 고유하고 검증 가능한 경우(예: BrowseComp 계열) 프로세스 감독을 자동으로 생성해 보상 신호를 보강하는 파이프라인으로 사용 가능하다.

코드 공개 여부: 공개

코드 저장소 보기

키워드

검색 에이전트단계별 보상답변 역추적GRPOQwen3.5-4B

용어 해설

단서 기반 단계 점수화(Clue-Anchored Step Scoring)
정답에서 역추적한 검증 가능한 단서 집합을 기준으로 각 탐색 단계의 행동을 1.0을 기본으로 점수화하고, 단서 발견(+0.8), 잘못된 배제(−0.8) 등 델타를 누적한 뒤 [0,2.0]로 클리핑하여 단계별 보상을 만든다.
답변 역추적 단서 회복(Answer-Backtracked Clue Recovery)
검증된 정답과 질의를 입력으로 삼아 LLM이 도구 호출을 통해 웹을 다시 조회하면서 정답으로 이어지는 중간 엔티티·사실·관계를 찾아내고 각 단서를 실제 웹 근거로 검증해 고정된 평가 기준을 생성한다.
ABC 기반 보상 가중 SFT(ABC-SFT)
수집한 궤적의 모든 단계 토큰에 대해 단계 보상 r_t를 w(r_t)=σ(α(r_t−β))로 매핑한 가중치를 곱해 교사 신호의 기여도를 조정하는 supervised fine-tuning 절차이다.
ABC 기반 단계 보상 GRPO(ABC-GRPO)
단계별 스코어 r_{i,t}를 정규화 후 미래할인 계수를 적용해 단계별 advantage를 계산하고(마스킹된 도구 응답 제외) GRPO의 궤적 수준 이득 대신 단계별 이득으로 정책을 최적화하는 RL 단계이다.
AI 분석 전체 내용 보기

AI 요약 · 북마크 · 개인 피드 설정 — 무료

출처 · 인용 안내

원문 발행 2026. 08. 05.수집 2026. 08. 07.출처 타입 PAPER

인용 시 "요약 출처: AI Trends (aitrends.kr)"를 표기하고, 사실 확인은 원문 보기 기준으로 진행해 주세요. 자세한 기준은 운영 정책을 참고해 주세요.