📚 my-arxiv
← Community

Loop Transformer

@hizieun·1개월 전(수정됨)

GPT-6 Astra가 쓴 Loop Transformer (Recurrent Depth) 정리

한 줄 요약

같은 트랜스포머 블록을 여러 번 반복 통과시켜, 파라미터를 늘리지 않고도 "더 깊게 생각"하게 만드는 구조. OpenAI는 이를 recurrent depth(순환 깊이) 라 부르지만, 연구계에서 몇 년째 있던 looped transformer 아이디어의 리브랜딩이다.

어떻게 동작하나

  • 입력 토큰이 트랜스포머 블록 스택(예: 22개)을 통과한다.
  • 1회 통과가 끝나면 출력 hidden state를 다시 맨 앞으로 되먹임한다.
  • 같은 22개 블록이 그 중간 표현을 한 번 더 처리한다. (가중치는 동일, 입력 데이터만 바뀜)
  • 즉 Nanbeige4.2-3B의 경우: 22블록을 2번 → 총 44번의 "블록 적용"이지만 저장하는 가중치는 22개뿐.

왜 도움이 되나

  1. 파라미터 효율 — 블록을 2번 재사용하면 블록을 2배(44개)로 늘린 모델 대비 파라미터가 약 50% 적다 → 메모리 절감.
  2. 추론 능력 — 저장 공간(파라미터)을 늘리지 않아도 연산량(computation)을 더 쓰는 것만으로 다단계 추론이 개선됨. "저장이 아니라 계산이 핵심"이라는 가설.

헷갈리기 쉬운 것과의 차이

비교 대상차이
Chain-of-ThoughtCoT는 보이는 추론 토큰을 생성. 루프는 내부 hidden state에서의 연산. 둘은 별개이며 함께 쓸 수 있음.
그냥 레이어 추가서로 다른 44블록을 쌓으면 파라미터·메모리가 2배. 루프는 절반의 파라미터로 비슷한 "계산 깊이"를 얻음(단, 학습 시 forward/backward 연산량은 동일).

트레이드오프 / 한계

  • KV 캐시는 안 줄어든다 — 패스마다 별도 캐시가 필요해 44블록 모델과 동일한 수준. (Nanbeige는 패스 간 KV 캐시 공유 시 오히려 성능 하락)
  • 학습 비용은 여전히 비쌈 — 반복 적용 전부에 그래디언트가 흘러야 함.
  • 최적 루프 횟수는 태스크 의존적 — Nanbeige는 2회가 가장 효율적이라 결론.

"추론을 숨긴다"는 소문에 대해

Sebastian Raschka는 루프가 chain-of-thought를 숨기는 주범은 아니라고 봄. 추론 흔적이 짧아진 건 은폐가 아니라, 모델이 더 유능해져 실수·백트래킹이 줄었기 때문 — "잘 준비된 학생일수록 연습장을 덜 쓴다".

GPT-6 Astra에 대한 추정

OpenAI 수석과학자: "Astra를 포함한 현재 프론티어 모델의 계산 그래프 깊이는 GPT-4의 2배 이내" → 2회 루핑과 일관되지만, 아키텍처 선택을 확정적으로 증명하진 않음.

참고 계보

  • Universal Transformers (2018) — 토큰별로 루프 횟수를 다르게 하는 adaptive halting의 시초
  • Mixture-of-Recursions (2025) — MoE식 라우터로 토큰별 루프 횟수 할당
  • Latent Reasoning (2025) — 학습 중 루프 횟수를 가변(8~64)으로 두어 다양한 추론 예산에 대비
  • SMELT (2026.09) — looped MoE가 동일 성능에 학습 연산 6.8~18% 절감

출처:

  • Sebastian Raschka, GPT-6 Astra, Looped Transformers, and Hidden Reasoning
  • arXiv: Universal Transformers / Mixture-of-Recursions / LoopMTP(2608.03624)

소스:

댓글 0

로그인하고 댓글을 남겨보세요.

아직 댓글이 없어요.